Hopiu/django-cachalot
1
0
Fork 0
mirror of https://github.com/Hopiu/django-cachalot.git synced 2026-03-16 21:30:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixes tests.

Browse source
This commit is contained in:
Bertrand Bordage 2018-06-14 15:57:24 +02:00
parent 51abf85f70
commit c832784df0
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cachalot/tests/test_utils.py
View file

@ -1,5 +1,5 @@
from django.core.management.color import no_style
from django.db import connection, DEFAULT_DB_ALIAS
from django.db import connection, transaction
from django.utils.six import string_types
from .models import PostgresModel
@ -15,9 +15,13 @@ class TestUtilsMixin:
# TODO: Remove this workaround when this issue is fixed:
# https://code.djangoproject.com/ticket/29494
def tearDown(self):
flush_sql_list = connection.ops.sql_flush(
no_style(), (PostgresModel._meta.db_table,), ())
connection.ops.execute_sql_flush(DEFAULT_DB_ALIAS, flush_sql_list)
if connection.vendor == 'postgresql':
flush_sql_list = connection.ops.sql_flush(
no_style(), (PostgresModel._meta.db_table,), ())
with transaction.atomic():
for sql in flush_sql_list:
with connection.cursor() as cursor:
cursor.execute(sql)
def force_repoen_connection(self):
if connection.vendor in ('mysql', 'postgresql'):