From 25139b3d4d0f262b516fc499119739624638d431 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mose=20M=C3=BCller?= Date: Thu, 15 Feb 2024 08:56:13 +0100 Subject: [PATCH] adds device connection test --- tests/components/test_device_connection.py | 32 ++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tests/components/test_device_connection.py diff --git a/tests/components/test_device_connection.py b/tests/components/test_device_connection.py new file mode 100644 index 0000000..d9436ee --- /dev/null +++ b/tests/components/test_device_connection.py @@ -0,0 +1,32 @@ +import asyncio +import logging + +import pydase +import pydase.components.device_connection +from pytest import LogCaptureFixture + +from tests.utils.test_serializer import pytest + +logger = logging.getLogger(__name__) + + +@pytest.mark.asyncio +async def test_reconnection(caplog: LogCaptureFixture) -> None: + class MyService(pydase.components.device_connection.DeviceConnection): + def __init__( + self, + ) -> None: + super().__init__() + self._reconnection_wait_time = 0.01 + + def connect(self) -> None: + self._connected = True + + service_instance = MyService() + + assert service_instance._connected is False + + service_instance._task_manager.start_autostart_tasks() + + await asyncio.sleep(0.01) + assert service_instance._connected is True