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
