mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-06-25 03:38:59 +02:00
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
import os
|
|
import importlib.util
|
|
import sys
|
|
|
|
# Dynamically load the module under test from the Ansible role's files directory
|
|
def load_optimizer_module():
|
|
module_path = os.path.abspath(os.path.join(
|
|
os.path.dirname(__file__),
|
|
'..', "..", 'roles', 'system-storage-optimizer', 'files', 'system-storage-optimizer.py'
|
|
))
|
|
spec = importlib.util.spec_from_file_location('storage_optimizer', module_path)
|
|
optimizer = importlib.util.module_from_spec(spec)
|
|
# Register module so patch('storage_optimizer...') works
|
|
sys.modules[spec.name] = optimizer
|
|
spec.loader.exec_module(optimizer)
|
|
return optimizer
|
|
|
|
storage_optimizer = load_optimizer_module()
|
|
|
|
class TestHealthCheckLogic(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
# Prevent sleeping delays in inline loops
|
|
patcher = patch('storage_optimizer.time.sleep', return_value=None)
|
|
self.sleep_patcher = patcher.start()
|
|
self.addCleanup(self.sleep_patcher.stop)
|
|
|
|
@patch('storage_optimizer.run_command')
|
|
def test_has_healthcheck_true(self, mock_run):
|
|
mock_run.return_value = '{"Status":"starting","FailingStreak":0}'
|
|
self.assertTrue(storage_optimizer.has_healthcheck('container-id'))
|
|
|
|
@patch('storage_optimizer.run_command')
|
|
def test_has_healthcheck_false(self, mock_run):
|
|
mock_run.return_value = 'null'
|
|
self.assertFalse(storage_optimizer.has_healthcheck('container-id'))
|
|
|
|
@patch('storage_optimizer.get_image')
|
|
def test_has_container_with_database(self, mock_get_image):
|
|
test_cases = [
|
|
('postgres:13', True),
|
|
('mariadb:latest', True),
|
|
('redis:6', True),
|
|
('mongo:4', True),
|
|
('nginx:alpine', False),
|
|
]
|
|
for image_name, expected in test_cases:
|
|
mock_get_image.return_value = image_name
|
|
result = storage_optimizer.has_container_with_database(['any'])
|
|
self.assertEqual(result, expected,
|
|
f"Image {image_name} expected {expected}")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|