Update unit tests

2025-04-17 18:32:44 +02:00
parent 34fb5d814f
commit 813c6a3f8f
18 changed files with 2365 additions and 2279 deletions
--- a/test/stop_command_test.py
+++ b/test/stop_command_test.py
@@ -1,49 +1,51 @@
-from datetime import datetime
-import os
-import time
-import unittest
-
-from sia.working_memory import WorkingMemory
-from sia.stop_command import StopCommand
-from sia.background_entry import BackgroundEntry
-
-class StopCommandTest(unittest.TestCase):
-    def setUp(self):
-        """Set up test cases with a fresh working memory"""
-        self.memory = WorkingMemory()
-        self.test_timestamp = datetime(2024, 10, 31, 12, 0, 0)
-
-    def wait_for_process_start(self, entry: BackgroundEntry, timeout=1.0):
-        """Wait for process to start, checking context pid"""
-        start_time = time.time()
-        while time.time() - start_time < timeout:
-            entry.update()
-            context = entry.generate_context()
-            if context.get("pid") is not None:
-                return True
-            time.sleep(0.1)
-        return False
-
-    def test_stop_command_cleanup(self):
-        """Test stop command cleans up and removes all entries"""
-        # Add background process
-        entry = BackgroundEntry("test-id", self.test_timestamp, "sleep 10")
-        self.memory.add_entry(entry)
-        
-        # Wait for process to start and get PID
-        self.assertTrue(self.wait_for_process_start(entry))
-        context = entry.generate_context()
-        pid = int(context.get("pid"))
-        
-        # Execute stop command
-        command = StopCommand()
-        result = command.execute(self.memory)
-        
-        # Verify entries removed and process terminated
-        self.assertTrue(result.should_stop)
-        self.assertEqual(self.memory.get_entries_count(), 0)
-        
-        # Verify process was terminated
-        time.sleep(0.1)  # Give process time to terminate
-        with self.assertRaises(ProcessLookupError):
+from datetime import datetime
+import os
+import time
+import unittest
+
+from sia.working_memory import WorkingMemory
+from sia.stop_command import StopCommand
+from sia.entry.background_entry import BackgroundEntry
+
+class StopCommandTest(unittest.TestCase):
+    def setUp(self):
+        """Set up test cases with a fresh working memory"""
+        self.memory = WorkingMemory()
+        self.test_timestamp = datetime(2024, 10, 31, 12, 0, 0)
+        # Use current directory as working directory
+        self.work_dir = os.getcwd()
+
+    def wait_for_process_start(self, entry: BackgroundEntry, timeout=1.0):
+        """Wait for process to start, checking context pid"""
+        start_time = time.time()
+        while time.time() - start_time < timeout:
+            entry.update()
+            context = entry.generate_context()
+            if context.get("pid") is not None:
+                return True
+            time.sleep(0.1)
+        return False
+
+    def test_stop_command_cleanup(self):
+        """Test stop command cleans up and removes all entries"""
+        # Add background process with correct work_dir parameter
+        entry = BackgroundEntry("test-id", self.work_dir, "sleep 10")
+        self.memory.add_entry(entry)
+        
+        # Wait for process to start and get PID
+        self.assertTrue(self.wait_for_process_start(entry))
+        context = entry.generate_context()
+        pid = int(context.get("pid"))
+        
+        # Execute stop command
+        command = StopCommand()
+        result = command.execute(self.memory)
+        
+        # Verify entries removed and process terminated
+        self.assertTrue(result.should_stop)
+        self.assertEqual(self.memory.get_entries_count(), 0)
+        
+        # Verify process was terminated
+        time.sleep(0.1)  # Give process time to terminate
+        with self.assertRaises(ProcessLookupError):
            os.kill(pid, 0)