24
24
rather than in tests/branch_implementations/*.py.
27
from bzrlib.repository import (_legacy_formats,
29
RepositoryTestProviderAdapter,
30
from bzrlib.repofmt import (
32
33
from bzrlib.tests import (
40
from bzrlib.tests.bzrdir_implementations.test_bzrdir import TestCaseWithBzrDir
41
from bzrlib.transport.memory import MemoryServer
44
class RepositoryTestProviderAdapter(TestScenarioApplier):
45
"""A tool to generate a suite testing multiple repository formats at once.
47
This is done by copying the test once for each transport and injecting
48
the transport_server, transport_readonly_server, and bzrdir_format and
49
repository_format classes into each copy. Each copy is also given a new id()
50
to make it easy to identify.
53
def __init__(self, transport_server, transport_readonly_server, formats,
54
vfs_transport_factory=None):
55
TestScenarioApplier.__init__(self)
56
self._transport_server = transport_server
57
self._transport_readonly_server = transport_readonly_server
58
self._vfs_transport_factory = vfs_transport_factory
59
self.scenarios = self.formats_to_scenarios(formats)
61
def formats_to_scenarios(self, formats):
62
"""Transform the input formats to a list of scenarios.
64
:param formats: A list of (repository_format, bzrdir_format).
67
for repository_format, bzrdir_format in formats:
68
scenario = (repository_format.__class__.__name__,
69
{"transport_server":self._transport_server,
70
"transport_readonly_server":self._transport_readonly_server,
71
"bzrdir_format":bzrdir_format,
72
"repository_format":repository_format,
74
# Only override the test's vfs_transport_factory if one was
75
# specified, otherwise just leave the default in place.
76
if self._vfs_transport_factory:
77
scenario[1]['vfs_transport_factory'] = self._vfs_transport_factory
78
result.append(scenario)
82
class TestCaseWithRepository(TestCaseWithBzrDir):
84
def make_repository(self, relpath, format=None):
86
# Create a repository of the type we are trying to test.
87
made_control = self.make_bzrdir(relpath)
88
repo = self.repository_format.initialize(made_control)
89
if getattr(self, "repository_to_test_repository", None):
90
repo = self.repository_to_test_repository(repo)
93
return super(TestCaseWithRepository, self).make_repository(
43
101
'bzrlib.tests.repository_implementations.test_break_lock',
44
102
'bzrlib.tests.repository_implementations.test_commit_builder',
45
103
'bzrlib.tests.repository_implementations.test_fileid_involved',
104
'bzrlib.tests.repository_implementations.test_iter_reverse_revision_history',
46
105
'bzrlib.tests.repository_implementations.test_reconcile',
47
106
'bzrlib.tests.repository_implementations.test_repository',
48
107
'bzrlib.tests.repository_implementations.test_revision',
108
'bzrlib.tests.repository_implementations.test_statistics',
111
from bzrlib.smart.server import (
112
SmartTCPServer_for_testing,
113
ReadonlySmartTCPServer_for_testing,
115
from bzrlib.remote import RemoteBzrDirFormat, RemoteRepositoryFormat
117
registry = repository.format_registry
118
all_formats = [registry.get(k) for k in registry.keys()]
119
all_formats.extend(weaverepo._legacy_formats)
50
120
adapter = RepositoryTestProviderAdapter(
51
121
default_transport,
52
122
# None here will cause a readonly decorator to be created
53
123
# by the TestCaseWithTransport.get_readonly_transport method.
55
[(format, format._matchingbzrdir) for format in
56
RepositoryFormat._formats.values() + _legacy_formats])
125
[(format, format._matchingbzrdir) for format in all_formats])
57
126
loader = TestLoader()
58
127
adapt_modules(test_repository_implementations, adapter, loader, result)
129
adapt_to_smart_server = RepositoryTestProviderAdapter(
130
SmartTCPServer_for_testing,
131
ReadonlySmartTCPServer_for_testing,
132
[(RemoteRepositoryFormat(), RemoteBzrDirFormat())],
135
adapt_modules(test_repository_implementations,
136
adapt_to_smart_server,