File tree Expand file tree Collapse file tree 1 file changed +14
-11
lines changed Expand file tree Collapse file tree 1 file changed +14
-11
lines changed Original file line number Diff line number Diff line change @@ -145,25 +145,28 @@ def _django_db_fixture_helper(
145
145
django_db_blocker .unblock ()
146
146
request .addfinalizer (django_db_blocker .restore )
147
147
148
+ import django .test
149
+ import django .db
150
+
148
151
if transactional :
149
- from django .test import TransactionTestCase as django_case
152
+ test_case_class = django .test .TransactionTestCase
153
+ else :
154
+ test_case_class = django .test .TestCase
150
155
151
- if reset_sequences :
156
+ _reset_sequences = reset_sequences
152
157
153
- class ResetSequenceTestCase (django_case ):
154
- reset_sequences = True
158
+ class PytestDjangoTestCase (test_case_class ): # type: ignore[misc,valid-type]
159
+ if transactional and _reset_sequences :
160
+ reset_sequences = True
155
161
156
- django_case = ResetSequenceTestCase
157
- else :
158
- from django .test import TestCase as django_case
159
- from django .db import transaction
160
- transaction .Atomic ._ensure_durability = False
162
+ if not transactional :
163
+ django .db .transaction .Atomic ._ensure_durability = False
161
164
162
165
def reset_durability () -> None :
163
- transaction .Atomic ._ensure_durability = True
166
+ django . db . transaction .Atomic ._ensure_durability = True
164
167
request .addfinalizer (reset_durability )
165
168
166
- test_case = django_case (methodName = "__init__" )
169
+ test_case = PytestDjangoTestCase (methodName = "__init__" )
167
170
test_case ._pre_setup ()
168
171
request .addfinalizer (test_case ._post_teardown )
169
172
You can’t perform that action at this time.
0 commit comments