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

18
test/write_entry_test.py
View File

@@ -14,7 +14,7 @@ class WriteEntryTest(unittest.TestCase):
def test_initialization(self):
"""Test entry initialization"""
content = "test message"
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
self.assertEqual(entry.content, content)
self.assertEqual(entry.id, self.test_id)
@@ -25,7 +25,7 @@ class WriteEntryTest(unittest.TestCase):
def test_single_write(self):
"""Test writing content once"""
content = "test message"
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
# Perform update and verify content was written
entry.update()
@@ -35,7 +35,7 @@ class WriteEntryTest(unittest.TestCase):
def test_multiple_updates(self):
"""Test that content is only written once even with multiple updates"""
content = "test message"
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
# Perform multiple updates
entry.update()
@@ -52,7 +52,7 @@ class WriteEntryTest(unittest.TestCase):
def test_empty_content(self):
"""Test writing empty content"""
entry = WriteEntry("", self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, "", self.io_buffer)
entry.update()
self.assertEqual(self.io_buffer.get_stdout(), "")
@@ -61,7 +61,7 @@ class WriteEntryTest(unittest.TestCase):
def test_special_characters(self):
"""Test writing content with special characters"""
content = "Special chars: \n\t\r\'\"\\"
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
entry.update()
self.assertEqual(self.io_buffer.get_stdout(), content)
@@ -69,7 +69,7 @@ class WriteEntryTest(unittest.TestCase):
def test_large_content(self):
"""Test writing large content"""
content = "x" * 10000
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
entry.update()
self.assertEqual(self.io_buffer.get_stdout(), content)
@@ -77,7 +77,7 @@ class WriteEntryTest(unittest.TestCase):
def test_generate_context(self):
"""Test XML context generation"""
content = "test message"
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
# Generate context before writing
element = entry.generate_context()
@@ -97,7 +97,7 @@ class WriteEntryTest(unittest.TestCase):
def test_unicode_content(self):
"""Test writing Unicode content"""
content = "Hello 世界 😊"
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
entry.update()
self.assertEqual(self.io_buffer.get_stdout(), content)
@@ -107,7 +107,7 @@ class WriteEntryTest(unittest.TestCase):
content = """Line 1
Line 2
Line 3"""
entry = WriteEntry(content, self.io_buffer, self.test_id, self.test_timestamp)
entry = WriteEntry(self.test_id, self.test_timestamp, content, self.io_buffer)
entry.update()
self.assertEqual(self.io_buffer.get_stdout(), content)