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

Migrate auth tests to use freezegun (#105243)

Browse source
This commit is contained in:
Jan-Philipp Benecke 2023-12-07 21:18:10 +01:00 • committed by GitHub
parent 2daa94b600
commit 6666b796f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/auth/test_init.py
View file

@ -894,10 +894,7 @@ async def test_auth_module_expired_session(mock_hass) -> None:
assert step["type"] == data_entry_flow.FlowResultType.FORM
assert step["step_id"] == "mfa"
with patch(
"homeassistant.util.dt.utcnow",
return_value=dt_util.utcnow() + MFA_SESSION_EXPIRATION,
):
with freeze_time(dt_util.utcnow() + MFA_SESSION_EXPIRATION):
step = await manager.login_flow.async_configure(
step["flow_id"], {"pin": "test-pin"}
)

18
tests/components/auth/test_init.py
View file

@ -4,6 +4,7 @@ from http import HTTPStatus
import logging
from unittest.mock import patch
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.auth import InvalidAuthError
@ -167,27 +168,24 @@ async def test_auth_code_checks_local_only_user(
assert error["error"] == "access_denied"
def test_auth_code_store_expiration(mock_credential) -> None:
def test_auth_code_store_expiration(
mock_credential, freezer: FrozenDateTimeFactory
) -> None:
"""Test that the auth code store will not return expired tokens."""
store, retrieve = auth._create_auth_code_store()
client_id = "bla"
now = utcnow()
with patch("homeassistant.util.dt.utcnow", return_value=now):
freezer.move_to(now)
code = store(client_id, mock_credential)
with patch(
"homeassistant.util.dt.utcnow", return_value=now + timedelta(minutes=10)
):
freezer.move_to(now + timedelta(minutes=10))
assert retrieve(client_id, code) is None
with patch("homeassistant.util.dt.utcnow", return_value=now):
freezer.move_to(now)
code = store(client_id, mock_credential)
with patch(
"homeassistant.util.dt.utcnow",
return_value=now + timedelta(minutes=9, seconds=59),
):
freezer.move_to(now + timedelta(minutes=9, seconds=59))
assert retrieve(client_id, code) == mock_credential