llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added timeout and limit to script entries

Browse Source
This commit is contained in:
Niels Geens
2024-11-13 17:48:39 +01:00
parent 65478e0b17
commit 834ce12e8f
21 changed files with 262 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/single_entry_test.py
View File

@@ -26,7 +26,7 @@ class SingleEntryTest(unittest.TestCase):
def test_initialization(self):
"""Test entry initialization"""
script = "echo test"
entry = SingleEntry(script, self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, script, None, None)
self.assertEqual(entry.script, script)
self.assertEqual(entry.stdout, "")
@@ -40,7 +40,7 @@ class SingleEntryTest(unittest.TestCase):
"""Test successful script execution"""
script = "echo 'test'"
entry = SingleEntry(script, self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, script, None, None)
entry.update()
# Verify entry state
@@ -51,7 +51,7 @@ class SingleEntryTest(unittest.TestCase):
def test_failed_execution(self):
"""Test failed script execution with invalid command"""
entry = SingleEntry("nonexistentcommand", self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, "nonexistentcommand", None, None)
entry.update()
# Verify entry state
@@ -64,7 +64,7 @@ class SingleEntryTest(unittest.TestCase):
"""Test script that creates a file"""
script = f"echo 'test' > {self.test_filename}"
entry = SingleEntry(script, self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, script, None, None)
entry.update()
# Verify file was created and contains expected content
@@ -77,7 +77,7 @@ class SingleEntryTest(unittest.TestCase):
"""Test that script only executes once"""
script = f"echo 'test' >> {self.test_filename}"
entry = SingleEntry(script, self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, script, None, None)
# Run update multiple times
entry.update()
@@ -92,7 +92,7 @@ class SingleEntryTest(unittest.TestCase):
def test_generate_context_before_execution(self):
"""Test XML context generation before execution"""
entry = SingleEntry("echo test", self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, "echo test", None, None)
element = entry.generate_context()
# Verify XML structure
@@ -109,7 +109,7 @@ class SingleEntryTest(unittest.TestCase):
"""Test XML context generation after execution"""
script = "echo 'test'"
entry = SingleEntry(script, self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, script, None, None)
entry.update()
element = entry.generate_context()
@@ -128,7 +128,7 @@ class SingleEntryTest(unittest.TestCase):
"""Test executing multiple commands in one script"""
script = f"echo 'first' > {self.test_filename} && echo 'second' >> {self.test_filename}"
entry = SingleEntry(script, self.test_id, self.test_timestamp)
entry = SingleEntry(self.test_id, self.test_timestamp, script, None, None)
entry.update()
# Verify file contains both lines