Test restarting channelfinder works ok

Author: jacomago

server/tests/docker.py

@@ -47,3 +47,9 @@ def restart_container(compose: DockerCompose, host_name: str) -> None:
     docker_client = DockerClient()
     docker_client.containers.get(container.ID).stop()
     docker_client.containers.get(container.ID).start()
+
+
+def shutdown_container(compose: DockerCompose, host_name: str) -> None:
+    container = compose.get_container(host_name)
+    docker_client = DockerClient()
+    docker_client.containers.get(container.ID).stop()

server/tests/test_restart.py

@@ -4,8 +4,14 @@
 from channelfinder import ChannelFinderClient
 from testcontainers.compose import DockerCompose
 
-from .client_checks import channels_match, check_channel_property, create_client_and_wait, wait_for_sync
-from .docker import restart_container, setup_compose  # noqa: F401
+from .client_checks import (
+    INACTIVE_PROPERTY,
+    channels_match,
+    check_channel_property,
+    create_client_and_wait,
+    wait_for_sync,
+)
+from .docker import restart_container, setup_compose, shutdown_container  # noqa: F401
 
 PROPERTIES_TO_MATCH = ["pvStatus", "recordType", "recordDesc", "alias", "hostName", "iocName", "recceiverID"]
 
@@ -45,3 +51,27 @@ def test_manual_channels_same_after_restart(
         channels_end = cf_client.find(name="*")
         assert len(channels_begin) == len(channels_end)
         channels_match(channels_begin, channels_end, PROPERTIES_TO_MATCH + ["test_property"])
+
+
+def check_connection_active(cf_client: ChannelFinderClient) -> bool:
+    try:
+        cf_client.find(name="*")
+        return True
+    except Exception:
+        return False
+
+
+class TestRestartChannelFinder:
+    def test_channels_same_after_restart(self, setup_compose: DockerCompose, cf_client: ChannelFinderClient) -> None:  # noqa: F811
+        channels_begin = cf_client.find(name="*")
+        restart_container(setup_compose, "cf")
+        assert wait_for_sync(cf_client, check_connection_active)
+        channels_end = cf_client.find(name="*")
+        assert len(channels_begin) == len(channels_end)
+        channels_match(channels_begin, channels_end, PROPERTIES_TO_MATCH)
+        shutdown_container(setup_compose, "ioc1-1")
+        assert wait_for_sync(
+            cf_client, lambda cf_client: check_channel_property(cf_client, "IOC1-1:Msg-I", INACTIVE_PROPERTY)
+        )
+        channels_inactive = cf_client.find(property=[("iocName", "IOC1-1")])
+        assert all(INACTIVE_PROPERTY in ch["properties"] for ch in channels_inactive)