Added Lifecycle Command line args

It will first try to find it in the lifecycle folder, and if not found
will treat it like a path.

--prerun="my/path/to.yaml"
or
--prerun="my_prerun"

Author: aprowe

agent/index.js

@@ -68,6 +68,8 @@ class TestDriverAgent extends EventEmitter2 {
     this.healMode = flags.healMode || flags.heal || false;
     this.sandboxId = flags.sandboxId || null;
     this.workingDir = flags.workingDir || process.cwd();
+    this.postrun = flags.postrun || null;
+    this.provision = flags.provision || null;
 
     // Resolve thisFile to absolute path with proper extension
     if (this.thisFile) {
@@ -167,8 +169,8 @@ class TestDriverAgent extends EventEmitter2 {
 
     this.analytics.track("exit", { failed });
 
-    if (shouldRunLifecycle) {
-      await this.runLifecycle("postrun");
+    if (shouldRunLifecycle && this.postrun) {
+      await this.runLifecycle(this.postrun);
     }
 
     // Emit exit event with exit code and close readline interface
@@ -1473,7 +1475,10 @@ ${regression}
     this.instance = instance;
     await this.renderSandbox(instance, headless);
     await this.newSession();
-    await this.runLifecycle("provision");
+
+    if (this.provision) {
+      await this.runLifecycle(this.provision);
+    }
   }
 
   async start() {
@@ -1741,6 +1746,12 @@ ${regression}
         }
       }
     }
+
+    // If lifecycleName is a filepath
+    if (fs.existsSync(path.resolve(lifecycleName))) {
+      lifecycleFile = path.resolve(lifecycleName);
+    }
+
     if (lifecycleFile) {
       await this.run(lifecycleFile, false, false);
     }

agent/interface.js

@@ -51,12 +51,27 @@ function createCommandDefinitions(agent) {
         summary: Flags.string({
           description: "Specify output file for summarize results",
         }),
+        prerun: Flags.string({
+          description: "File to run before the main test file. Can be the name of a file in the lifecycle folder, or an absolute path",
+          default: "prerun",
+        }),
+        postrun: Flags.string({
+          description:
+            "File to after before the main test file. This will run regardless of if the test passes. can be the name of a file in the lifecycle folder, or an absolute path",
+          default: "postrun",
+        }),
+        provision: Flags.string({
+          description:
+            "File to run when a new sandbox is provisioned. Can be the name of a file in the lifecycle folder, or an absolute path",
+          default: "provision",
+        }),
       },
       handler: async (args, flags) => {
         // Use --path flag if provided, otherwise fall back to args.file
         const file = normalizeFilePath(args.file);
 
-        await agent.runLifecycle("prerun");
+        await agent.runLifecycle(flags.prerun);
+
         // When run() is called through run.js CLI command, shouldExit should be true
         const shouldExit = agent.cliArgs?.command === "run";
         await agent.run(file, flags.write, shouldExit);