fix cancelation tests

eleanorjboyd · Oct 29, 2024 · f694061 · f694061
1 parent 17e23ac
commit f694061
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 5 deletions.
diff --git a/python_files/tests/pytestadapter/helpers.py b/python_files/tests/pytestadapter/helpers.py
@@ -130,7 +130,8 @@ def parse_rpc_message(data: str) -> Tuple[Dict[str, str], str]:
 
 def _listen_on_fifo(pipe_name: str, result: List[str], completed: threading.Event):
     # Open the FIFO for reading
-    with open(pipe_name) as fifo:
+    fifo_path = pathlib.Path(pipe_name)
+    with fifo_path.open() as fifo:
         print("Waiting for data...")
         while True:
             if completed.is_set():

diff --git a/src/client/testing/testController/pytest/pytestExecutionAdapter.ts b/src/client/testing/testController/pytest/pytestExecutionAdapter.ts
@@ -188,6 +188,7 @@ export class PytestTestExecutionAdapter implements ITestExecutionAdapter {
                         resultProc?.kill();
                     } else {
                         deferredTillExecClose.resolve();
+                        serverCancel.cancel();
                     }
                 });
 

diff --git a/src/client/testing/testController/unittest/testExecutionAdapter.ts b/src/client/testing/testController/unittest/testExecutionAdapter.ts
@@ -190,6 +190,7 @@ export class UnittestTestExecutionAdapter implements ITestExecutionAdapter {
                         resultProc?.kill();
                     } else {
                         deferredTillExecClose?.resolve();
+                        serverCancel.cancel();
                     }
                 });
 

diff --git a/src/test/testing/testController/testCancellationRunAdapters.unit.test.ts b/src/test/testing/testController/testCancellationRunAdapters.unit.test.ts
@@ -74,7 +74,6 @@ suite('Execution Flow Run Adapters', () => {
                 .setup((x) => x.execObservable(typeMoq.It.isAny(), typeMoq.It.isAny()))
                 .returns(() => {
                     cancellationToken.cancel();
-                    deferredTillServerCloseTester?.resolve();
                     return {
                         proc: mockProc as any,
                         out: typeMoq.Mock.ofType<Observable<Output<string>>>().object,
@@ -96,8 +95,12 @@ suite('Execution Flow Run Adapters', () => {
                 return Promise.resolve('named-pipe');
             });
 
-            utilsStartRunResultNamedPipe.callsFake((_callback, deferredTillServerClose, _token) => {
+            utilsStartRunResultNamedPipe.callsFake((_callback, deferredTillServerClose, token) => {
                 deferredTillServerCloseTester = deferredTillServerClose;
+                token?.onCancellationRequested(() => {
+                    deferredTillServerCloseTester?.resolve();
+                });
+
                 return Promise.resolve('named-pipes-socket-name');
             });
             serverDisposeStub.callsFake(() => {
@@ -140,7 +143,6 @@ suite('Execution Flow Run Adapters', () => {
                 .setup((x) => x.execObservable(typeMoq.It.isAny(), typeMoq.It.isAny()))
                 .returns(() => {
                     cancellationToken.cancel();
-                    deferredTillServerCloseTester?.resolve();
                     return {
                         proc: mockProc as any,
                         out: typeMoq.Mock.ofType<Observable<Output<string>>>().object,
@@ -164,6 +166,9 @@ suite('Execution Flow Run Adapters', () => {
 
             utilsStartRunResultNamedPipe.callsFake((_callback, deferredTillServerClose, _token) => {
                 deferredTillServerCloseTester = deferredTillServerClose;
+                token?.onCancellationRequested(() => {
+                    deferredTillServerCloseTester?.resolve();
+                });
                 return Promise.resolve('named-pipes-socket-name');
             });
             serverDisposeStub.callsFake(() => {
@@ -188,7 +193,6 @@ suite('Execution Flow Run Adapters', () => {
                 })
                 .returns(async () => {
                     cancellationToken.cancel();
-                    deferredTillServerCloseTester?.resolve();
                     return Promise.resolve();
                 });