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

package/dist/index.cjs

@@ -6088,6 +6088,9 @@ export default class ProboReporter implements Reporter {
             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__namespace.join(testSuiteDir, 'tests');
@@ -6113,8 +6116,35 @@ export default class ProboReporter implements Reporter {
          */
         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__namespace.join(outputDir, 'tests');
+                if (fs__namespace.existsSync(testsDir)) {
+                    const specFiles = fs__namespace.readdirSync(testsDir).filter(f => f.endsWith('.spec.ts'));
+                    for (const specFile of specFiles) {
+                        const filePath = path__namespace.join(testsDir, specFile);
+                        const content = fs__namespace.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,