← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_graph.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-12-10 21:06:38 UTC
  • mfrom: (3835.1.18 remote-stacking-graph)
  • Revision ID: pqm@pqm.ubuntu.com-20081210210638-e569azc2bomqq5qx
(abentley) Move RemoteRepository caching to CachingParentsProvider

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_graph.py
1413
1413
        self.assertEqual(['a', 'b'], sorted(self.inst_pp.calls))
1414
1414
 
1415
1415
 
 
1416
class TestCachingParentsProviderExtras(tests.TestCaseWithTransport):
 
1417
    """Test the behaviour when parents are provided that were not requested."""
 
1418
 
 
1419
    def setUp(self):
 
1420
        super(TestCachingParentsProviderExtras, self).setUp()
 
1421
        class ExtraParentsProvider(object):
 
1422
 
 
1423
            def get_parent_map(self, keys):
 
1424
                return {'rev1': [], 'rev2': ['rev1',]}
 
1425
 
 
1426
        self.inst_pp = InstrumentedParentsProvider(ExtraParentsProvider())
 
1427
        self.caching_pp = _mod_graph.CachingParentsProvider(
 
1428
            get_parent_map=self.inst_pp.get_parent_map)
 
1429
 
 
1430
    def test_uncached(self):
 
1431
        self.caching_pp.disable_cache()
 
1432
        self.assertEqual({'rev1': []},
 
1433
                         self.caching_pp.get_parent_map(['rev1']))
 
1434
        self.assertEqual(['rev1'], self.inst_pp.calls)
 
1435
        self.assertIs(None, self.caching_pp._cache)
 
1436
 
 
1437
    def test_cache_initially_empty(self):
 
1438
        self.assertEqual({}, self.caching_pp._cache)
 
1439
 
 
1440
    def test_cached(self):
 
1441
        self.assertEqual({'rev1': []},
 
1442
                         self.caching_pp.get_parent_map(['rev1']))
 
1443
        self.assertEqual(['rev1'], self.inst_pp.calls)
 
1444
        self.assertEqual({'rev1': [], 'rev2': ['rev1']},
 
1445
                         self.caching_pp._cache)
 
1446
        self.assertEqual({'rev1': []},
 
1447
                          self.caching_pp.get_parent_map(['rev1']))
 
1448
        self.assertEqual(['rev1'], self.inst_pp.calls)
 
1449
 
 
1450
    def test_disable_cache_clears_cache(self):
 
1451
        # Put something in the cache
 
1452
        self.caching_pp.get_parent_map(['rev1'])
 
1453
        self.assertEqual(2, len(self.caching_pp._cache))
 
1454
        self.caching_pp.disable_cache()
 
1455
        self.assertIs(None, self.caching_pp._cache)
 
1456
 
 
1457
    def test_cache_misses(self):
 
1458
        self.caching_pp.get_parent_map(['rev3'])
 
1459
        self.caching_pp.get_parent_map(['rev3'])
 
1460
        self.assertEqual(['rev3'], self.inst_pp.calls)
 
1461
 
 
1462
    def test_no_cache_misses(self):
 
1463
        self.caching_pp.enable_cache(cache_misses=False)
 
1464
        self.caching_pp.get_parent_map(['rev3'])
 
1465
        self.caching_pp.get_parent_map(['rev3'])
 
1466
        self.assertEqual(['rev3', 'rev3'], self.inst_pp.calls)
 
1467
 
 
1468
    def test_cache_extras(self):
 
1469
        self.assertEqual({}, self.caching_pp.get_parent_map(['rev3']))
 
1470
        self.assertEqual({'rev2': ['rev1']},
 
1471
                         self.caching_pp.get_parent_map(['rev2']))
 
1472
        self.assertEqual(['rev3'], self.inst_pp.calls)
 
1473
 
 
1474
 
1416
1475
class TestCollapseLinearRegions(tests.TestCase):
1417
1476
 
1418
1477
    def assertCollapsed(self, collapsed, original):