661
675
self.assertEqual((set(['e']), set(['f', 'g'])),
662
676
graph.find_difference('e', 'f'))
664
679
def test_stacked_parents_provider(self):
665
680
parents1 = _mod_graph.DictParentsProvider({'rev2': ['rev3']})
666
681
parents2 = _mod_graph.DictParentsProvider({'rev1': ['rev4']})
667
stacked = _mod_graph._StackedParentsProvider([parents1, parents2])
682
stacked = _mod_graph.StackedParentsProvider([parents1, parents2])
683
self.assertEqual({'rev1':['rev4'], 'rev2':['rev3']},
684
stacked.get_parent_map(['rev1', 'rev2']))
685
self.assertEqual({'rev2':['rev3'], 'rev1':['rev4']},
686
stacked.get_parent_map(['rev2', 'rev1']))
687
self.assertEqual({'rev2':['rev3']},
688
stacked.get_parent_map(['rev2', 'rev2']))
689
self.assertEqual({'rev1':['rev4']},
690
stacked.get_parent_map(['rev1', 'rev1']))
692
def test_stacked_parents_provider_overlapping(self):
693
# rev2 is availible in both providers.
697
parents1 = _mod_graph.DictParentsProvider({'rev2': ['rev1']})
698
parents2 = _mod_graph.DictParentsProvider({'rev2': ['rev1']})
699
stacked = _mod_graph.StackedParentsProvider([parents1, parents2])
700
self.assertEqual({'rev2': ['rev1']},
701
stacked.get_parent_map(['rev2']))
703
def test__stacked_parents_provider_deprecated(self):
704
parents1 = _mod_graph.DictParentsProvider({'rev2': ['rev3']})
705
parents2 = _mod_graph.DictParentsProvider({'rev1': ['rev4']})
706
stacked = self.applyDeprecated(deprecated_in((1, 16, 0)),
707
_mod_graph._StackedParentsProvider, [parents1, parents2])
668
708
self.assertEqual({'rev1':['rev4'], 'rev2':['rev3']},
669
709
stacked.get_parent_map(['rev1', 'rev2']))
670
710
self.assertEqual({'rev2':['rev3'], 'rev1':['rev4']},
1312
1394
self.assertFindDistance(6, graph, 'g', [('i', 8)])
1397
class TestFindMergeOrder(TestGraphBase):
1399
def assertMergeOrder(self, expected, graph, tip, base_revisions):
1400
self.assertEqual(expected, graph.find_merge_order(tip, base_revisions))
1402
def test_parents(self):
1403
graph = self.make_graph(ancestry_1)
1404
self.assertMergeOrder(['rev3', 'rev2b'], graph, 'rev4',
1406
self.assertMergeOrder(['rev3', 'rev2b'], graph, 'rev4',
1409
def test_ancestors(self):
1410
graph = self.make_graph(ancestry_1)
1411
self.assertMergeOrder(['rev1', 'rev2b'], graph, 'rev4',
1413
self.assertMergeOrder(['rev1', 'rev2b'], graph, 'rev4',
1416
def test_shortcut_one_ancestor(self):
1417
# When we have enough info, we can stop searching
1418
graph = self.make_breaking_graph(ancestry_1, ['rev3', 'rev2b', 'rev4'])
1419
# Single ancestors shortcut right away
1420
self.assertMergeOrder(['rev3'], graph, 'rev4', ['rev3'])
1422
def test_shortcut_after_one_ancestor(self):
1423
graph = self.make_breaking_graph(ancestry_1, ['rev2a', 'rev2b'])
1424
self.assertMergeOrder(['rev3', 'rev1'], graph, 'rev4', ['rev1', 'rev3'])
1315
1427
class TestCachingParentsProvider(tests.TestCase):
1428
"""These tests run with:
1430
self.inst_pp, a recording parents provider with a graph of a->b, and b is a
1432
self.caching_pp, a CachingParentsProvider layered on inst_pp.
1317
1435
def setUp(self):
1318
1436
super(TestCachingParentsProvider, self).setUp()
1354
1471
# Use sorted because we don't care about the order, just that each is
1355
1472
# only present 1 time.
1356
1473
self.assertEqual(['a', 'b'], sorted(self.inst_pp.calls))
1475
def test_note_missing_key(self):
1476
"""After noting that a key is missing it is cached."""
1477
self.caching_pp.note_missing_key('b')
1478
self.assertEqual({}, self.caching_pp.get_parent_map(['b']))
1479
self.assertEqual([], self.inst_pp.calls)
1480
self.assertEqual(set(['b']), self.caching_pp.missing_keys)
1483
class TestCachingParentsProviderExtras(tests.TestCaseWithTransport):
1484
"""Test the behaviour when parents are provided that were not requested."""
1487
super(TestCachingParentsProviderExtras, self).setUp()
1488
class ExtraParentsProvider(object):
1490
def get_parent_map(self, keys):
1491
return {'rev1': [], 'rev2': ['rev1',]}
1493
self.inst_pp = InstrumentedParentsProvider(ExtraParentsProvider())
1494
self.caching_pp = _mod_graph.CachingParentsProvider(
1495
get_parent_map=self.inst_pp.get_parent_map)
1497
def test_uncached(self):
1498
self.caching_pp.disable_cache()
1499
self.assertEqual({'rev1': []},
1500
self.caching_pp.get_parent_map(['rev1']))
1501
self.assertEqual(['rev1'], self.inst_pp.calls)
1502
self.assertIs(None, self.caching_pp._cache)
1504
def test_cache_initially_empty(self):
1505
self.assertEqual({}, self.caching_pp._cache)
1507
def test_cached(self):
1508
self.assertEqual({'rev1': []},
1509
self.caching_pp.get_parent_map(['rev1']))
1510
self.assertEqual(['rev1'], self.inst_pp.calls)
1511
self.assertEqual({'rev1': [], 'rev2': ['rev1']},
1512
self.caching_pp._cache)
1513
self.assertEqual({'rev1': []},
1514
self.caching_pp.get_parent_map(['rev1']))
1515
self.assertEqual(['rev1'], self.inst_pp.calls)
1517
def test_disable_cache_clears_cache(self):
1518
# Put something in the cache
1519
self.caching_pp.get_parent_map(['rev1'])
1520
self.assertEqual(2, len(self.caching_pp._cache))
1521
self.caching_pp.disable_cache()
1522
self.assertIs(None, self.caching_pp._cache)
1524
def test_enable_cache_raises(self):
1525
e = self.assertRaises(AssertionError, self.caching_pp.enable_cache)
1526
self.assertEqual('Cache enabled when already enabled.', str(e))
1528
def test_cache_misses(self):
1529
self.caching_pp.get_parent_map(['rev3'])
1530
self.caching_pp.get_parent_map(['rev3'])
1531
self.assertEqual(['rev3'], self.inst_pp.calls)
1533
def test_no_cache_misses(self):
1534
self.caching_pp.disable_cache()
1535
self.caching_pp.enable_cache(cache_misses=False)
1536
self.caching_pp.get_parent_map(['rev3'])
1537
self.caching_pp.get_parent_map(['rev3'])
1538
self.assertEqual(['rev3', 'rev3'], self.inst_pp.calls)
1540
def test_cache_extras(self):
1541
self.assertEqual({}, self.caching_pp.get_parent_map(['rev3']))
1542
self.assertEqual({'rev2': ['rev1']},
1543
self.caching_pp.get_parent_map(['rev2']))
1544
self.assertEqual(['rev3'], self.inst_pp.calls)
1547
class TestCollapseLinearRegions(tests.TestCase):
1549
def assertCollapsed(self, collapsed, original):
1550
self.assertEqual(collapsed,
1551
_mod_graph.collapse_linear_regions(original))
1553
def test_collapse_nothing(self):
1554
d = {1:[2, 3], 2:[], 3:[]}
1555
self.assertCollapsed(d, d)
1556
d = {1:[2], 2:[3, 4], 3:[5], 4:[5], 5:[]}
1557
self.assertCollapsed(d, d)
1559
def test_collapse_chain(self):
1560
# Any time we have a linear chain, we should be able to collapse
1561
d = {1:[2], 2:[3], 3:[4], 4:[5], 5:[]}
1562
self.assertCollapsed({1:[5], 5:[]}, d)
1563
d = {5:[4], 4:[3], 3:[2], 2:[1], 1:[]}
1564
self.assertCollapsed({5:[1], 1:[]}, d)
1565
d = {5:[3], 3:[4], 4:[1], 1:[2], 2:[]}
1566
self.assertCollapsed({5:[2], 2:[]}, d)
1568
def test_collapse_with_multiple_children(self):
1579
# 4 and 5 cannot be removed because 6 has 2 children
1580
# 2 and 3 cannot be removed because 1 has 2 parents
1581
d = {1:[2, 3], 2:[4], 4:[6], 3:[5], 5:[6], 6:[7], 7:[]}
1582
self.assertCollapsed(d, d)
1585
class TestPendingAncestryResultGetKeys(TestCaseWithMemoryTransport):
1586
"""Tests for bzrlib.graph.PendingAncestryResult."""
1588
def test_get_keys(self):
1589
builder = self.make_branch_builder('b')
1590
builder.start_series()
1591
builder.build_snapshot('rev-1', None, [
1592
('add', ('', 'root-id', 'directory', ''))])
1593
builder.build_snapshot('rev-2', ['rev-1'], [])
1594
builder.finish_series()
1595
repo = builder.get_branch().repository
1597
self.addCleanup(repo.unlock)
1598
result = _mod_graph.PendingAncestryResult(['rev-2'], repo)
1599
self.assertEqual(set(['rev-1', 'rev-2']), set(result.get_keys()))
1601
def test_get_keys_excludes_ghosts(self):
1602
builder = self.make_branch_builder('b')
1603
builder.start_series()
1604
builder.build_snapshot('rev-1', None, [
1605
('add', ('', 'root-id', 'directory', ''))])
1606
builder.build_snapshot('rev-2', ['rev-1', 'ghost'], [])
1607
builder.finish_series()
1608
repo = builder.get_branch().repository
1610
self.addCleanup(repo.unlock)
1611
result = _mod_graph.PendingAncestryResult(['rev-2'], repo)
1612
self.assertEqual(sorted(['rev-1', 'rev-2']), sorted(result.get_keys()))
1614
def test_get_keys_excludes_null(self):
1615
# Make a 'graph' with an iter_ancestry that returns NULL_REVISION
1616
# somewhere other than the last element, which can happen in real
1618
class StubGraph(object):
1619
def iter_ancestry(self, keys):
1620
return [(NULL_REVISION, ()), ('foo', (NULL_REVISION,))]
1621
result = _mod_graph.PendingAncestryResult(['rev-3'], None)
1622
result_keys = result._get_keys(StubGraph())
1623
# Only the non-null keys from the ancestry appear.
1624
self.assertEqual(set(['foo']), set(result_keys))
1627
class TestPendingAncestryResultRefine(TestGraphBase):
1629
def test_refine(self):
1630
# Used when pulling from a stacked repository, so test some revisions
1631
# being satisfied from the stacking branch.
1632
g = self.make_graph(
1633
{"tip":["mid"], "mid":["base"], "tag":["base"],
1634
"base":[NULL_REVISION], NULL_REVISION:[]})
1635
result = _mod_graph.PendingAncestryResult(['tip', 'tag'], None)
1636
result = result.refine(set(['tip']), set(['mid']))
1637
self.assertEqual(set(['mid', 'tag']), result.heads)
1638
result = result.refine(set(['mid', 'tag', 'base']),
1639
set([NULL_REVISION]))
1640
self.assertEqual(set([NULL_REVISION]), result.heads)
1641
self.assertTrue(result.is_empty())
1644
class TestSearchResultRefine(TestGraphBase):
1646
def test_refine(self):
1647
# Used when pulling from a stacked repository, so test some revisions
1648
# being satisfied from the stacking branch.
1649
g = self.make_graph(
1650
{"tip":["mid"], "mid":["base"], "tag":["base"],
1651
"base":[NULL_REVISION], NULL_REVISION:[]})
1652
result = _mod_graph.SearchResult(set(['tip', 'tag']),
1653
set([NULL_REVISION]), 4, set(['tip', 'mid', 'tag', 'base']))
1654
result = result.refine(set(['tip']), set(['mid']))
1655
recipe = result.get_recipe()
1656
# We should be starting from tag (original head) and mid (seen ref)
1657
self.assertEqual(set(['mid', 'tag']), recipe[1])
1658
# We should be stopping at NULL (original stop) and tip (seen head)
1659
self.assertEqual(set([NULL_REVISION, 'tip']), recipe[2])
1660
self.assertEqual(3, recipe[3])
1661
result = result.refine(set(['mid', 'tag', 'base']),
1662
set([NULL_REVISION]))
1663
recipe = result.get_recipe()
1664
# We should be starting from nothing (NULL was known as a cut point)
1665
self.assertEqual(set([]), recipe[1])
1666
# We should be stopping at NULL (original stop) and tip (seen head) and
1667
# tag (seen head) and mid(seen mid-point head). We could come back and
1668
# define this as not including mid, for minimal results, but it is
1669
# still 'correct' to include mid, and simpler/easier.
1670
self.assertEqual(set([NULL_REVISION, 'tip', 'tag', 'mid']), recipe[2])
1671
self.assertEqual(0, recipe[3])
1672
self.assertTrue(result.is_empty())