31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
"""
|
|
Utilities related to nose.
|
|
"""
|
|
import django_nose
|
|
from django.core.management import call_command
|
|
from django.db import DEFAULT_DB_ALIAS, connections, transaction
|
|
|
|
|
|
class NoseTestSuiteRunner(django_nose.NoseTestSuiteRunner):
|
|
"""Custom NoseTestSuiteRunner."""
|
|
|
|
def setup_databases(self):
|
|
""" Setup databases and then flush to remove data added by migrations. """
|
|
return_value = super(NoseTestSuiteRunner, self).setup_databases()
|
|
|
|
# Delete all data added by data migrations. Unit tests should setup their own data using factories.
|
|
call_command('flush', verbosity=0, interactive=False, load_initial_data=False)
|
|
|
|
# Through Django 1.8, auto increment sequences are not reset when calling flush on a SQLite db.
|
|
# So we do it ourselves.
|
|
# http://sqlite.org/autoinc.html
|
|
connection = connections[DEFAULT_DB_ALIAS]
|
|
if connection.vendor == 'sqlite' and not connection.features.supports_sequence_reset:
|
|
with transaction.atomic(using=DEFAULT_DB_ALIAS):
|
|
cursor = connection.cursor()
|
|
cursor.execute(
|
|
"delete from sqlite_sequence;"
|
|
)
|
|
|
|
return return_value
|