chore: update cache test

web-infra-dev · Sep 4, 2024 · 726802a · 726802a
1 parent 0165108
commit 726802a
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 20 deletions.
diff --git a/packages/web-integration/playwright.config.ts b/packages/web-integration/playwright.config.ts
@@ -18,7 +18,7 @@ dotenv.config({
  */
 export default defineConfig({
   // testDir: './tests/ai/e2e',
-  // testIgnore: 'generate-test-data.spec.ts',
+  testIgnore: 'generate-test-data.spec.ts',
   timeout: 900 * 1000,
   /* Run tests in files in parallel */
   fullyParallel: true,

diff --git a/packages/web-integration/src/common/task-cache.ts b/packages/web-integration/src/common/task-cache.ts
@@ -45,7 +45,7 @@ export type AiTaskCache = {
 };
 
 export class TaskCache {
-  cache: AiTaskCache | undefined;
+  cache: AiTaskCache;
 
   cacheId: string;
 
@@ -56,7 +56,9 @@ export class TaskCache {
   constructor(opts?: { fileName?: string }) {
     this.midscenePkgInfo = getMidscenePkgInfo(__dirname);
     this.cacheId = generateCacheId(opts?.fileName);
-    this.cache = this.readCacheFromFile();
+    this.cache = this.readCacheFromFile() || {
+      aiTasks: [],
+    };
     this.newCache = {
       aiTasks: [],
     };

diff --git a/packages/web-integration/tests/ai/e2e/ai-online-order.spec.ts b/packages/web-integration/tests/ai/e2e/ai-online-order.spec.ts
@@ -35,12 +35,12 @@ test('ai online order', async ({ ai, page, aiQuery }) => {
   //   productPrice: "商品价格",
   //   productDescription: "商品描述（饮品的各种参数，吸管、冰沙等），在价格下面",
   // }));
-  expect(cardDetail.productName).toContain('多肉葡萄');
-  expect(cardDetail.productDescription).toContain('绿妍');
 
   console.log('商品订单详情：', {
     productName: cardDetail.productName,
     productPrice: cardDetail.productPrice,
     productDescription: cardDetail.productDescription,
   });
+  expect(cardDetail.productName).toContain('多肉葡萄');
+  expect(cardDetail.productDescription).toContain('绿妍');
 });
diff --git a/packages/web-integration/tests/unit-test/task-cache.test.ts b/packages/web-integration/tests/unit-test/task-cache.test.ts
@@ -33,21 +33,19 @@ describe('TaskCache', () => {
   });
 
   it('should return false if the prompt does not match', async () => {
-    if (taskCache.cache) {
-      taskCache.cache.aiTasks = [
-        {
-          prompt: 'different prompt',
-          tasks: [
-            {
-              type: 'plan',
-              prompt: 'different prompt',
-              pageContext,
-              response: { plans: [] },
-            },
-          ],
-        },
-      ];
-    }
+    taskCache.cache.aiTasks = [
+      {
+        prompt: 'different prompt',
+        tasks: [
+          {
+            type: 'plan',
+            prompt: 'different prompt',
+            pageContext,
+            response: { plans: [] },
+          },
+        ],
+      },
+    ];
     const cacheGroup = taskCache.getCacheGroupByPrompt('test prompt');
     const result = cacheGroup.readCache(
       formalPageContext,