Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Use hass_client_no_auth test fixture in integrations s-x (#55585)

Browse Source
This commit is contained in:
Erik Montnemery
2021-09-02 14:50:10 +02:00
committed by GitHub
parent acdddabe1f
commit bfd799dc04
13 changed files with 60 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/websocket_api/conftest.py
View File

@@ -13,12 +13,12 @@ async def websocket_client(hass, hass_ws_client):
@pytest.fixture
async def no_auth_websocket_client(hass, aiohttp_client):
async def no_auth_websocket_client(hass, hass_client_no_auth):
"""Websocket connection that requires authentication."""
assert await async_setup_component(hass, "websocket_api", {})
await hass.async_block_till_done()
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
ws = await client.ws_connect(URL)
auth_ok = await ws.receive_json()

18
tests/components/websocket_api/test_auth.py
View File

@@ -121,14 +121,14 @@ async def test_auth_active_with_token(
assert auth_msg["type"] == TYPE_AUTH_OK
async def test_auth_active_user_inactive(hass, aiohttp_client, hass_access_token):
async def test_auth_active_user_inactive(hass, hass_client_no_auth, hass_access_token):
"""Test authenticating with a token."""
refresh_token = await hass.auth.async_validate_access_token(hass_access_token)
refresh_token.user.is_active = False
assert await async_setup_component(hass, "websocket_api", {})
await hass.async_block_till_done()
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
async with client.ws_connect(URL) as ws:
auth_msg = await ws.receive_json()
@@ -140,12 +140,12 @@ async def test_auth_active_user_inactive(hass, aiohttp_client, hass_access_token
assert auth_msg["type"] == TYPE_AUTH_INVALID
async def test_auth_active_with_password_not_allow(hass, aiohttp_client):
async def test_auth_active_with_password_not_allow(hass, hass_client_no_auth):
"""Test authenticating with a token."""
assert await async_setup_component(hass, "websocket_api", {})
await hass.async_block_till_done()
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
async with client.ws_connect(URL) as ws:
auth_msg = await ws.receive_json()
@@ -157,12 +157,14 @@ async def test_auth_active_with_password_not_allow(hass, aiohttp_client):
assert auth_msg["type"] == TYPE_AUTH_INVALID
async def test_auth_legacy_support_with_password(hass, aiohttp_client, legacy_auth):
async def test_auth_legacy_support_with_password(
hass, hass_client_no_auth, legacy_auth
):
"""Test authenticating with a token."""
assert await async_setup_component(hass, "websocket_api", {})
await hass.async_block_till_done()
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
async with client.ws_connect(URL) as ws:
auth_msg = await ws.receive_json()
@@ -174,12 +176,12 @@ async def test_auth_legacy_support_with_password(hass, aiohttp_client, legacy_au
assert auth_msg["type"] == TYPE_AUTH_INVALID
async def test_auth_with_invalid_token(hass, aiohttp_client):
async def test_auth_with_invalid_token(hass, hass_client_no_auth):
"""Test authenticating with a token."""
assert await async_setup_component(hass, "websocket_api", {})
await hass.async_block_till_done()
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
async with client.ws_connect(URL) as ws:
auth_msg = await ws.receive_json()

6
tests/components/websocket_api/test_commands.py
View File

@@ -459,12 +459,14 @@ async def test_ping(websocket_client):
assert msg["type"] == "pong"
async def test_call_service_context_with_user(hass, aiohttp_client, hass_access_token):
async def test_call_service_context_with_user(
hass, hass_client_no_auth, hass_access_token
):
"""Test that the user is set in the service call context."""
assert await async_setup_component(hass, "websocket_api", {})
calls = async_mock_service(hass, "domain_test", "test_service")
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
async with client.ws_connect(URL) as ws:
auth_msg = await ws.receive_json()

4
tests/components/websocket_api/test_sensor.py
View File

@@ -7,14 +7,14 @@ from homeassistant.components.websocket_api.http import URL
from .test_auth import test_auth_active_with_token
async def test_websocket_api(hass, aiohttp_client, hass_access_token, legacy_auth):
async def test_websocket_api(hass, hass_client_no_auth, hass_access_token, legacy_auth):
"""Test API streams."""
await async_setup_component(
hass, "sensor", {"sensor": {"platform": "websocket_api"}}
)
await hass.async_block_till_done()
client = await aiohttp_client(hass.http.app)
client = await hass_client_no_auth()
ws = await client.ws_connect(URL)
auth_ok = await ws.receive_json()