def test_monkey_patch_test_db(self, mock_method): monkey_patch_test_db() creation = BaseDatabaseCreation(None) creation.create_test_db() creation.destroy_test_db('foo_db') mock_method.assert_called_once_with(None)
def test_monkey_patch_test_db(self, mock_method): monkey_patch_test_db() creation = BaseDatabaseCreation(None) creation.create_test_db() creation.destroy_test_db('foo_db') mock_method.assert_called_once_with(None)
from django.db.utils import ConnectionHandler from mock import MagicMock import django.core.management.commands.loaddata from django.db.utils import NotSupportedError from django_mock_queries.mocks import monkey_patch_test_db import django_mock_queries.mocks # noinspection PyUnresolvedReferences from kive.settings import * monkey_patch_test_db() django.core.management.commands.loaddata.Command.handle = lambda *args, **kwargs: None original_mock_django_connection = django_mock_queries.mocks.mock_django_connection # Patch for django_mock_queries until a PR is merged. def patched_mock_django_connection(disabled_features=None): original_mock_django_connection(disabled_features=disabled_features) # noinspection PyUnusedLocal def patched_compiler(queryset, connection, using, **kwargs): result = MagicMock(name='mock_connection.ops.compiler()') # noinspection PyProtectedMember result.execute_sql.side_effect = NotSupportedError( "Mock database tried to execute SQL for {} model.".format( queryset.model._meta.object_name)) result.has_results.side_effect = result.execute_sql.side_effect return result
from django_mock_queries.mocks import monkey_patch_test_db from .common import * class Test(Common): pass monkey_patch_test_db()