← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__rio.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-02-11 04:02:41 UTC
  • mfrom: (5017.2.2 tariff)
  • Revision ID: pqm@pqm.ubuntu.com-20100211040241-w6n021dz0uus341n
(mbp) add import-tariff tests

Show diffs side-by-side

added added

removed removed

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