Added tests for the post_migrate signal.

1 files changed, 34 insertions, 28 deletions

diff --git a/tests/migrate_signals/tests.py b/tests/migrate_signals/tests.py
index cdfffe5de7..603980b740 100644
--- a/tests/migrate_signals/tests.py
+++ b/tests/migrate_signals/tests.py
@@ -11,10 +11,11 @@ MIGRATE_VERBOSITY = 1
 MIGRATE_INTERACTIVE = False
 
 
-class PreMigrateReceiver(object):
-    def __init__(self):
+class Receiver(object):
+    def __init__(self, signal):
         self.call_counter = 0
         self.call_args = None
+        signal.connect(self, sender=APP_CONFIG)
 
     def __call__(self, signal, sender, **kwargs):
         self.call_counter = self.call_counter + 1
@@ -27,9 +28,11 @@ class OneTimeReceiver(object):
     several databases and several times for some of them.
     """
 
-    def __init__(self):
+    def __init__(self, signal):
+        self.signal = signal
         self.call_counter = 0
         self.call_args = None
+        self.signal.connect(self, sender=APP_CONFIG)
 
     def __call__(self, signal, sender, **kwargs):
         # Although test runner calls migrate for several databases,
@@ -38,7 +41,7 @@ class OneTimeReceiver(object):
             self.call_counter = self.call_counter + 1
             self.call_args = kwargs
             # we need to test only one call of migrate
-            signals.pre_migrate.disconnect(pre_migrate_receiver, sender=APP_CONFIG)
+            self.signal.disconnect(self, sender=APP_CONFIG)
 
 
 # We connect receiver here and not in unit test code because we need to
@@ -49,49 +52,52 @@ class OneTimeReceiver(object):
 #   2. We connect receiver.
 #   3. Test runner calls migrate for create default database.
 #   4. Test runner execute our unit test code.
-pre_migrate_receiver = OneTimeReceiver()
-signals.pre_migrate.connect(pre_migrate_receiver, sender=APP_CONFIG)
+pre_migrate_receiver = OneTimeReceiver(signals.pre_migrate)
+post_migrate_receiver = OneTimeReceiver(signals.post_migrate)
 
 
 class MigrateSignalTests(TestCase):
 
     available_apps = ['migrate_signals']
 
-    def test_pre_migrate_call_time(self):
+    def test_call_time(self):
         self.assertEqual(pre_migrate_receiver.call_counter, 1)
+        self.assertEqual(post_migrate_receiver.call_counter, 1)
 
-    def test_pre_migrate_args(self):
-        r = PreMigrateReceiver()
-        signals.pre_migrate.connect(r, sender=APP_CONFIG)
+    def test_args(self):
+        pre_migrate_receiver = Receiver(signals.pre_migrate)
+        post_migrate_receiver = Receiver(signals.post_migrate)
         management.call_command(
             'migrate', database=MIGRATE_DATABASE, verbosity=MIGRATE_VERBOSITY,
             interactive=MIGRATE_INTERACTIVE, stdout=six.StringIO(),
         )
 
-        args = r.call_args
-        self.assertEqual(r.call_counter, 1)
-        self.assertEqual(set(args), set(PRE_MIGRATE_ARGS))
-        self.assertEqual(args['app_config'], APP_CONFIG)
-        self.assertEqual(args['verbosity'], MIGRATE_VERBOSITY)
-        self.assertEqual(args['interactive'], MIGRATE_INTERACTIVE)
-        self.assertEqual(args['using'], 'default')
+        for receiver in [pre_migrate_receiver, post_migrate_receiver]:
+            args = receiver.call_args
+            self.assertEqual(receiver.call_counter, 1)
+            self.assertEqual(set(args), set(SIGNAL_ARGS))
+            self.assertEqual(args['app_config'], APP_CONFIG)
+            self.assertEqual(args['verbosity'], MIGRATE_VERBOSITY)
+            self.assertEqual(args['interactive'], MIGRATE_INTERACTIVE)
+            self.assertEqual(args['using'], 'default')
 
     @override_settings(MIGRATION_MODULES={'migrate_signals': 'migrate_signals.custom_migrations'})
-    def test_pre_migrate_migrations_only(self):
+    def test_migrations_only(self):
         """
-        If all apps have migrations, pre_migrate should be sent.
+        If all apps have migrations, migration signals should be sent.
         """
-        r = PreMigrateReceiver()
-        signals.pre_migrate.connect(r, sender=APP_CONFIG)
+        pre_migrate_receiver = Receiver(signals.pre_migrate)
+        post_migrate_receiver = Receiver(signals.post_migrate)
         stdout = six.StringIO()
         management.call_command(
             'migrate', database=MIGRATE_DATABASE, verbosity=MIGRATE_VERBOSITY,
             interactive=MIGRATE_INTERACTIVE, stdout=stdout,
         )
-        args = r.call_args
-        self.assertEqual(r.call_counter, 1)
-        self.assertEqual(set(args), set(PRE_MIGRATE_ARGS))
-        self.assertEqual(args['app_config'], APP_CONFIG)
-        self.assertEqual(args['verbosity'], MIGRATE_VERBOSITY)
-        self.assertEqual(args['interactive'], MIGRATE_INTERACTIVE)
-        self.assertEqual(args['using'], 'default')
+        for receiver in [pre_migrate_receiver, post_migrate_receiver]:
+            args = receiver.call_args
+            self.assertEqual(receiver.call_counter, 1)
+            self.assertEqual(set(args), set(SIGNAL_ARGS))
+            self.assertEqual(args['app_config'], APP_CONFIG)
+            self.assertEqual(args['verbosity'], MIGRATE_VERBOSITY)
+            self.assertEqual(args['interactive'], MIGRATE_INTERACTIVE)
+            self.assertEqual(args['using'], 'default')