From c93c06711a383e92aa29eeb2555db400edeb16c7 Mon Sep 17 00:00:00 2001
From: Gal Zahavi <38544478+galz10@users.noreply.github.com>
Date: Tue, 19 Aug 2025 18:24:42 -0700
Subject: [PATCH] Update shell service integration tests (#6598)

---
 integration-tests/shell-service.test.ts | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/integration-tests/shell-service.test.ts b/integration-tests/shell-service.test.ts
index 58cf65e5..387d635c 100644
--- a/integration-tests/shell-service.test.ts
+++ b/integration-tests/shell-service.test.ts
@@ -16,7 +16,7 @@ describe('ShellExecutionService programmatic integration tests', () => {
   beforeAll(async () => {
     // Create a dedicated directory for this test suite to avoid conflicts.
     testDir = path.join(
-      process.env.INTEGRATION_TEST_FILE_DIR!,
+      process.env['INTEGRATION_TEST_FILE_DIR']!,
       'shell-service-tests',
     );
     await fs.mkdir(testDir, { recursive: true });
@@ -27,11 +27,12 @@ describe('ShellExecutionService programmatic integration tests', () => {
     const onOutputEvent = vi.fn();
     const abortController = new AbortController();
 
-    const handle = ShellExecutionService.execute(
+    const handle = await ShellExecutionService.execute(
       command,
       testDir,
       onOutputEvent,
       abortController.signal,
+      false,
     );
 
     const result = await handle.result;
@@ -52,11 +53,12 @@ describe('ShellExecutionService programmatic integration tests', () => {
       const onOutputEvent = vi.fn();
       const abortController = new AbortController();
 
-      const handle = ShellExecutionService.execute(
+      const handle = await ShellExecutionService.execute(
         command,
         testDir,
         onOutputEvent,
         abortController.signal,
+        false,
       );
 
       const result = await handle.result;
@@ -77,11 +79,12 @@ describe('ShellExecutionService programmatic integration tests', () => {
       const onOutputEvent = vi.fn();
       const abortController = new AbortController();
 
-      const handle = ShellExecutionService.execute(
+      const handle = await ShellExecutionService.execute(
         command,
         testDir,
         onOutputEvent,
         abortController.signal,
+        false,
       );
 
       const result = await handle.result;
@@ -98,11 +101,12 @@ describe('ShellExecutionService programmatic integration tests', () => {
     const onOutputEvent = vi.fn();
     const abortController = new AbortController();
 
-    const handle = ShellExecutionService.execute(
+    const handle = await ShellExecutionService.execute(
       command,
       testDir,
       onOutputEvent,
       abortController.signal,
+      false,
     );
 
     // Abort shortly after starting