Use mocks instead of instances

Author: kennethjyang

tests/back_end/test_platform_handler.py

@@ -37,6 +37,39 @@
 class TestPlatformHandler:
     """Tests for the PlatformHandler class."""
 
+    @pytest.fixture
+    def console(self, mocker: MockerFixture) -> Console:
+        """Console instance for testing."""
+
+        def exception_to_string(exception: Exception) -> str:
+            return str(exception)
+
+        # Create mock.
+        mocked_console = mocker.Mock(spec=Console)
+
+        # Patch internally used functions.
+        _ = mocker.patch.object(mocked_console, "pretty_exception", side_effect=exception_to_string)
+
+        # Return mocked console.
+        return mocked_console
+
+    @pytest.fixture
+    def binding(self, mocker: MockerFixture) -> BaseBinding:
+        """FakeBinding instance for testing."""
+
+        def reflect_vector4(input_vector: Vector4) -> Vector4:
+            return input_vector
+
+        # Create mock.
+        mocked_binding = mocker.Mock(spec=BaseBinding)
+
+        # Patch internally used functions.
+        _ = mocker.patch.object(mocked_binding, "platform_space_to_unified_space", side_effect=reflect_vector4)
+        _ = mocker.patch.object(mocked_binding, "unified_space_to_platform_space", side_effect=reflect_vector4)
+
+        # Return mocked binding.
+        return mocked_binding
+
     @pytest.fixture
     def platform_handler(self, binding: BaseBinding, console: Console) -> PlatformHandler:
         """Fixture for creating a PlatformHandler instance."""
@@ -546,7 +579,7 @@ async def test_emergency_stop(
         """Platform should call stop_all and print to critical console."""
         # Mock binding.
         patched_stop_all = mocker.patch.object(
-            PlatformHandler,
+            platform_handler,
             "stop_all",
         )
         spied_critical_print = mocker.spy(console, "critical_print")

tests/conftest.py

@@ -1,10 +1,5 @@
-import pytest
-from pytest_mock import MockerFixture
 from vbl_aquarium.models.unity import Vector3, Vector4
 
-from ephys_link.front_end.console import Console
-from ephys_link.utils.base_binding import BaseBinding
-
 # Dummy values for testing.
 DUMMY_STRING = "Dummy String"
 DUMMY_SMALL_STRING = "dummy"
@@ -13,23 +8,3 @@
 DUMMY_VECTOR3 = Vector3(x=1.0, y=2.0, z=3.0)
 DUMMY_VECTOR4 = Vector4(x=1.0, y=2.0, z=3.0, w=4.0)
 DUMMY_EXCEPTION = RuntimeError("Test runtime error")
-
-
-@pytest.fixture
-def console() -> Console:
-    """Console instance for testing."""
-    return Console()
-
-
-@pytest.fixture
-def binding(mocker: MockerFixture) -> BaseBinding:
-    """FakeBinding instance for testing."""
-
-    def reflect_vector4(input_vector: Vector4) -> Vector4:
-        return input_vector
-
-    # Patch the BaseBinding class.
-    mocked_binding = mocker.Mock(spec=BaseBinding)
-    _ = mocker.patch.object(mocked_binding, "platform_space_to_unified_space", side_effect=reflect_vector4)
-    _ = mocker.patch.object(mocked_binding, "unified_space_to_platform_space", side_effect=reflect_vector4)
-    return mocked_binding