Test server launch

Author: kennethjyang

tests/back_end/test_server.py

@@ -0,0 +1,65 @@
+import pytest
+from pytest_mock import MockerFixture
+from vbl_aquarium.models.ephys_link import EphysLinkOptions, GetManipulatorsResponse
+
+import ephys_link.back_end.server
+from ephys_link.back_end.platform_handler import PlatformHandler
+from ephys_link.back_end.server import Server
+from ephys_link.front_end.console import Console
+from tests.conftest import DUMMY_STRING, DUMMY_STRING_LIST
+
+
+class TestServer:
+    @pytest.fixture
+    def platform_handler(self, mocker: MockerFixture) -> PlatformHandler:
+        """Fixture for mock PlatformHandler."""
+        return mocker.Mock(spec=PlatformHandler)
+
+    @pytest.fixture
+    def console(self, mocker: MockerFixture) -> Console:
+        """Fixture for mock console."""
+        return mocker.Mock(spec=Console)
+
+    @pytest.fixture
+    def server(self, platform_handler: PlatformHandler, console: Console) -> Server:
+        """Fixture for server."""
+        return Server(EphysLinkOptions(use_proxy=False), platform_handler, console)
+
+    @pytest.fixture
+    def proxy_client(self, platform_handler: PlatformHandler, console: Console) -> Server:
+        """Fixture for server as proxy client."""
+        return Server(EphysLinkOptions(use_proxy=True), platform_handler, console)
+
+    def test_launch_server(
+        self, server: Server, platform_handler: PlatformHandler, console: Console, mocker: MockerFixture
+    ) -> None:
+        """Server should print info then start."""
+        # Add mocks and spies.
+        spied_info_print = mocker.spy(console, "info_print")
+
+        patched_get_display_name = mocker.patch.object(platform_handler, "get_display_name", return_value=DUMMY_STRING)
+
+        # Mock out get manipulators.
+        patched_get_manipulators = mocker.patch.object(platform_handler, "get_manipulators", new=mocker.Mock())
+        asyncio_loop = mocker.Mock()
+        patched_run_until_complete = mocker.patch.object(
+            asyncio_loop, "run_until_complete", return_value=GetManipulatorsResponse(manipulators=DUMMY_STRING_LIST)
+        )
+        patched_get_event_loop = mocker.patch.object(
+            ephys_link.back_end.server, "get_event_loop", return_value=asyncio_loop
+        )
+
+        # Mock out run_app.
+        mocked_run_app = mocker.patch.object(ephys_link.back_end.server, "run_app")
+
+        # Act.
+        server.launch()
+
+        # Assert.
+        patched_get_display_name.assert_called_once()
+        patched_get_manipulators.assert_called_once()
+        patched_run_until_complete.assert_called_once()
+        patched_get_event_loop.assert_called_once()
+        spied_info_print.assert_any_call("PLATFORM", platform_handler.get_display_name())
+        spied_info_print.assert_any_call("MANIPULATORS", str(DUMMY_STRING_LIST))
+        mocked_run_app.assert_called_once()