From 479a35c49938d688ee689fc18f311baadb14c074 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Wed, 5 Apr 2023 19:12:21 -1000
Subject: [PATCH] Revert "Move local calendar diagnostics to
 pytest.mark.freezetime" (#90899)

Revert "Move local calendar diagnostics to pytest.mark.freezetime (#90886)"

This reverts commit a01952981fe91e08fd96cde976959cb212197947.
---
 .../local_calendar/test_diagnostics.py        | 81 ++++++++++++++++---
 1 file changed, 70 insertions(+), 11 deletions(-)

diff --git a/tests/components/local_calendar/test_diagnostics.py b/tests/components/local_calendar/test_diagnostics.py
index 1f7b95873db..9a1da25d770 100644
--- a/tests/components/local_calendar/test_diagnostics.py
+++ b/tests/components/local_calendar/test_diagnostics.py
@@ -1,15 +1,43 @@
 """Tests for diagnostics platform of local calendar."""
+from aiohttp.test_utils import TestClient
+from freezegun import freeze_time
 import pytest
 from syrupy.assertion import SnapshotAssertion
 
+from homeassistant.auth.models import Credentials
 from homeassistant.core import HomeAssistant
 from homeassistant.setup import async_setup_component
 
-from .conftest import TEST_ENTITY, ClientFixture
+from .conftest import TEST_ENTITY, Client
 
-from tests.common import MockConfigEntry
+from tests.common import CLIENT_ID, MockConfigEntry, MockUser
 from tests.components.diagnostics import get_diagnostics_for_config_entry
-from tests.typing import ClientSessionGenerator
+from tests.typing import ClientSessionGenerator, WebSocketGenerator
+
+
+async def generate_new_hass_access_token(
+    hass: HomeAssistant, hass_admin_user: MockUser, hass_admin_credential: Credentials
+) -> str:
+    """Return an access token to access Home Assistant."""
+    await hass.auth.async_link_user(hass_admin_user, hass_admin_credential)
+
+    refresh_token = await hass.auth.async_create_refresh_token(
+        hass_admin_user, CLIENT_ID, credential=hass_admin_credential
+    )
+    return hass.auth.async_create_access_token(refresh_token)
+
+
+def _get_test_client_generator(
+    hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, new_token: str
+):
+    """Return a test client generator.""."""
+
+    async def auth_client() -> TestClient:
+        return await aiohttp_client(
+            hass.http.app, headers={"Authorization": f"Bearer {new_token}"}
+        )
+
+    return auth_client
 
 
 @pytest.fixture(autouse=True)
@@ -18,30 +46,58 @@ async def setup_diag(hass):
     assert await async_setup_component(hass, "diagnostics", {})
 
 
-@pytest.mark.freeze_time("2023-03-13 12:05:00-07:00")
+@freeze_time("2023-03-13 12:05:00-07:00")
 async def test_empty_calendar(
     hass: HomeAssistant,
     setup_integration: None,
-    hass_client: ClientSessionGenerator,
+    hass_admin_user: MockUser,
+    hass_admin_credential: Credentials,
     config_entry: MockConfigEntry,
+    aiohttp_client: ClientSessionGenerator,
+    socket_enabled: None,
     snapshot: SnapshotAssertion,
 ) -> None:
     """Test diagnostics against an empty calendar."""
-    data = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
+    # Since we are freezing time only when we enter this test, we need to
+    # manually create a new token and clients since the token created by
+    # the fixtures would not be valid.
+    #
+    # Ideally we would use pytest.mark.freeze_time before the fixtures, but that does not
+    # work with the ical library and freezegun because
+    # `TypeError: '<' not supported between instances of 'FakeDatetimeMeta' and 'FakeDateMeta'`
+    new_token = await generate_new_hass_access_token(
+        hass, hass_admin_user, hass_admin_credential
+    )
+    data = await get_diagnostics_for_config_entry(
+        hass, _get_test_client_generator(hass, aiohttp_client, new_token), config_entry
+    )
     assert data == snapshot
 
 
-@pytest.mark.freeze_time("2023-03-13 12:05:00-07:00")
+@freeze_time("2023-03-13 12:05:00-07:00")
 async def test_api_date_time_event(
     hass: HomeAssistant,
     setup_integration: None,
+    hass_admin_user: MockUser,
+    hass_admin_credential: Credentials,
     config_entry: MockConfigEntry,
-    hass_client: ClientSessionGenerator,
-    ws_client: ClientFixture,
+    hass_ws_client: WebSocketGenerator,
+    aiohttp_client: ClientSessionGenerator,
+    socket_enabled: None,
     snapshot: SnapshotAssertion,
 ) -> None:
     """Test an event with a start/end date time."""
-    client = await ws_client()
+    # Since we are freezing time only when we enter this test, we need to
+    # manually create a new token and clients since the token created by
+    # the fixtures would not be valid.
+    #
+    # Ideally we would use pytest.mark.freeze_time before the fixtures, but that does not
+    # work with the ical library and freezegun because
+    # `TypeError: '<' not supported between instances of 'FakeDatetimeMeta' and 'FakeDateMeta'`
+    new_token = await generate_new_hass_access_token(
+        hass, hass_admin_user, hass_admin_credential
+    )
+    client = Client(await hass_ws_client(hass, access_token=new_token))
     await client.cmd_result(
         "create",
         {
@@ -54,5 +110,8 @@ async def test_api_date_time_event(
             },
         },
     )
-    data = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
+
+    data = await get_diagnostics_for_config_entry(
+        hass, _get_test_client_generator(hass, aiohttp_client, new_token), config_entry
+    )
     assert data == snapshot