fix: csharp executor proxy test

microsoft · Apr 25, 2024 · b74cf62 · b74cf62
1 parent 9b4076a
commit b74cf62
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/src/promptflow/tests/executor/unittests/batch/test_csharp_executor_proxy.py b/src/promptflow/tests/executor/unittests/batch/test_csharp_executor_proxy.py
@@ -1,6 +1,4 @@
 import json
-import platform
-import signal
 import socket
 import subprocess
 from pathlib import Path
@@ -64,10 +62,7 @@ async def test_destroy_with_terminates_gracefully(self):
         await executor_proxy.destroy()
 
         mock_process.poll.assert_called_once()
-        if platform.system() != "Windows":
-            mock_process.terminate.assert_called_once()
-        else:
-            mock_process.send_signal.assert_called_once_with(signal.CTRL_BREAK_EVENT)
+        mock_process.terminate.assert_called_once()
         mock_process.wait.assert_called_once_with(timeout=5)
         mock_process.kill.assert_not_called()
 
@@ -82,10 +77,7 @@ async def test_destroy_with_force_kill(self):
         await executor_proxy.destroy()
 
         mock_process.poll.assert_called_once()
-        if platform.system() != "Windows":
-            mock_process.terminate.assert_called_once()
-        else:
-            mock_process.send_signal.assert_called_once_with(signal.CTRL_BREAK_EVENT)
+        mock_process.terminate.assert_called_once()
         mock_process.wait.assert_called_once_with(timeout=5)
         mock_process.kill.assert_called_once()