Work on tango tests

Author: marcelldls

src/fastcs/backends/tango/dsr.py

@@ -155,11 +155,10 @@ def _collect_dsr_args(options: TangoDSROptions) -> list[str]:
 class TangoDSR:
     def __init__(self, mapping: Mapping):
         self._mapping = mapping
+        self.dev_class = self._mapping.controller.__class__.__name__
+        self._device = self._create_device()
 
-    def run(self, options: TangoDSROptions | None = None) -> None:
-        if options is None:
-            options = TangoDSROptions()
-
+    def _create_device(self):
         class_dict: dict = {
             **_collect_dev_attributes(self._mapping),
             **_collect_dev_commands(self._mapping),
@@ -169,12 +168,17 @@ def run(self, options: TangoDSROptions | None = None) -> None:
         }
 
         class_bases = (server.Device,)
-        pytango_class = type(options.dev_class, class_bases, class_dict)
+        pytango_class = type(self.dev_class, class_bases, class_dict)
+        return pytango_class
+
+    def run(self, options: TangoDSROptions | None = None) -> None:
+        if options is None:
+            options = TangoDSROptions()
 
         dsr_args = _collect_dsr_args(options)
 
         server.run(
-            (pytango_class,),
+            (self.dev_class,),
             [options.dev_class, options.dsr_instance, *dsr_args],
             green_mode=server.GreenMode.Asyncio,
         )

tests/backends/tango/test_dsr.py

@@ -1,8 +1,11 @@
 import pytest
 from pytest_mock import MockerFixture
+from tango import DevState
 from tango._tango import AttrWriteType, CmdArgType
 
+from fastcs.attributes import AttrR
 from fastcs.backends.tango.dsr import _collect_dev_attributes, _collect_dev_commands
+from fastcs.datatypes import Bool, Int
 
 
 def test_collect_attributes(mapping):
@@ -35,3 +38,70 @@ async def test_collect_commands(mapping, mocker: MockerFixture):
     # Check that command is created and it can be called
     assert list(commands.keys()) == ["Go"]
     await commands["Go"](mocker.MagicMock())
+
+
+class TestTangoDevice:
+    def test_list_attributes(self, tango_context):
+        assert list(tango_context.get_attribute_list()) == [
+            "BigEnum",
+            "ReadBool",
+            "ReadInt",
+            "ReadWriteFloat",
+            "ReadWriteInt",
+            "StringEnum",
+            "WriteBool",
+            "State",
+            "Status",
+        ]
+
+    def test_list_commands(self, tango_context):
+        assert list(tango_context.get_command_list()) == [
+            "Go",
+            "Init",
+            "State",
+            "Status",
+        ]
+
+    def test_read_int(self, tango_context):
+        result = tango_context.read_attribute("ReadInt")
+        assert result.value == AttrR(Int())._value
+
+    def test_read_write_int(self, tango_context):
+        expected = 42
+        tango_context.write_attribute("ReadWriteInt", expected)
+        result = tango_context.read_attribute("ReadWriteInt")
+        assert result.w_value == expected
+
+    def test_read_write_float(self, tango_context):
+        expected = 23.1
+        tango_context.write_attribute("ReadWriteFloat", expected)
+        result = tango_context.read_attribute("ReadWriteFloat")
+        assert result.w_value == expected
+
+    def test_read_bool(self, tango_context):
+        result = tango_context.read_attribute("ReadBool")
+        assert result.value == AttrR(Bool())._value
+
+    def test_write_bool(self, tango_context):
+        tango_context.write_attribute("WriteBool", AttrR(Bool())._value)
+
+    # # We need to discuss enums
+    # def test_string_enum(self, tango_context):
+    #     expected = 1
+    #     tango_context.write_attribute("StringEnum", "green")
+    #     result = tango_context.read_attribute("StringEnum")
+    #     assert result.w_value == expected
+
+    def test_big_enum(self, tango_context):
+        result = tango_context.read_attribute("BigEnum")
+        assert result.value == AttrR(Int(), allowed_values=list(range(1, 18)))._value
+
+    def test_go(self, tango_context):
+        tango_context.command_inout("go")
+
+    def test_state(self, tango_context):
+        assert tango_context.command_inout("State") == DevState.ON
+
+    def test_status(self, tango_context):
+        expected = "The device is in ON state."
+        assert tango_context.command_inout("Status") == expected

tests/conftest.py

@@ -8,8 +8,10 @@
 
 import pytest
 from aioca import purge_channel_caches
+from tango.test_context import DeviceTestContext
 
 from fastcs.attributes import AttrR, AttrRW, AttrW, Handler, Sender, Updater
+from fastcs.backends.tango.backend import TangoBackend
 from fastcs.controller import Controller
 from fastcs.datatypes import Bool, Float, Int, String
 from fastcs.mapping import Mapping
@@ -58,7 +60,7 @@ class TestController(Controller):
     string_enum: AttrRW = AttrRW(String(), allowed_values=["red", "green", "blue"])
     big_enum: AttrR = AttrR(
         Int(),
-        allowed_values=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17],
+        allowed_values=list(range(1, 18)),
     )
 
     initialised = False
@@ -120,3 +122,10 @@ def ioc():
     except ValueError:
         # Someone else already called communicate
         pass
+
+
+@pytest.fixture(scope="class")
+def tango_context():
+    device = TangoBackend(TestController())._dsr._device
+    with DeviceTestContext(device) as proxy:
+        yield proxy