Skip to content

Commit 1b6919e

Browse files
committed
Refactor reset_checks django fixture
- remove version check - autouse so that checks are cleared after every test run
1 parent 658befa commit 1b6919e

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/django/test_django.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66

77
import pytest
88
import redis
9-
from django import VERSION as django_version
109
from django.core.checks.registry import registry
1110
from django.core.exceptions import ImproperlyConfigured
1211
from django.db import connection
@@ -20,14 +19,11 @@
2019
from dockerflow.django.middleware import DockerflowMiddleware
2120

2221

23-
@pytest.fixture
22+
@pytest.fixture(autouse=True)
2423
def reset_checks():
25-
if django_version[0] < 2:
26-
registry.registered_checks = []
27-
registry.deployment_checks = []
28-
else:
29-
registry.registered_checks = set()
30-
registry.deployment_checks = set()
24+
yield
25+
registry.registered_checks = set()
26+
registry.deployment_checks = set()
3127

3228

3329
@pytest.fixture

0 commit comments

Comments
 (0)