Пример #1
0
    def setup(self):
        """
        Creates or sets up test database name, loads fixtures and mocks
        ``SMTPConnection`` class from ``django.core.mail``.
        """
        # Creates test database
        create_test_db = True

        settings.original_DATABASE_ENGINE = settings.DATABASE_ENGINE
        settings.original_DATABASE_NAME = settings.DATABASE_NAME

        if self.database_name is None or self.database_name == ':memory:':
            # Closes current database connection and changes
            # ``DATABASE_ENGINE`` to ``'sqlite3'``
            connection.close()
            settings.DATABASE_ENGINE = 'sqlite3'

            settings.TEST_DATABASE_NAME = ':memory:'

            # Flushes in-memory database if it exists
            if db_exists(settings.TEST_DATABASE_NAME):
                flush()
        elif self.database_name == ':original:':
            # Disable database flush by default
            if self.database_flush is None:
                self.database_flush = False

            self.database_name = settings.DATABASE_NAME
            settings.TEST_DATABASE_NAME = settings.DATABASE_NAME

            # If original database not exist - tries to create it
            if db_exists(self.database_name):
                create_test_db = False
        else:
            settings.TEST_DATABASE_NAME = self.database_name

            # Do not re-creates tests database if it exists on
            # ``database_flush == False``
            if self.database_flush == False and \
               db_exists(self.database_name):
                create_test_db = False

        if create_test_db:
            self.database_name = \
                connection.creation.create_test_db(autoclobber=True)

        tables = connection.introspection.table_names()

        if self.database_flush and tables:
            call_command('flush', interactive=False)

        # Load data from fixtures
        if self.fixtures:
            call_command('loaddata', *self.fixtures)

        # Mock original SMTPConnection
        mail.original_SMTPConnection = mail.SMTPConnection
        mail.SMTPConnection = TestSMTPConnection

        mail.outbox = []
Пример #2
0
    def teardown(self):
        # Destroys test database
        if self.database_name == ":memory:" and self.database_flush != False:
            flush()

        if self.database_name != settings.original_DATABASE_NAME and self.database_flush != False:
            connection.creation.destroy_test_db(self.database_name)

        settings.DATABASE_ENGINE = settings.original_DATABASE_ENGINE
        settings.DATABASE_NAME = settings.original_DATABASE_NAME

        # Unmock original SMTPConnection
        mail.SMTPConnection = mail.original_SMTPConnection
        del mail.original_SMTPConnection

        del mail.outbox
Пример #3
0
    def teardown(self):
        # Destroys test database
        if self.database_name == ':memory:' and self.database_flush != False:
            flush()

        if self.database_name != settings.original_DATABASE_NAME and \
           self.database_flush != False:
            connection.creation.destroy_test_db(self.database_name)

        settings.DATABASE_ENGINE = settings.original_DATABASE_ENGINE
        settings.DATABASE_NAME = settings.original_DATABASE_NAME

        # Unmock original SMTPConnection
        mail.SMTPConnection = mail.original_SMTPConnection
        del mail.original_SMTPConnection

        del mail.outbox

        if not self.old_installed_apps == settings.INSTALLED_APPS:
            settings.INSTALLED_APPS = self.old_installed_apps