Django, MongoDB, and MongoEngine: A custom TestRunner managing mongo test database creation and cleanup.

test_runner.py

# coding: utf-8
from django.conf import settings
from django_nose import NoseTestSuiteRunner


class MongoTestRunner(NoseTestSuiteRunner):
    """
        A test runner that can be used to create, connect to, disconnect from, 
        and destroy a mongo test database for standard django testing.
        Redefine TEST_RUNNER in settings
        Example: TEST_RUNNER = 'apps.common.test_runner.MongoTestRunner'
    """
    mongodb_name = f'test_{settings.MONGOENGINE_DATABASE}'

    def setup_databases(self):
        from mongoengine.connection import connect, disconnect
        disconnect()
        connect(self.mongodb_name,
                host=settings.MONGOENGINE_LINK.replace('/' + settings.MONGOENGINE_DATABASE, '/' + self.mongodb_name))
        print(f'Creating mongo test database: {self.mongodb_name}')
        return super().setup_databases()

    def teardown_databases(self, *args, **kwargs):
        from mongoengine.connection import get_connection, disconnect
        connection = get_connection()
        connection.drop_database(self.mongodb_name)
        print(f'Dropping mongo test database: {self.mongodb_name}')
        disconnect()
        super().teardown_databases(*args, **kwargs)