From 9cbf7f25efa6ba3c4419b46fa11977e8b967aa26 Mon Sep 17 00:00:00 2001
From: Andrew Godwin <andrew@aeracode.org>
Date: Thu, 29 May 2014 16:43:49 -0700
Subject: [PATCH] Fix additional test failures caused by migration pollution

---
 tests/migrations/test_loader.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/migrations/test_loader.py b/tests/migrations/test_loader.py
index 34bc340495..01ead5e8ea 100644
--- a/tests/migrations/test_loader.py
+++ b/tests/migrations/test_loader.py
@@ -18,23 +18,23 @@ class RecorderTests(TestCase):
         """
         recorder = MigrationRecorder(connection)
         self.assertEqual(
-            recorder.applied_migrations(),
+            set((x, y) for (x, y) in recorder.applied_migrations() if x == "myapp"),
             set(),
         )
         recorder.record_applied("myapp", "0432_ponies")
         self.assertEqual(
-            recorder.applied_migrations(),
+            set((x, y) for (x, y) in recorder.applied_migrations() if x == "myapp"),
             set([("myapp", "0432_ponies")]),
         )
         # That should not affect records of another database
         recorder_other = MigrationRecorder(connections['other'])
         self.assertEqual(
-            recorder_other.applied_migrations(),
+            set((x, y) for (x, y) in recorder_other.applied_migrations() if x == "myapp"),
             set(),
         )
         recorder.record_unapplied("myapp", "0432_ponies")
         self.assertEqual(
-            recorder.applied_migrations(),
+            set((x, y) for (x, y) in recorder.applied_migrations() if x == "myapp"),
             set(),
         )
 
@@ -136,14 +136,14 @@ class LoaderTests(TestCase):
         recorder = MigrationRecorder(connection)
         # Loading with nothing applied should just give us the one node
         self.assertEqual(
-            len(migration_loader.graph.nodes),
+            len([x for x in migration_loader.graph.nodes if x[0] == "migrations"]),
             1,
         )
         # However, fake-apply one migration and it should now use the old two
         recorder.record_applied("migrations", "0001_initial")
         migration_loader.build_graph()
         self.assertEqual(
-            len(migration_loader.graph.nodes),
+            len([x for x in migration_loader.graph.nodes if x[0] == "migrations"]),
             2,
         )
         recorder.flush()