Added tests for the post_migrate signal. (18900e55) · Commits · Dom Sekotill / django

tests/migrate_signals/tests.py

+34 −28

Original line number	Diff line number	Diff line
		@@ -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,48 +52,51 @@ 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))
		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))
		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)