Add some tests

tests/system/test_introspection.py

@@ -6,6 +6,8 @@
 from time import sleep
 
 import pytest
+from fastcs.attributes import AttrR
+from fastcs.datatypes import Float
 
 from eiger_fastcs.eiger_controller import EigerController, EigerParameter
 
@@ -78,4 +80,11 @@ async def test_introspection(sim_eiger_controller: EigerController):
 
     assert parameters == expected_parameters, "Detector API does not match"
 
+    attributes = controller._create_attributes(_parameters)
+
+    assert len(attributes) == 91
+    assert isinstance(attributes["humidity"], AttrR)
+    assert isinstance(attributes["humidity"].datatype, Float)
+    assert attributes["humidity"]._group == "DetectorStatus"
+
     await controller.connection.close()

tests/test_controller.py

@@ -0,0 +1,45 @@
+import pytest
+from pytest_mock import MockerFixture
+
+from eiger_fastcs.eiger_controller import EigerController
+
+
+@pytest.mark.asyncio
+async def test_initialise(mocker: MockerFixture):
+    controller = EigerController("127.0.0.1", 80)
+
+    connection = mocker.patch.object(controller, "connection")
+    connection.get = mocker.AsyncMock()
+    connection.get.return_value = {"value": "idle"}
+    initialize = mocker.patch.object(controller, "initialize")
+    introspect = mocker.patch.object(controller, "_introspect_detector")
+    create_attributes = mocker.patch.object(controller, "_create_attributes")
+    attr = mocker.MagicMock()
+    create_attributes.return_value = {"attr_name": attr}
+
+    await controller.initialise()
+
+    connection.get.assert_called_once_with("detector/api/1.8.0/status/state")
+    initialize.assert_not_called()
+    introspect.assert_awaited_once_with()
+    create_attributes.assert_called_once_with(introspect.return_value)
+    assert controller.attr_name == attr, "Attribute not added to controller"
+
+
+@pytest.mark.asyncio
+async def test_initialise_state_na(mocker: MockerFixture):
+    controller = EigerController("127.0.0.1", 80)
+
+    connection = mocker.patch.object(controller, "connection")
+    connection.get = mocker.AsyncMock()
+    connection.get.return_value = {"value": "na"}
+    initialize = mocker.patch.object(controller, "initialize")
+    introspect = mocker.patch.object(controller, "_introspect_detector")
+    create_attributes = mocker.patch.object(controller, "_create_attributes")
+
+    await controller.initialise()
+
+    connection.get.assert_called_once_with("detector/api/1.8.0/status/state")
+    initialize.assert_awaited_once_with()
+    introspect.assert_awaited_once_with()
+    create_attributes.assert_called_once_with(introspect.return_value)