diff --git a/tests/migrations/test_graph.py b/tests/migrations/test_graph.py index b6a7c62561..a72619eea5 100644 --- a/tests/migrations/test_graph.py +++ b/tests/migrations/test_graph.py @@ -133,3 +133,16 @@ class GraphTests(TestCase): CircularDependencyError, graph.forwards_plan, ("app_a", "0003"), ) + + def test_plan_invalid_node(self): + """ + Tests for forwards/backwards_plan of nonexistent node. + """ + graph = MigrationGraph() + message = "Node ('app_b', '0001') not a valid node" + + with self.assertRaisesMessage(ValueError, message): + graph.forwards_plan(("app_b", "0001")) + + with self.assertRaisesMessage(ValueError, message): + graph.backwards_plan(("app_b", "0001"))