speed up test

Author: marcusschiesser

e2e/resolve_ts_dependencies.spec.ts

@@ -19,7 +19,6 @@ if (
   templateFramework == "nextjs" ||
   templateFramework == "express" // test is only relevant for TS projects
 ) {
-  const llamaParseOptions = [true, false];
   // vectorDBs combinations to test
   const vectorDbs: TemplateVectorDB[] = [
     "mongo",
@@ -34,68 +33,79 @@ if (
   ];
 
   test.describe("Test resolve TS dependencies", () => {
-    for (const llamaParseOpt of llamaParseOptions) {
-      for (const vectorDb of vectorDbs) {
-        const optionDescription = `vectorDb: ${vectorDb}, dataSource: ${dataSource}, llamaParse: ${llamaParseOpt}`;
+    // Test vector DBs without LlamaParse
+    for (const vectorDb of vectorDbs) {
+      const optionDescription = `vectorDb: ${vectorDb}, dataSource: ${dataSource}`;
 
-        test(`options: ${optionDescription}`, async () => {
-          const cwd = await createTestDir();
+      test(`Vector DB test - ${optionDescription}`, async () => {
+        await runTest(vectorDb, false);
+      });
+    }
+
+    // Test LlamaParse with vectorDB 'none'
+    test(`LlamaParse test - vectorDb: none, dataSource: ${dataSource}, llamaParse: true`, async () => {
+      await runTest("none", true);
+    });
 
-          const result = await runCreateLlama({
-            cwd: cwd,
-            templateType: "streaming",
-            templateFramework: templateFramework,
-            dataSource: dataSource,
-            vectorDb: vectorDb,
-            port: 3000,
-            externalPort: 8000,
-            postInstallAction: "none",
-            templateUI: undefined,
-            appType: templateFramework === "nextjs" ? "" : "--no-frontend",
-            llamaCloudProjectName: undefined,
-            llamaCloudIndexName: undefined,
-            tools: undefined,
-            useLlamaParse: llamaParseOpt,
-          });
-          const name = result.projectName;
+    async function runTest(
+      vectorDb: TemplateVectorDB | "none",
+      useLlamaParse: boolean,
+    ) {
+      const cwd = await createTestDir();
 
-          // Check if the app folder exists
-          const appDir = path.join(cwd, name);
-          const dirExists = fs.existsSync(appDir);
-          expect(dirExists).toBeTruthy();
+      const result = await runCreateLlama({
+        cwd: cwd,
+        templateType: "streaming",
+        templateFramework: templateFramework,
+        dataSource: dataSource,
+        vectorDb: vectorDb,
+        port: 3000,
+        externalPort: 8000,
+        postInstallAction: "none",
+        templateUI: undefined,
+        appType: templateFramework === "nextjs" ? "" : "--no-frontend",
+        llamaCloudProjectName: undefined,
+        llamaCloudIndexName: undefined,
+        tools: undefined,
+        useLlamaParse: useLlamaParse,
+      });
+      const name = result.projectName;
 
-          // Install dependencies using pnpm
-          try {
-            const { stderr: installStderr } = await execAsync(
-              "pnpm install --prefer-offline",
-              {
-                cwd: appDir,
-              },
-            );
-            expect(installStderr).toBeFalsy();
-          } catch (error) {
-            console.error("Error installing dependencies:", error);
-            throw error;
-          }
+      // Check if the app folder exists
+      const appDir = path.join(cwd, name);
+      const dirExists = fs.existsSync(appDir);
+      expect(dirExists).toBeTruthy();
+
+      // Install dependencies using pnpm
+      try {
+        const { stderr: installStderr } = await execAsync(
+          "pnpm install --prefer-offline",
+          {
+            cwd: appDir,
+          },
+        );
+        expect(installStderr).toBeFalsy();
+      } catch (error) {
+        console.error("Error installing dependencies:", error);
+        throw error;
+      }
 
-          // Run tsc type check and capture the output
-          try {
-            const { stdout, stderr } = await execAsync(
-              "pnpm exec tsc -b --diagnostics",
-              {
-                cwd: appDir,
-              },
-            );
-            // Check if there's any error output
-            expect(stderr).toBeFalsy();
+      // Run tsc type check and capture the output
+      try {
+        const { stdout, stderr } = await execAsync(
+          "pnpm exec tsc -b --diagnostics",
+          {
+            cwd: appDir,
+          },
+        );
+        // Check if there's any error output
+        expect(stderr).toBeFalsy();
 
-            // Log the stdout for debugging purposes
-            console.log("TypeScript type-check output:", stdout);
-          } catch (error) {
-            console.error("Error running tsc:", error);
-            throw error;
-          }
-        });
+        // Log the stdout for debugging purposes
+        console.log("TypeScript type-check output:", stdout);
+      } catch (error) {
+        console.error("Error running tsc:", error);
+        throw error;
       }
     }
   });