@probolabs/playwright 1.4.0-rc.1 → 1.4.0-rc.3

package/dist/index.js

@@ -6069,6 +6069,9 @@ class TestSuiteRunner {
         const testSuiteDir = outputDir || getDefaultTestSuiteDir(testSuiteId, testSuiteName);
         // Generate code for all scenarios
         const codeGenResult = await ProboCodeGenerator.generateCodeForTestSuite(testSuiteId, apiToken, apiUrl);
+        if (codeGenResult.scenarios.length === 0) {
+            throw new Error(`No scenarios found in test suite ${testSuiteId}. Cannot generate test files.`);
+        }
         // Ensure directories exist
         ensureDirectoryExists(testSuiteDir);
         const testsDir = path.join(testSuiteDir, 'tests');
@@ -6094,8 +6097,35 @@ class TestSuiteRunner {
      */
     static async generatePackageJson(outputDir, codeGenResult) {
         // Check if any scenario has secrets by examining the generated code
-        // We'll check for dotenv imports in the generated code
-        const hasSecrets = codeGenResult.scenarios.some(scenario => scenario.code.includes("import { config } from 'dotenv'"));
+        // We'll check for dotenv imports in the generated code (check for multiple patterns to be robust)
+        let hasSecrets = codeGenResult.scenarios.some(scenario => {
+            const code = scenario.code;
+            return code.includes("from 'dotenv'") ||
+                code.includes('from "dotenv"') ||
+                code.includes("import { config } from 'dotenv'") ||
+                code.includes('import { config } from "dotenv"') ||
+                code.includes("require('dotenv')") ||
+                code.includes('require("dotenv")');
+        });
+        // Fallback: if not detected in codeGenResult, check the actual written files
+        // This handles edge cases where the code might have been modified or generated differently
+        if (!hasSecrets) {
+            const testsDir = path.join(outputDir, 'tests');
+            if (fs.existsSync(testsDir)) {
+                const specFiles = fs.readdirSync(testsDir).filter(f => f.endsWith('.spec.ts'));
+                for (const specFile of specFiles) {
+                    const filePath = path.join(testsDir, specFile);
+                    const content = fs.readFileSync(filePath, 'utf-8');
+                    if (content.includes("from 'dotenv'") ||
+                        content.includes('from "dotenv"') ||
+                        content.includes("import { config } from 'dotenv'") ||
+                        content.includes('import { config } from "dotenv"')) {
+                        hasSecrets = true;
+                        break;
+                    }
+                }
+            }
+        }
         const sanitizedName = slugify(codeGenResult.testSuiteName);
         const packageJsonContent = generatePackageJson({
             name: sanitizedName,