51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
"""维护窗口恢复逻辑回归，不触碰真实服务。"""
|
|
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import call, patch
|
|
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'tools'))
|
|
import build_frontend_review as review
|
|
|
|
|
|
class RestoreServicesTest(unittest.TestCase):
|
|
unit = 'smail-frontend-build-20260918-010000'
|
|
|
|
def test_stop_build_before_restoring_services(self):
|
|
with patch.object(review.subprocess, 'run') as run:
|
|
self.assertEqual(review.restore_services(self.unit), [])
|
|
self.assertEqual(run.call_args_list, [
|
|
call(['systemctl', 'stop', self.unit], timeout=90, check=False),
|
|
call(['systemctl', 'start', 'mysql.service'], check=True, timeout=90),
|
|
call(['systemctl', 'start', 'Scert.service'], check=True, timeout=90),
|
|
])
|
|
|
|
def test_attempt_remaining_restore_after_database_failure(self):
|
|
error = subprocess.CalledProcessError(1, 'systemctl')
|
|
with patch.object(review.subprocess, 'run', side_effect=[None, error, None]) as run:
|
|
self.assertEqual(review.restore_services(self.unit), [
|
|
{'service': 'mysql.service', 'error': 'CalledProcessError'},
|
|
])
|
|
self.assertEqual(run.call_args_list[-1].args[0], ['systemctl', 'start', 'Scert.service'])
|
|
|
|
def test_site_timeout_is_reported(self):
|
|
error = subprocess.TimeoutExpired('systemctl', 90)
|
|
with patch.object(review.subprocess, 'run', side_effect=[None, None, error]):
|
|
self.assertEqual(review.restore_services(self.unit), [
|
|
{'service': 'Scert.service', 'error': 'TimeoutExpired'},
|
|
])
|
|
|
|
def test_reject_unrelated_service_targets(self):
|
|
for name in ('nginx', 'smail.service', self.unit + ' other', '../mysql.service'):
|
|
with self.subTest(name=name), patch.object(review.subprocess, 'run') as run:
|
|
with self.assertRaises(AssertionError):
|
|
review.restore_services(name)
|
|
run.assert_not_called()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|