From eb5e50215a630ed028eca2d85d4131e9e186377d Mon Sep 17 00:00:00 2001
From: Andrew Godwin <andrew@aeracode.org>
Date: Fri, 10 May 2013 16:18:19 +0100
Subject: [PATCH] Do some basic testing of the recorder

---
 tests/migrations/tests.py | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/tests/migrations/tests.py b/tests/migrations/tests.py
index 338e28aa567..9ef5e37b8f4 100644
--- a/tests/migrations/tests.py
+++ b/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 @@ 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(),
+        )