2020-10-23 16:28:21 +02:00
|
|
|
"""deconz conftest."""
|
2021-03-12 21:03:29 +01:00
|
|
|
|
|
|
|
|
from typing import Optional
|
|
|
|
|
from unittest.mock import patch
|
|
|
|
|
|
|
|
|
|
import pytest
|
|
|
|
|
|
2021-03-02 10:02:04 +02:00
|
|
|
from tests.components.light.conftest import mock_light_profiles # noqa: F401
|
2021-03-12 21:03:29 +01:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
|
|
|
def mock_deconz_websocket():
|
|
|
|
|
"""No real websocket allowed."""
|
|
|
|
|
with patch("pydeconz.gateway.WSClient") as mock:
|
|
|
|
|
|
|
|
|
|
async def make_websocket_call(data: Optional[dict] = None, state: str = ""):
|
|
|
|
|
"""Generate a websocket call."""
|
|
|
|
|
pydeconz_gateway_session_handler = mock.call_args[0][3]
|
|
|
|
|
|
|
|
|
|
if data:
|
|
|
|
|
mock.return_value.data = data
|
|
|
|
|
await pydeconz_gateway_session_handler(signal="data")
|
|
|
|
|
elif state:
|
|
|
|
|
mock.return_value.state = state
|
|
|
|
|
await pydeconz_gateway_session_handler(signal="state")
|
|
|
|
|
else:
|
|
|
|
|
raise NotImplementedError
|
|
|
|
|
|
|
|
|
|
yield make_websocket_call
|
