From 87ec582546d5da31ed5c29e8a9f0f42381e73101 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Sat, 17 Aug 2024 15:10:12 +0800
Subject: [PATCH] core: fix testdata generate realtime update

---
 packages/hydrooj/src/handler/record.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/hydrooj/src/handler/record.ts b/packages/hydrooj/src/handler/record.ts
index 13ea1c691..3a7cc9023 100644
--- a/packages/hydrooj/src/handler/record.ts
+++ b/packages/hydrooj/src/handler/record.ts
@@ -379,7 +379,7 @@ class RecordDetailConnectionHandler extends ConnectionHandler {
     async prepare(domainId: string, rid: ObjectId) {
         const rdoc = await record.get(domainId, rid);
         if (!rdoc) return;
-        if (rdoc.contest && rdoc.input === undefined) {
+        if (rdoc.contest && ![record.RECORD_GENERATE, record.RECORD_PRETEST].some((i) => i.toHexString() === rdoc.contest.toHexString())) {
             this.tdoc = await contest.get(domainId, rdoc.contest);
             let canView = this.user.own(this.tdoc);
             canView ||= contest.canShowRecord.call(this, this.tdoc);