Support fixtures inside classes decorated with freeze_time

Author: bblommers

freezegun/api.py

@@ -693,7 +693,6 @@ def tearDown(*args: Any, **kwargs: Any) -> None:
             klass.tearDown = tearDown  # type: ignore[method-assign]
 
         else:
-
             seen = set()
 
             klasses = klass.mro()
@@ -707,7 +706,20 @@ def tearDown(*args: Any, **kwargs: Any) -> None:
                         continue
 
                     try:
-                        setattr(klass, attr, self(attr_value))
+                        if attr_value.__dict__.get("_pytestfixturefunction") and hasattr(attr_value, "__pytest_wrapped__"):
+                            # PYTEST==8.2.x (and maybe others)
+                            # attr_value is a pytest fixture
+                            # In other words: attr_value == fixture(original_method)
+                            # We need to keep the fixture itself intact to ensure pytest still treats it as a fixture
+                            # We still want to freeze time inside the original_method though
+                            attr_value.__pytest_wrapped__.obj = self(attr_value.__pytest_wrapped__.obj)
+                        elif attr_value.__dict__.get("_fixture_function"):
+                            # PYTEST==8.4.x
+                            # Same
+                            attr_value._fixture_function = self(attr_value._fixture_function)
+                        else:
+                            # Wrap the entire method inside 'freeze_time'
+                            setattr(klass, attr, self(attr_value))
                     except (AttributeError, TypeError):
                         # Sometimes we can't set this for built-in types and custom callables
                         continue

tests/test_class_decorator.py

@@ -0,0 +1,19 @@
+from datetime import datetime
+from unittest import TestCase
+
+import pytest
+from freezegun import freeze_time
+from freezegun.api import FakeDatetime
+
+@freeze_time("2022-10-01")
+class TestClassDecoratorWithFixture:
+    @pytest.fixture
+    def ff(self) -> datetime:
+        return datetime.now()
+
+    def test_with_fixture(self, ff: datetime) -> None:
+        assert ff == FakeDatetime(2022, 10, 1, 0, 0)
+        assert datetime.now() == FakeDatetime(2022, 10, 1, 0, 0)
+
+    def test_without_fixture(self) -> None:
+        assert datetime.now() == FakeDatetime(2022, 10, 1, 0, 0)