pydase/tests/components/test_device_connection.py
2024-09-13 16:09:39 +02:00

29 lines
780 B
Python

import asyncio
import pydase
import pydase.components.device_connection
import pytest
from pydase.task.autostart import autostart_service_tasks
from pytest import LogCaptureFixture
@pytest.mark.asyncio(scope="function")
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()
autostart_service_tasks(service_instance)
assert service_instance._connected is False
await asyncio.sleep(0.01)
assert service_instance._connected is True