From be4baaefe2e8e7effa3c510b31746dbb41ec422a Mon Sep 17 00:00:00 2001 From: Raffaele Salmaso Date: Sat, 23 Aug 2014 11:37:25 +0200 Subject: [PATCH] Fixed #23352 -- Added tests for MigrationGraph.{forwards,backwards}_plan --- tests/migrations/test_graph.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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"))