fix: static-page compatibility

web-infra-dev · Feb 5, 2025 · 05d0dc3 · 05d0dc3
1 parent c46dbf3
commit 05d0dc3
Show file tree

Hide file tree

Showing 7 changed files with 36 additions and 174 deletions.
diff --git a/packages/midscene/src/ai-model/prompt/util.ts b/packages/midscene/src/ai-model/prompt/util.ts
@@ -322,7 +322,7 @@ export async function descriptionOfTree<
     const indentStr = '  '.repeat(indent);
 
     let children = '';
-    for (let i = 0; i < node.children.length; i++) {
+    for (let i = 0; i < (node.children || []).length; i++) {
       const childContent = buildContentTree(node.children[i], indent + 1);
       if (childContent) {
         children += `\n${childContent}`;
@@ -405,11 +405,11 @@ export async function describeUserPage<
   const idElementMap: Record<string, ElementType> = {};
   const flatElements: ElementType[] = [];
   function dfsTree(node: ElementTreeNode<ElementType>) {
-    if (node.node) {
+    if (node?.node) {
       idElementMap[node.node.id] = node.node;
       flatElements.push(node.node);
     }
-    for (let i = 0; i < node.children.length; i++) {
+    for (let i = 0; i < (node.children || []).length; i++) {
       dfsTree(node.children[i]);
     }
   }

diff --git a/packages/midscene/tests/ai/connectivity.test.ts b/packages/midscene/tests/ai/connectivity.test.ts
@@ -7,6 +7,11 @@ import dotenv from 'dotenv';
 import { getFixture } from 'tests/utils';
 import { beforeAll, describe, expect, it, vi } from 'vitest';
 
+dotenv.config({
+  debug: true,
+  override: true,
+});
+
 vi.setConfig({
   testTimeout: 20 * 1000,
 });

diff --git a/packages/midscene/tests/ai/evaluate/assertion.test.ts b/packages/midscene/tests/ai/evaluate/assertion.test.ts
@@ -9,7 +9,12 @@ import {
   repeatFile,
 } from './test-suite/util';
 import 'dotenv/config';
-import { repeatTime } from '../util';
+import dotenv from 'dotenv';
+
+dotenv.config({
+  debug: true,
+  override: true,
+});
 
 const testSources = [
   'online_order',

diff --git a/packages/midscene/tests/ai/evaluate/computer.test.ts b/packages/midscene/tests/ai/evaluate/computer.test.ts
diff --git a/packages/midscene/tests/ai/evaluate/inspect.test.ts b/packages/midscene/tests/ai/evaluate/inspect.test.ts
@@ -14,8 +14,14 @@ import {
   runTestCases,
 } from './test-suite/util';
 import 'dotenv/config';
+import dotenv from 'dotenv';
 import { repeatTime } from '../util';
 
+dotenv.config({
+  debug: true,
+  override: true,
+});
+
 const relocateAfterPlanning = false;
 const failCaseThreshold = process.env.CI ? 1 : 0;
 const testSources = [

diff --git a/packages/midscene/vitest.config.ts b/packages/midscene/vitest.config.ts
@@ -10,6 +10,8 @@ import { version } from './package.json';
  */
 dotenv.config({
   path: path.join(__dirname, '../../.env'),
+  override: true,
+  debug: true,
 });
 
 const enableAiTest = Boolean(process.env.AITEST);

diff --git a/packages/web-integration/src/playground/static-page.ts b/packages/web-integration/src/playground/static-page.ts
@@ -17,7 +17,20 @@ export default class StaticPage implements AbstractPage {
   private uiContext: WebUIContext;
 
   constructor(uiContext: WebUIContext) {
-    this.uiContext = uiContext;
+    if (uiContext.tree) {
+      this.uiContext = uiContext;
+    } else {
+      const contents = uiContext.content;
+      this.uiContext = Object.assign(uiContext, {
+        tree: {
+          node: null,
+          children: contents.map((content) => ({
+            node: content,
+            children: [],
+          })),
+        },
+      });
+    }
   }
 
   async getElementsInfo() {