test(aider): modify script execution to source environment variables and improve debug output

Author: DevelopmentCats

aider/main.test.ts

@@ -34,8 +34,17 @@ const executeScriptInContainerWithBash = async (
   `,
   ]);
 
-  // Run the script
-  const resp = await execContainer(id, ["bash", "-c", instance.script]);
+  // Run the script with the environment variables from extraCommands
+  // Modifying to preserve environment variables
+  const resp = await execContainer(id, ["bash", "-c", `
+    # Source any environment variables that might have been set
+    if [ -f /tmp/env_vars.sh ]; then
+      source /tmp/env_vars.sh
+    fi
+    
+    # Run the script
+    ${instance.script}
+  `]);
   const stdout = resp.stdout.trim().split("\n");
   const stderr = resp.stderr.trim().split("\n");
   return {
@@ -109,17 +118,21 @@ describe("aider", async () => {
     const output = await executeScriptInContainerWithBash(
       state,
       "alpine",
-      `export CODER_MCP_AIDER_TASK_PROMPT="${testPrompt}"`,
+      `echo 'export CODER_MCP_AIDER_TASK_PROMPT="${testPrompt}"' > /tmp/env_vars.sh`
     );
 
+    // Debug: print all output lines
+    console.log("DEBUG OUTPUT LINES:");
+    output.stdout.forEach(line => console.log(`> ${line}`));
+
     // Check if script contains the proper command construction with --message and --yes-always flags
     const instance = findResourceInstance(state, "coder_script");
 
-    // Verify the script uses --message flag
-    expect(instance.script.includes("aider --message")).toBe(true);
-
     // Verify the script uses --yes-always flag
-    expect(instance.script.includes("--yes-always")).toBe(true);
+    expect(instance.script.includes("aider --yes-always")).toBe(true);
+
+    // Verify the script uses --message flag
+    expect(instance.script.includes("--message")).toBe(true);
 
     // Verify the script creates a flag file to prevent duplicate execution
     expect(instance.script.includes('touch "$HOME/.aider_task_executed"')).toBe(
@@ -128,7 +141,7 @@ describe("aider", async () => {
 
     // Verify the output shows the right message
     expect(
-      output.stdout.some((line) => line.includes("Running Aider with message")),
+      output.stdout.some((line) => line.includes("Running Aider with message in screen session")),
     ).toBe(true);
   });