Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Do some basic testing of the recorder

  • Loading branch information...
commit eb5e50215a630ed028eca2d85d4131e9e186377d 1 parent 8a1f017
@andrewgodwin andrewgodwin authored
Showing with 28 additions and 1 deletion.
  1. +28 −1 tests/migrations/tests.py
View
29 tests/migrations/tests.py
@@ -1,7 +1,8 @@
-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 @@ def test_load(self):
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(),
+ )
Please sign in to comment.
Something went wrong with that request. Please try again.