test: replace pytest-freezegun with freezgun

mozilla-services · Nov 14, 2022 · a13fe46 · a13fe46
1 parent e8d2001
commit a13fe46
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 23 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -84,7 +84,7 @@ types-PyYAML = "^6.0.11"
 types-requests = "^2.28.10"
 types-geoip2 = "^3.0.0"
 scalene = "^1.5.13"
-pytest-freezegun = "^0.4.2"
+freezegun = "^1.2.2"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

diff --git a/tests/integration/api/v1/providers/test_providers.py b/tests/integration/api/v1/providers/test_providers.py
@@ -10,6 +10,7 @@
 
 import pytest
 from fastapi.testclient import TestClient
+from freezegun import freeze_time
 from pytest import LogCaptureFixture
 
 from merino.providers import BaseProvider
@@ -85,7 +86,7 @@ def test_providers(
     assert response.json() == expected_response
 
 
-@pytest.mark.freeze_time("1998-03-31")
+@freeze_time("1998-03-31")
 @pytest.mark.parametrize("providers", [{}])
 def test_providers_request_logs(
     caplog: LogCaptureFixture,

diff --git a/tests/integration/api/v1/suggest/test_suggest.py b/tests/integration/api/v1/suggest/test_suggest.py
@@ -9,6 +9,7 @@
 
 import pytest
 from fastapi.testclient import TestClient
+from freezegun import freeze_time
 from pytest import LogCaptureFixture
 from pytest_mock import MockerFixture
 
@@ -81,7 +82,7 @@ def test_client_variants(client: TestClient) -> None:
     assert result["client_variants"] == ["foo", "bar"]
 
 
-@pytest.mark.freeze_time("1998-03-31")
+@freeze_time("1998-03-31")
 def test_suggest_request_logs(
     mocker: MockerFixture,
     caplog: LogCaptureFixture,