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

Disconnect websockets after token is revoked (#57091)

Browse Source 
Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
This commit is contained in:
David Boslee
2021-10-08 10:38:22 -06:00
committed by GitHub
parent 830e2bc47a
commit 7f966613bd
6 changed files with 94 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/auth/test_init.py
View File

@@ -363,11 +363,15 @@ async def test_ws_refresh_tokens(hass, hass_ws_client, hass_access_token):
assert token["last_used_ip"] == refresh_token.last_used_ip
async def test_ws_delete_refresh_token(hass, hass_ws_client, hass_access_token):
async def test_ws_delete_refresh_token(
hass, hass_admin_user, hass_admin_credential, hass_ws_client, hass_access_token
):
"""Test deleting a refresh token."""
assert await async_setup_component(hass, "auth", {"http": {}})
refresh_token = await hass.auth.async_validate_access_token(hass_access_token)
refresh_token = await hass.auth.async_create_refresh_token(
hass_admin_user, CLIENT_ID, credential=hass_admin_credential
)
ws_client = await hass_ws_client(hass, hass_access_token)
@@ -382,7 +386,7 @@ async def test_ws_delete_refresh_token(hass, hass_ws_client, hass_access_token):
result = await ws_client.receive_json()
assert result["success"], result
refresh_token = await hass.auth.async_validate_access_token(hass_access_token)
refresh_token = await hass.auth.async_get_refresh_token(refresh_token.id)
assert refresh_token is None