mirror of https://github.com/django/django.git
99 lines
3.9 KiB
Python
99 lines
3.9 KiB
Python
|
from django.test import TransactionTestCase
|
||
|
|
from django.db.migrations.graph import MigrationsGraph, CircularDependencyException
|
||
|
|
|
||
|
|
|
||
|
|
class GraphTests(TransactionTestCase):
|
||
|
|
"""
|
||
|
|
Tests the digraph structure.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def test_simple_graph(self):
|
||
|
|
"""
|
||
|
|
Tests a basic dependency graph:
|
||
|
|
|
||
|
|
app_a: 0001 <-- 0002 <--- 0003 <-- 0004
|
||
|
|
/
|
||
|
|
app_b: 0001 <-- 0002 <-/
|
||
|
|
"""
|
||
|
|
# Build graph
|
||
|
|
graph = MigrationsGraph()
|
||
|
|
graph.add_dependency(("app_a", "0004"), ("app_a", "0003"))
|
||
|
|
graph.add_dependency(("app_a", "0003"), ("app_a", "0002"))
|
||
|
|
graph.add_dependency(("app_a", "0002"), ("app_a", "0001"))
|
||
|
|
graph.add_dependency(("app_a", "0003"), ("app_b", "0002"))
|
||
|
|
graph.add_dependency(("app_b", "0002"), ("app_b", "0001"))
|
||
|
|
# Test root migration case
|
||
|
|
self.assertEqual(
|
||
|
|
graph.forwards_plan(("app_a", "0001")),
|
||
|
|
[('app_a', '0001')],
|
||
|
|
)
|
||
|
|
# Test branch B only
|
||
|
|
self.assertEqual(
|
||
|
|
graph.forwards_plan(("app_b", "0002")),
|
||
|
|
[("app_b", "0001"), ("app_b", "0002")],
|
||
|
|
)
|
||
|
|
# Test whole graph
|
||
|
|
self.assertEqual(
|
||
|
|
graph.forwards_plan(("app_a", "0004")),
|
||
|
|
[('app_b', '0001'), ('app_b', '0002'), ('app_a', '0001'), ('app_a', '0002'), ('app_a', '0003'), ('app_a', '0004')],
|
||
|
|
)
|
||
|
|
# Test reverse to b:0002
|
||
|
|
self.assertEqual(
|
||
|
|
graph.backwards_plan(("app_b", "0002")),
|
||
|
|
[('app_a', '0004'), ('app_a', '0003'), ('app_b', '0002')],
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_complex_graph(self):
|
||
|
|
"""
|
||
|
|
Tests a complex dependency graph:
|
||
|
|
|
||
|
|
app_a: 0001 <-- 0002 <--- 0003 <-- 0004
|
||
|
|
\ \ / /
|
||
|
|
app_b: 0001 <-\ 0002 <-X /
|
||
|
|
\ \ /
|
||
|
|
app_c: \ 0001 <-- 0002 <-
|
||
|
|
"""
|
||
|
|
# Build graph
|
||
|
|
graph = MigrationsGraph()
|
||
|
|
graph.add_dependency(("app_a", "0004"), ("app_a", "0003"))
|
||
|
|
graph.add_dependency(("app_a", "0003"), ("app_a", "0002"))
|
||
|
|
graph.add_dependency(("app_a", "0002"), ("app_a", "0001"))
|
||
|
|
graph.add_dependency(("app_a", "0003"), ("app_b", "0002"))
|
||
|
|
graph.add_dependency(("app_b", "0002"), ("app_b", "0001"))
|
||
|
|
graph.add_dependency(("app_a", "0004"), ("app_c", "0002"))
|
||
|
|
graph.add_dependency(("app_c", "0002"), ("app_c", "0001"))
|
||
|
|
graph.add_dependency(("app_c", "0001"), ("app_b", "0001"))
|
||
|
|
graph.add_dependency(("app_c", "0002"), ("app_a", "0002"))
|
||
|
|
# Test branch C only
|
||
|
|
self.assertEqual(
|
||
|
|
graph.forwards_plan(("app_c", "0002")),
|
||
|
|
[('app_b', '0001'), ('app_c', '0001'), ('app_a', '0001'), ('app_a', '0002'), ('app_c', '0002')],
|
||
|
|
)
|
||
|
|
# Test whole graph
|
||
|
|
self.assertEqual(
|
||
|
|
graph.forwards_plan(("app_a", "0004")),
|
||
|
|
[('app_b', '0001'), ('app_c', '0001'), ('app_a', '0001'), ('app_a', '0002'), ('app_c', '0002'), ('app_b', '0002'), ('app_a', '0003'), ('app_a', '0004')],
|
||
|
|
)
|
||
|
|
# Test reverse to b:0001
|
||
|
|
self.assertEqual(
|
||
|
|
graph.backwards_plan(("app_b", "0001")),
|
||
|
|
[('app_a', '0004'), ('app_c', '0002'), ('app_c', '0001'), ('app_a', '0003'), ('app_b', '0002'), ('app_b', '0001')],
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_circular_graph(self):
|
||
|
|
"""
|
||
|
|
Tests a circular dependency graph.
|
||
|
|
"""
|
||
|
|
# Build graph
|
||
|
|
graph = MigrationsGraph()
|
||
|
|
graph.add_dependency(("app_a", "0003"), ("app_a", "0002"))
|
||
|
|
graph.add_dependency(("app_a", "0002"), ("app_a", "0001"))
|
||
|
|
graph.add_dependency(("app_a", "0001"), ("app_b", "0002"))
|
||
|
|
graph.add_dependency(("app_b", "0002"), ("app_b", "0001"))
|
||
|
|
graph.add_dependency(("app_b", "0001"), ("app_a", "0003"))
|
||
|
|
# Test whole graph
|
||
|
|
self.assertRaises(
|
||
|
|
CircularDependencyException,
|
||
|
|
graph.forwards_plan, ("app_a", "0003"),
|
||
|
|
)
|