From 5561a308a3e31499f3f66de018f02a9e01ff41d7 Mon Sep 17 00:00:00 2001 From: David Wobrock Date: Fri, 31 May 2019 16:33:10 +0200 Subject: [PATCH] Add tests for ignore applied/unapplied migrations --- tests/unit/test_linter.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/unit/test_linter.py b/tests/unit/test_linter.py index a3f644b..b15b5bc 100644 --- a/tests/unit/test_linter.py +++ b/tests/unit/test_linter.py @@ -73,3 +73,17 @@ class LinterFunctionsTestCase(unittest.TestCase): linter = MigrationLinter() migrations = linter._gather_all_migrations() self.assertGreater(len(list(migrations)), 1) + + def test_ignore_unapplied_migrations(self): + linter = MigrationLinter(only_applied_migrations=True) + linter.migration_loader.applied_migrations = {("app_correct", "0002_foo")} + + self.assertTrue(linter.should_ignore_migration("app_correct", "0001_initial")) + self.assertFalse(linter.should_ignore_migration("app_correct", "0002_foo")) + + def test_ignore_applied_migrations(self): + linter = MigrationLinter(only_unapplied_migrations=True) + linter.migration_loader.applied_migrations = {("app_correct", "0002_foo")} + + self.assertFalse(linter.should_ignore_migration("app_correct", "0001_initial")) + self.assertTrue(linter.should_ignore_migration("app_correct", "0002_foo"))