pydase/tests/components/test_device_connection.py

29 lines
735 B
Python

import asyncio
import pydase
import pydase.components.device_connection
import pytest
from pytest import LogCaptureFixture
@pytest.mark.asyncio(scope="session")
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