← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__known_graph.py

  • Committer: Martin Pool
  • Date: 2010-10-15 10:19:25 UTC
  • mto: This revision was merged to the branch mainline in revision 5503.
  • Revision ID: mbp@sourcefrog.net-20101015101925-qr2m0i5sv6ad0bs2
Superstitious argument quoting in makefile

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test__known_graph.py
1
 
# Copyright (C) 2009, 2010, 2011 Canonical Ltd
 
1
# Copyright (C) 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
20
20
 
21
21
from bzrlib import (
22
22
    errors,
 
23
    graph as _mod_graph,
23
24
    _known_graph_py,
24
25
    tests,
25
26
    )
26
27
from bzrlib.tests import test_graph
27
28
from bzrlib.revision import NULL_REVISION
28
 
from bzrlib.tests.scenarios import load_tests_apply_scenarios
29
 
 
30
 
 
31
 
def caching_scenarios():
 
29
 
 
30
 
 
31
def load_tests(standard_tests, module, loader):
 
32
    """Parameterize tests for all versions of groupcompress."""
32
33
    scenarios = [
33
34
        ('python', {'module': _known_graph_py, 'do_cache': True}),
34
35
    ]
 
36
    caching_scenarios = [
 
37
        ('python-nocache', {'module': _known_graph_py, 'do_cache': False}),
 
38
    ]
 
39
    suite = loader.suiteClass()
35
40
    if compiled_known_graph_feature.available():
36
41
        scenarios.append(('C', {'module': compiled_known_graph_feature.module,
37
42
                                'do_cache': True}))
38
 
    return scenarios
39
 
 
40
 
 
41
 
def non_caching_scenarios():
42
 
    scenarios = [
43
 
        ('python-nocache', {'module': _known_graph_py, 'do_cache': False}),
44
 
    ]
45
 
    if compiled_known_graph_feature.available():
46
 
        scenarios.append(
 
43
        caching_scenarios.append(
47
44
            ('C-nocache', {'module': compiled_known_graph_feature.module,
48
45
                           'do_cache': False}))
49
 
    return scenarios
50
 
 
51
 
 
52
 
load_tests = load_tests_apply_scenarios
 
46
    else:
 
47
        # the compiled module isn't available, so we add a failing test
 
48
        class FailWithoutFeature(tests.TestCase):
 
49
            def test_fail(self):
 
50
                self.requireFeature(compiled_known_graph_feature)
 
51
        suite.addTest(loader.loadTestsFromTestCase(FailWithoutFeature))
 
52
    # TestKnownGraphHeads needs to be permutated with and without caching.
 
53
    # All other TestKnownGraph tests only need to be tested across module
 
54
    heads_suite, other_suite = tests.split_suite_by_condition(
 
55
        standard_tests, tests.condition_isinstance(TestKnownGraphHeads))
 
56
    suite = tests.multiply_tests(other_suite, scenarios, suite)
 
57
    suite = tests.multiply_tests(heads_suite, scenarios + caching_scenarios,
 
58
                                 suite)
 
59
    return suite
53
60
 
54
61
 
55
62
compiled_known_graph_feature = tests.ModuleAvailableFeature(
56
 
    'bzrlib._known_graph_pyx')
 
63
                                    'bzrlib._known_graph_pyx')
57
64
 
58
65
 
59
66
#  a
68
75
 
69
76
class TestCaseWithKnownGraph(tests.TestCase):
70
77
 
71
 
    scenarios = caching_scenarios()
72
78
    module = None # Set by load_tests
73
79
 
74
80
    def make_known_graph(self, ancestry):
208
214
 
209
215
class TestKnownGraphHeads(TestCaseWithKnownGraph):
210
216
 
211
 
    scenarios = caching_scenarios() + non_caching_scenarios()
212
217
    do_cache = None # Set by load_tests
213
218
 
214
219
    def test_heads_null(self):