evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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