This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions

View file

@ -11,25 +11,23 @@ from . import API_PASSWORD
@pytest.fixture
def websocket_client(hass, hass_ws_client, hass_access_token):
"""Create a websocket client."""
return hass.loop.run_until_complete(
hass_ws_client(hass, hass_access_token))
return hass.loop.run_until_complete(hass_ws_client(hass, hass_access_token))
@pytest.fixture
def no_auth_websocket_client(hass, loop, aiohttp_client):
"""Websocket connection that requires authentication."""
assert loop.run_until_complete(
async_setup_component(hass, 'websocket_api', {
'http': {
'api_password': API_PASSWORD
}
}))
async_setup_component(
hass, "websocket_api", {"http": {"api_password": API_PASSWORD}}
)
)
client = loop.run_until_complete(aiohttp_client(hass.http.app))
ws = loop.run_until_complete(client.ws_connect(URL))
auth_ok = loop.run_until_complete(ws.receive_json())
assert auth_ok['type'] == TYPE_AUTH_REQUIRED
assert auth_ok["type"] == TYPE_AUTH_REQUIRED
yield ws