testdriverai 4.1.37 → 4.1.39

package/agent.js

@@ -863,9 +863,6 @@ ${regression}
 // it parses the markdown file and executes the codeblocks exactly as if they were
 // generated by the AI in a single prompt
 let run = async (file, overwrite = false, shouldExit = true) => {
-  // parse potential string value for exit
-  shouldExit = shouldExit === "false" ? false : true;
-  overwrite = overwrite === "false" ? false : true;
 
   setTerminalWindowTransparency(true);
 
@@ -887,6 +884,9 @@ let run = async (file, overwrite = false, shouldExit = true) => {
     await exit(true);
   }
 
+  // Inject environment variables into any ${VAR} strings
+  yml = parser.interpolate(yml, process.env);
+
   let ymlObj = null;
   try {
     ymlObj = await yaml.load(yml);
@@ -934,13 +934,13 @@ ${yaml.dump(step)}
     await actOnMarkdown(markdown, 0, true);
   }
 
-  if (overwrite) {
+  if (overwrite || overwrite == "true") {
     await save({ filepath: file });
   }
 
   setTerminalWindowTransparency(false);
 
-  if (shouldExit) {
+  if (shouldExit || shouldExit == "true") {
     await summarize();
     await exit(false);
   }

package/eslint.config.js

@@ -10,6 +10,7 @@ module.exports = [
       globals: {
         ...globals.browser,
         ...globals.node,
+        ...globals.jest
       },
     },
   },

package/lib/parser.js

@@ -76,10 +76,24 @@ const parseYAML = async function (inputYaml) {
   return doc;
 };
 
+// Replace ${VAR} with the value from the vars object
+// Will skip variables that are not in the vars object
+// Will skip escaped variables like \${VAR} 
+function interpolate (yaml, vars) {
+  let newyaml = yaml;
+  Object.keys(vars).forEach((key) => {
+    newyaml = newyaml.replace(new RegExp(`(?<!\\\\)\\$\\{${key}\\}`, "g"), vars[key]);
+  });
+  // Replace \$ with $
+  newyaml = newyaml.replace(/\\(\${[^}]+})/g, "$1");
+  return newyaml;
+}
+
 module.exports = {
   findCodeBlocks,
   findGenerativePrompts,
   getYAMLFromCodeBlock,
+  interpolate,
   getCommands: async function (codeBlock) {
     const yml = getYAMLFromCodeBlock(codeBlock);
     let yamlArray = await parseYAML(yml);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "4.1.37",
+  "version": "4.1.39",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "index.js",
   "bin": {
@@ -10,7 +10,8 @@
     "start": "node index.js",
     "dev": "DEV=true node index.js",
     "debug": "DEV=true VERBOSE=true node index.js",
-    "bundle": "node build.mjs"
+    "bundle": "node build.mjs",
+    "test": "mocha test/*"
   },
   "author": "",
   "license": "ISC",
@@ -56,6 +57,8 @@
     "globals": "^15.9.0",
     "node-addon-api": "^8.0.0",
     "node-gyp": "^10.1.0",
+    "mocha": "^10.8.2",
+    "chai": "^5.1.2",
     "prettier": "3.3.3"
   },
   "optionalDependencies": {

package/test/test_parser.js

@@ -0,0 +1,48 @@
+const chai = require('chai');
+const expect = chai.expect;
+const { interpolate } = require('../lib/parser');
+
+describe('interpolate', function() {
+  it('should replace variables in the string with values from the dictionary', function() {
+    const template = 'Hello, ${name}!';
+    const vars = { name: 'World' };
+    const result = interpolate(template, vars);
+    expect(result).to.equal('Hello, World!');
+  });
+
+  it('should replace multiple variables in the string', function() {
+    const template = 'Hi, ${firstName} ${lastName}!';
+    const vars = { firstName: 'Marcy', lastName: 'the Dog' };
+    const result = interpolate(template, vars);
+    expect(result).to.equal('Hi, Marcy the Dog!');
+  });
+
+  it('should leave variables unchanged if they are not in the dictionary', function() {
+    const template = 'Hi, ${name} ${surname}!';
+    const vars = { name: 'John' };
+    const result = interpolate(template, vars);
+    expect(result).to.equal('Hi, John ${surname}!');
+  });
+
+  it('should return the original string if no variables are present', function() {
+    const template = 'Hello, World!';
+    const vars = { name: 'John' };
+    const result = interpolate(template, vars);
+    expect(result).to.equal('Hello, World!');
+  });
+
+  it('should handle an empty dictionary gracefully', function() {
+    const template = 'Hi, ${name}!';
+    const vars = {};
+    const result = interpolate(template, vars);
+    expect(result).to.equal('Hi, ${name}!');
+  });
+
+  it('should not replace escaped variables', function() {
+    const template = 'Hello, \\${name}!';
+    const vars = { name: 'World' };
+    const result = interpolate(template, vars);
+    expect(result).to.equal('Hello, ${name}!');
+  });
+});
+