~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__chk_map.py

  • Committer: Martin Pool
  • Date: 2009-11-16 02:26:32 UTC
  • mto: This revision was merged to the branch mainline in revision 4880.
  • Revision ID: mbp@sourcefrog.net-20091116022632-261trs2osy8oupe3
Convert version-info to use TextUIOutputStream

Show diffs side-by-side

added added

removed removed

Lines of Context:
25
25
 
26
26
 
27
27
def load_tests(standard_tests, module, loader):
28
 
    suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
29
 
        'bzrlib._chk_map_py', 'bzrlib._chk_map_pyx')
 
28
    # parameterize all tests in this module
 
29
    suite = loader.suiteClass()
 
30
    import bzrlib._chk_map_py as py_module
 
31
    scenarios = [('python', {'module': py_module})]
 
32
    if CompiledChkMapFeature.available():
 
33
        import bzrlib._chk_map_pyx as c_module
 
34
        scenarios.append(('C', {'module': c_module}))
 
35
    else:
 
36
        # the compiled module isn't available, so we add a failing test
 
37
        class FailWithoutFeature(tests.TestCase):
 
38
            def test_fail(self):
 
39
                self.requireFeature(CompiledChkMapFeature)
 
40
        suite.addTest(loader.loadTestsFromTestCase(FailWithoutFeature))
 
41
    tests.multiply_tests(standard_tests, scenarios, suite)
30
42
    return suite
31
43
 
32
44
 
 
45
class _CompiledChkMapFeature(tests.Feature):
 
46
 
 
47
    def _probe(self):
 
48
        try:
 
49
            import bzrlib._chk_map_pyx
 
50
        except ImportError:
 
51
            return False
 
52
        return True
 
53
 
 
54
    def feature_name(self):
 
55
        return 'bzrlib._chk_map_pyx'
 
56
 
 
57
CompiledChkMapFeature = _CompiledChkMapFeature()
 
58
 
 
59
 
33
60
class TestSearchKeys(tests.TestCase):
34
61
 
35
62
    module = None # Filled in by test parameterization