123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208 |
- """Tests for dulwich.graph."""
- from dulwich.graph import WorkList, _find_lcas, can_fast_forward
- from dulwich.repo import MemoryRepo
- from dulwich.tests.utils import make_commit
- from . import TestCase
- class FindMergeBaseTests(TestCase):
- @staticmethod
- def run_test(dag, inputs):
- def lookup_parents(commit_id):
- return dag[commit_id]
- def lookup_stamp(commit_id):
-
-
- return 100
- c1 = inputs[0]
- c2s = inputs[1:]
- return set(_find_lcas(lookup_parents, c1, c2s, lookup_stamp))
- def test_multiple_lca(self):
-
- graph = {
- "5": ["1", "2"],
- "4": ["3", "1"],
- "3": ["2"],
- "2": ["0"],
- "1": [],
- "0": [],
- }
- self.assertEqual(self.run_test(graph, ["4", "5"]), {"1", "2"})
- def test_no_common_ancestor(self):
-
- graph = {
- "4": ["2"],
- "3": ["1"],
- "2": [],
- "1": ["0"],
- "0": [],
- }
- self.assertEqual(self.run_test(graph, ["4", "3"]), set())
- def test_ancestor(self):
-
- graph = {
- "G": ["D", "F"],
- "F": ["E"],
- "D": ["C"],
- "C": ["B"],
- "E": ["B"],
- "B": ["A"],
- "A": [],
- }
- self.assertEqual(self.run_test(graph, ["D", "C"]), {"C"})
- def test_direct_parent(self):
-
- graph = {
- "G": ["D", "F"],
- "F": ["E"],
- "D": ["C"],
- "C": ["B"],
- "E": ["B"],
- "B": ["A"],
- "A": [],
- }
- self.assertEqual(self.run_test(graph, ["G", "D"]), {"D"})
- def test_another_crossover(self):
-
- graph = {
- "G": ["D", "F"],
- "F": ["E", "C"],
- "D": ["C", "E"],
- "C": ["B"],
- "E": ["B"],
- "B": ["A"],
- "A": [],
- }
- self.assertEqual(self.run_test(graph, ["D", "F"]), {"E", "C"})
- def test_three_way_merge_lca(self):
-
- graph = {
- "C": ["C1"],
- "C1": ["C2"],
- "C2": ["C3"],
- "C3": ["C4"],
- "C4": ["2"],
- "B": ["B1"],
- "B1": ["B2"],
- "B2": ["B3"],
- "B3": ["1"],
- "A": ["A1"],
- "A1": ["A2"],
- "A2": ["A3"],
- "A3": ["1"],
- "1": ["2"],
- "2": [],
- }
-
-
- self.assertEqual(self.run_test(graph, ["A", "B", "C"]), {"1"})
- def test_octopus(self):
-
-
-
- graph = {
- "C": ["C1"],
- "C1": ["C2"],
- "C2": ["C3"],
- "C3": ["C4"],
- "C4": ["2"],
- "B": ["B1"],
- "B1": ["B2"],
- "B2": ["B3"],
- "B3": ["1"],
- "A": ["A1"],
- "A1": ["A2"],
- "A2": ["A3"],
- "A3": ["1"],
- "1": ["2"],
- "2": [],
- }
- def lookup_parents(cid):
- return graph[cid]
- def lookup_stamp(commit_id):
-
-
- return 100
- lcas = ["A"]
- others = ["B", "C"]
- for cmt in others:
- next_lcas = []
- for ca in lcas:
- res = _find_lcas(lookup_parents, cmt, [ca], lookup_stamp)
- next_lcas.extend(res)
- lcas = next_lcas[:]
- self.assertEqual(set(lcas), {"2"})
- class CanFastForwardTests(TestCase):
- def test_ff(self):
- r = MemoryRepo()
- base = make_commit()
- c1 = make_commit(parents=[base.id])
- c2 = make_commit(parents=[c1.id])
- r.object_store.add_objects([(base, None), (c1, None), (c2, None)])
- self.assertTrue(can_fast_forward(r, c1.id, c1.id))
- self.assertTrue(can_fast_forward(r, base.id, c1.id))
- self.assertTrue(can_fast_forward(r, c1.id, c2.id))
- self.assertFalse(can_fast_forward(r, c2.id, c1.id))
- def test_diverged(self):
- r = MemoryRepo()
- base = make_commit()
- c1 = make_commit(parents=[base.id])
- c2a = make_commit(parents=[c1.id], message=b"2a")
- c2b = make_commit(parents=[c1.id], message=b"2b")
- r.object_store.add_objects([(base, None), (c1, None), (c2a, None), (c2b, None)])
- self.assertTrue(can_fast_forward(r, c1.id, c2a.id))
- self.assertTrue(can_fast_forward(r, c1.id, c2b.id))
- self.assertFalse(can_fast_forward(r, c2a.id, c2b.id))
- self.assertFalse(can_fast_forward(r, c2b.id, c2a.id))
- class WorkListTest(TestCase):
- def test_WorkList(self):
-
-
-
- wlst = WorkList()
- wlst.add((100, "Test Value 1"))
- wlst.add((50, "Test Value 2"))
- wlst.add((200, "Test Value 3"))
- self.assertTrue(wlst.get() == (200, "Test Value 3"))
- self.assertTrue(wlst.get() == (100, "Test Value 1"))
- wlst.add((150, "Test Value 4"))
- self.assertTrue(wlst.get() == (150, "Test Value 4"))
- self.assertTrue(wlst.get() == (50, "Test Value 2"))
|