Skip to content

Commit a5d510e

Browse files
committed
Fixes suggested by checkqa
1 parent 9a77e95 commit a5d510e

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

pytest_django/fixtures.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
from contextlib import contextmanager
66
import os
7+
import sys
78
import warnings
89

910
import pytest
@@ -223,6 +224,7 @@ def wrapper(request):
223224
# Use atomic instead of calling .savepoint* directly.
224225
# This way works for both top-level transactions and "subtransactions".
225226
atomic = transaction.atomic()
227+
226228
def finalize():
227229
# Only run __exit__ if there was no error running the wrapped function.
228230
# Otherwise we've run it already.

pytest_django/plugin.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
# Silence linters for imported fixtures.
2525
(_django_db_setup, _live_server_helper, admin_client, admin_user, client, db,
2626
django_user_model, django_username_field, live_server, rf, settings,
27-
transactional_db)
27+
shared_db_wrapper, transactional_db)
2828

2929

3030
SETTINGS_MODULE_ENV = 'DJANGO_SETTINGS_MODULE'

0 commit comments

Comments
 (0)