start implementation on new architecture
This commit is contained in:
66
test/delete_command_test.py
Normal file
66
test/delete_command_test.py
Normal file
@@ -0,0 +1,66 @@
|
||||
import unittest
|
||||
from datetime import datetime
|
||||
import time
|
||||
|
||||
from sia.working_memory import WorkingMemory
|
||||
from sia.delete_command import DeleteCommand
|
||||
from sia.reasoning_entry import ReasoningEntry
|
||||
from sia.background_entry import BackgroundEntry
|
||||
|
||||
class DeleteCommandTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
"""Set up test cases with a fresh working memory"""
|
||||
self.memory = WorkingMemory()
|
||||
self.test_id = "test-id-1234"
|
||||
self.test_timestamp = datetime(2024, 10, 31, 12, 0, 0)
|
||||
|
||||
def test_delete_existing_entry(self):
|
||||
"""Test deleting an existing entry"""
|
||||
# Add test entry
|
||||
entry = ReasoningEntry("test content", self.test_id, self.test_timestamp)
|
||||
self.memory.add_entry(entry)
|
||||
|
||||
# Execute delete command
|
||||
command = DeleteCommand(self.test_id)
|
||||
result = command.execute(self.memory)
|
||||
|
||||
# Verify result and memory state
|
||||
self.assertTrue(result.success)
|
||||
self.assertFalse(result.should_stop)
|
||||
self.assertEqual(result.message, "")
|
||||
self.assertEqual(self.memory.get_entries_count(), 0)
|
||||
|
||||
def test_delete_running_background_entry(self):
|
||||
"""Test deleting a background entry that is still running"""
|
||||
# Create a background entry with a long-running command
|
||||
entry = BackgroundEntry("sleep 10", self.test_id, self.test_timestamp)
|
||||
self.memory.add_entry(entry)
|
||||
|
||||
# Start the process
|
||||
entry.update()
|
||||
self.assertIsNotNone(entry.process)
|
||||
process_pid = entry.process.pid
|
||||
|
||||
# Execute delete command
|
||||
command = DeleteCommand(self.test_id)
|
||||
result = command.execute(self.memory)
|
||||
|
||||
# Verify process was terminated
|
||||
time.sleep(0.1) # Give process time to terminate
|
||||
self.assertTrue(result.success)
|
||||
self.assertIsNone(entry.process)
|
||||
|
||||
# Verify process no longer exists
|
||||
with self.assertRaises(ProcessLookupError):
|
||||
import os
|
||||
os.kill(process_pid, 0)
|
||||
|
||||
def test_delete_nonexistent_entry(self):
|
||||
"""Test attempting to delete a nonexistent entry"""
|
||||
command = DeleteCommand("nonexistent-id")
|
||||
result = command.execute(self.memory)
|
||||
|
||||
# Verify error result
|
||||
self.assertFalse(result.success)
|
||||
self.assertFalse(result.should_stop)
|
||||
self.assertIn("not found", result.message)
|
||||
Reference in New Issue
Block a user