34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
import unittest
|
|
from datetime import datetime
|
|
from sia.context_template import generate_context
|
|
|
|
class TestContextTemplate(unittest.TestCase):
|
|
def test_empty_containers(self):
|
|
"""Test context generation with no containers."""
|
|
context = generate_context([])
|
|
self.assertIn("<context>", context)
|
|
self.assertIn("<containers />", context)
|
|
self.assertIn("</context>", context)
|
|
|
|
def test_single_container(self):
|
|
"""Test context generation with a single container."""
|
|
container_status = {
|
|
'name': 'test-container',
|
|
'status': 'running',
|
|
'started_at': '2024-10-25T10:00:00Z',
|
|
'stdout_size': 100,
|
|
'stderr_size': 50
|
|
}
|
|
context = generate_context([container_status])
|
|
self.assertIn("<context>", context)
|
|
self.assertIn("<containers", context)
|
|
self.assertIn('name="test-container"', context)
|
|
self.assertIn('status="running"', context)
|
|
self.assertIn('started_at="2024-10-25T10:00:00Z"', context)
|
|
self.assertIn('stdout="100"', context)
|
|
self.assertIn('stderr="50"', context)
|
|
self.assertIn("</context>", context)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|