96 lines
3.8 KiB
Python
96 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from monitors.yumi.service_release import build_update_targets_payload
|
|
|
|
|
|
class ServiceReleaseTargetsTests(unittest.TestCase):
|
|
@patch("monitors.yumi.service_release.frontend_release_enabled", return_value=False)
|
|
@patch("monitors.yumi.service_release.service_repository_ref", side_effect=lambda service: f"repo/{service}")
|
|
@patch("monitors.yumi.service_release.buildable_service_names", return_value=["auth", "gateway"])
|
|
@patch("monitors.yumi.service_release.service_records_by_name")
|
|
@patch("monitors.yumi.service_release.load_local_host_compose_model")
|
|
def test_build_update_targets_payload_reuses_host_compose_model(
|
|
self,
|
|
load_local_host_compose_model_mock,
|
|
service_records_by_name_mock,
|
|
buildable_service_names_mock,
|
|
service_repository_ref_mock,
|
|
frontend_release_enabled_mock,
|
|
) -> None:
|
|
del buildable_service_names_mock, service_repository_ref_mock, frontend_release_enabled_mock
|
|
service_records_by_name_mock.return_value = {
|
|
"auth": [
|
|
{
|
|
"host": "app-1",
|
|
"ip": "10.0.0.1",
|
|
"instanceId": "ins-app-1",
|
|
"port": 8080,
|
|
"path": "/auth",
|
|
}
|
|
],
|
|
"gateway": [
|
|
{
|
|
"host": "app-1",
|
|
"ip": "10.0.0.1",
|
|
"instanceId": "ins-app-1",
|
|
"port": 9000,
|
|
"path": "/gateway",
|
|
}
|
|
],
|
|
}
|
|
load_local_host_compose_model_mock.return_value = {
|
|
"services": {
|
|
"auth": {"image": "registry/auth:main"},
|
|
"gateway": {"image": "registry/gateway:main"},
|
|
}
|
|
}
|
|
|
|
payload = build_update_targets_payload()
|
|
|
|
self.assertTrue(payload["ok"])
|
|
self.assertEqual(load_local_host_compose_model_mock.call_count, 1)
|
|
self.assertEqual([item["service"] for item in payload["items"]], ["auth", "gateway"])
|
|
self.assertEqual(payload["items"][0]["currentImages"], ["registry/auth:main"])
|
|
self.assertEqual(payload["items"][1]["currentImages"], ["registry/gateway:main"])
|
|
|
|
@patch("monitors.yumi.service_release.frontend_release_enabled", return_value=False)
|
|
@patch("monitors.yumi.service_release.service_repository_ref", side_effect=lambda service: f"repo/{service}")
|
|
@patch("monitors.yumi.service_release.buildable_service_names", return_value=["auth"])
|
|
@patch("monitors.yumi.service_release.service_records_by_name")
|
|
@patch("monitors.yumi.service_release.load_local_host_compose_model", side_effect=RuntimeError("broken compose"))
|
|
def test_build_update_targets_payload_keeps_items_when_compose_unavailable(
|
|
self,
|
|
load_local_host_compose_model_mock,
|
|
service_records_by_name_mock,
|
|
buildable_service_names_mock,
|
|
service_repository_ref_mock,
|
|
frontend_release_enabled_mock,
|
|
) -> None:
|
|
del load_local_host_compose_model_mock
|
|
del buildable_service_names_mock, service_repository_ref_mock, frontend_release_enabled_mock
|
|
service_records_by_name_mock.return_value = {
|
|
"auth": [
|
|
{
|
|
"host": "app-1",
|
|
"ip": "10.0.0.1",
|
|
"instanceId": "ins-app-1",
|
|
"port": 8080,
|
|
"path": "/auth",
|
|
}
|
|
]
|
|
}
|
|
|
|
payload = build_update_targets_payload()
|
|
|
|
self.assertTrue(payload["ok"])
|
|
self.assertEqual(len(payload["items"]), 1)
|
|
self.assertEqual(payload["items"][0]["currentImages"], [])
|
|
self.assertEqual(payload["items"][0]["warnings"], ["app-1: broken compose"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|