def mock_client_fixture(controller_state, version_state): """Mock a client.""" with patch("homeassistant.components.zwave_js.ZwaveClient", autospec=True) as client_class: client = client_class.return_value async def connect(): await asyncio.sleep(0) client.connected = True async def listen(driver_ready: asyncio.Event) -> None: driver_ready.set() await asyncio.sleep(30) assert False, "Listen wasn't canceled!" async def disconnect(): client.connected = False client.connect = AsyncMock(side_effect=connect) client.listen = AsyncMock(side_effect=listen) client.disconnect = AsyncMock(side_effect=disconnect) client.driver = Driver(client, controller_state) client.version = VersionInfo.from_message(version_state) client.ws_server_url = "ws://test:3000/zjs" yield client
def driver_fixture(client, controller_state): """Return a driver instance with a supporting client.""" return Driver(client, controller_state)
def driver_fixture(client, controller_state, log_config): """Return a driver instance with a supporting client.""" client.driver = Driver(client, controller_state, log_config) return client.driver