From 5031979ca9c46ed00597a36f6412e321f6fdcdcf Mon Sep 17 00:00:00 2001
From: brother7 <7brother7@gmail.com>
Date: Tue, 25 Aug 2026 18:00:02 +0800
Subject: [PATCH] test(pi): hydrate packaged fork source before proof

---
 scripts/run-pi-subagent-packaged-smoke.mjs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/scripts/run-pi-subagent-packaged-smoke.mjs b/scripts/run-pi-subagent-packaged-smoke.mjs
index 209d654..b381c54 100644
--- a/scripts/run-pi-subagent-packaged-smoke.mjs
+++ b/scripts/run-pi-subagent-packaged-smoke.mjs
@@ -476,6 +476,14 @@ export async function runPackagedProductProof(options) {
       || settledProxy?.proxy?.activeProviderRequests?.child !== 0) {
       throw new Error(`Packaged proxy source did not settle before fork: ${JSON.stringify(settledProxy?.proxy)}`);
     }
+    await page.reload();
+    await page.waitForLoadState('domcontentloaded');
+    if (await page.getByTestId('ai-module-selection-page').count()) {
+      await page.getByTestId('ai-module-option-programming').click();
+      await page.getByTestId('main-layout').waitFor({ state: 'visible', timeout: 10_000 });
+    }
+    await page.evaluate(() => { window.location.hash = '/chat'; });
+    await page.getByText('REAL_PARENT_COMPLETE').waitFor({ state: 'visible', timeout: 30_000 });
     const timeline = page.getByTestId('coding-conversation-timeline');
     const forkActions = timeline.getByRole('button', { name: '从这里创建新对话分支' });
     await forkActions.waitFor({ state: 'visible', timeout: 30_000 });