# -*- coding: utf-8 -*-
from django.test.simple import DjangoTestSuiteRunner as TestRunner
from django.db import connections, DEFAULT_DB_ALIAS


class CustomTestRunner(TestRunner):
    """
    Из-за того, что проект достаточно сильно разросся, мы хотим для
    тестов использовать копию рабочей бд
    (без создания и удаления бд для тестов)

    В Django >= 1.8 для этого можно использовать ключ keepdb
    """
    def setup_databases(self, **kwargs):
        db = connections.databases[DEFAULT_DB_ALIAS]['NAME']
        connections[DEFAULT_DB_ALIAS].settings_dict['NAME'] = 'test_%s' % db

    def teardown_databases(self, old_config, **kwargs):
        pass