From 6666b796f2d08dc8409d954104d7c3350dd37742 Mon Sep 17 00:00:00 2001
From: Jan-Philipp Benecke <jan-philipp@bnck.me>
Date: Thu, 7 Dec 2023 21:18:10 +0100
Subject: [PATCH] Migrate auth tests to use freezegun (#105243)

---
 tests/auth/test_init.py            |  5 +----
 tests/components/auth/test_init.py | 26 ++++++++++++--------------
 2 files changed, 13 insertions(+), 18 deletions(-)

diff --git a/tests/auth/test_init.py b/tests/auth/test_init.py
index ef7beab488b..9e9b48a07f6 100644
--- a/tests/auth/test_init.py
+++ b/tests/auth/test_init.py
@@ -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"}
         )
diff --git a/tests/components/auth/test_init.py b/tests/components/auth/test_init.py
index a33ca702bcf..4088b1819fa 100644
--- a/tests/components/auth/test_init.py
+++ b/tests/components/auth/test_init.py
@@ -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,28 +168,25 @@ 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):
-        code = store(client_id, mock_credential)
+    freezer.move_to(now)
+    code = store(client_id, mock_credential)
 
-    with patch(
-        "homeassistant.util.dt.utcnow", return_value=now + timedelta(minutes=10)
-    ):
-        assert retrieve(client_id, code) is None
+    freezer.move_to(now + timedelta(minutes=10))
+    assert retrieve(client_id, code) is None
 
-    with patch("homeassistant.util.dt.utcnow", return_value=now):
-        code = store(client_id, mock_credential)
+    freezer.move_to(now)
+    code = store(client_id, mock_credential)
 
-    with patch(
-        "homeassistant.util.dt.utcnow",
-        return_value=now + timedelta(minutes=9, seconds=59),
-    ):
-        assert retrieve(client_id, code) == mock_credential
+    freezer.move_to(now + timedelta(minutes=9, seconds=59))
+    assert retrieve(client_id, code) == mock_credential
 
 
 def test_auth_code_store_requires_credentials(mock_credential) -> None: