• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-test-migrations-master
  • .github
    • FUNDING.yml
    • workflows
      • test.yml
      • misspell.yml
  • poetry.lock
  • Makefile
  • LICENSE
  • .editorconfig
  • CHANGELOG.md
  • django_test_migrations
    • sql.py
    • logic
      • __init__.py
      • migrations.py
    • exceptions.py
    • autonames.py
    • py.typed
    • migrator.py
    • __init__.py
    • types.py
    • plan.py
    • contrib
      • pytest_plugin.py
      • django_checks.py
      • __init__.py
      • unittest_case.py
  • setup.cfg
  • README.md
  • tests
    • test_contrib
      • test_pytest_plugin
        • test_pytest_plugin_direct.py
        • test_pytest_plugin_reverse.py
        • test_signals.py
        • __init__.py
        • test_plugin.py
      • test_django_checks
        • test_autonames_check.py
      • test_unittest_case
        • test_unittest_case.py
        • test_signals.py
        • __init__.py
    • test_plan
      • test_truncate_plan.py
      • test_all_migrations.py
      • conftest.py
    • test_migrator
      • test_migrator.py
    • test_autonames
      • test_autonames.py
    • test_logic
      • test_migrations.py
    • test_exceptions
      • test_migration_not_found.py
    • test_sql
      • test_flush_utils.py
      • test_drop_models_table.py
      • test_django_migrations_table_utils.py
    • conftest.py
  • pyproject.toml
  • django_test_app
    • manage.py
    • django_test_app
      • wsgi.py
      • urls.py
      • __init__.py
      • settings.py
    • main_app
      • apps.py
      • logic
        • pure
          • __init__.py
          • migrations.py
        • __init__.py
      • models.py
      • views.py
      • __init__.py
      • migrations
        • 0003_update_is_clean.py
        • 0002_someitem_is_clean.py
        • 0001_initial.py
        • 0004_auto_20191119_2125.py
        • 0005_auto_20200329_1118.py
        • __init__.py
  • .gitignore

from unittest import mock

from django.apps import apps
from django.core.management import call_command
from django.db import DEFAULT_DB_ALIAS
from django.db.models.signals import post_migrate, pre_migrate

from django_test_migrations.contrib.unittest_case import MigratorTestCase


class TestSignalMuting(MigratorTestCase):
    """Test that the `post_migrate` signal has been muted."""

    migrate_from = ('main_app', '0002_someitem_is_clean')
    migrate_to = ('main_app', '0001_initial')

    def test_pre_migrate_muted(self):
        """Ensure ``pre_migrate`` signal has been muted."""
        assert not pre_migrate.receivers
        assert self._pre_migrate_receivers

    def test_post_migrate_muted(self):
        """Ensure ``post_migrate`` signal has been muted."""
        assert not post_migrate.receivers
        assert self._post_migrate_receivers


class TestSignalConnectInTest(MigratorTestCase):
    """Ensure test ``pre_migrate`` or ``post_migrate`` receiver are called.

    Ensure that ``pre_migrate`` or ``post_migrate`` signals receivers
    connected directly in tests are called.

    """

    migrate_from = ('main_app', '0001_initial')
    migrate_to = ('main_app', '0002_someitem_is_clean')

    def tearDown(self):
        """Disconnect ``pre_migrate`` and ``post_migrate`` testing receivers."""
        pre_migrate.disconnect(
            self.pre_migrate_receiver_mock,
            sender=self.main_app_config,
        )
        post_migrate.disconnect(
            self.post_migrate_receiver_mock,
            sender=self.main_app_config,
        )

    def prepare(self):
        """Connect testing ``pre_migrate`` and ``post_migrate`` receivers."""
        self.pre_migrate_receiver_mock = mock.MagicMock()
        self.post_migrate_receiver_mock = mock.MagicMock()
        # ``old_apps`` is not real ``ProjectState`` instance, so we cannot use
        # it to get "original" main_app ``AppConfig`` instance needed to
        # connect signal receiver, that's the reason we are using
        # ``apps`` imported directly from ``django.apps``
        self.main_app_config = apps.get_app_config('main_app')
        pre_migrate.connect(
            self.pre_migrate_receiver_mock,
            sender=self.main_app_config,
        )
        post_migrate.connect(
            self.post_migrate_receiver_mock,
            sender=self.main_app_config,
        )

    def test_signal_receivers_added_in_tests(self):
        """Ensure migration signals receivers connected in tests are called."""
        verbosity = 0
        interactive = False
        # call `migrate` management command to trigger ``pre_migrate`` and
        # ``post_migrate`` signals
        call_command('migrate', verbosity=verbosity, interactive=interactive)

        common_kwargs = {
            'sender': self.main_app_config,
            'app_config': self.main_app_config,
            'apps': mock.ANY,  # we don't have any reference to this object
            'using': DEFAULT_DB_ALIAS,
            'verbosity': verbosity,
            'interactive': interactive,
            'plan': mock.ANY,  # not important for this test
        }
        self.pre_migrate_receiver_mock.assert_called_once_with(
            **common_kwargs,
            signal=pre_migrate,
        )
        self.post_migrate_receiver_mock.assert_called_once_with(
            **common_kwargs,
            signal=post_migrate,
        )