Do some basic testing of the recorder (eb5e5021) · Commits · Dom Sekotill / django

tests/migrations/tests.py

+28 −1

Original line number	Diff line number	Diff line
		from django.test import TransactionTestCase
		from django.test import TransactionTestCase, TestCase
		from django.db import connection
		from django.db.migrations.graph import MigrationGraph, CircularDependencyError
		from django.db.migrations.loader import MigrationLoader
		from django.db.migrations.recorder import MigrationRecorder


		class GraphTests(TransactionTestCase):
		@@ -133,3 +134,29 @@ class LoaderTests(TransactionTestCase):
		graph.forwards_plan(("migrations", "0002_second")),
		[("migrations", "0001_initial"), ("migrations", "0002_second")],
		)


		class RecorderTests(TestCase):
		"""
		Tests the disk and database loader.
		"""

		def test_apply(self):
		"""
		Tests marking migrations as applied/unapplied.
		"""
		recorder = MigrationRecorder(connection)
		self.assertEqual(
		recorder.applied_migrations(),
		set(),
		)
		recorder.record_applied("myapp", "0432_ponies")
		self.assertEqual(
		recorder.applied_migrations(),
		set([("myapp", "0432_ponies")]),
		)
		recorder.record_unapplied("myapp", "0432_ponies")
		self.assertEqual(
		recorder.applied_migrations(),
		set(),
		)