1 files changed, 145 insertions, 0 deletions
diff --git a/bzrlib/tests/per_workingtree/__init__.py b/bzrlib/tests/per_workingtree/__init__.py
new file mode 100644
index 0000000..4dc2612
--- /dev/null
+++ b/bzrlib/tests/per_workingtree/__init__.py
@@ -0,0 +1,145 @@
+# Copyright (C) 2006-2011 Canonical Ltd
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+
+
+"""WorkingTree implementation tests for bzr.
+
+This test the conformance of all the workingtre variations to the expected API.
+Specific tests for individual formats are in the tests/test_workingtree file
+rather than in tests/per_workingtree/*.py.
+"""
+
+from bzrlib import (
+    branchbuilder,
+    tests,
+    workingtree,
+    )
+from bzrlib.tests import per_controldir
+
+
+def make_scenarios(transport_server, transport_readonly_server, formats):
+    result = []
+    for workingtree_format in formats:
+        result.append((workingtree_format.__class__.__name__,
+                       make_scenario(transport_server,
+                                     transport_readonly_server,
+                                     workingtree_format)))
+    return result
+
+
+def make_scenario(transport_server, transport_readonly_server,
+                  workingtree_format):
+    return {
+        "transport_server": transport_server,
+        "transport_readonly_server": transport_readonly_server,
+        "bzrdir_format": workingtree_format._matchingbzrdir,
+        "workingtree_format": workingtree_format,
+        }
+
+
+def wt_scenarios():
+    """Returns the scenarios for all registered working trees.
+
+    This can used by plugins that want to define tests against these working
+    trees.
+    """
+    scenarios = make_scenarios(
+        tests.default_transport,
+        # None here will cause a readonly decorator to be created
+        # by the TestCaseWithTransport.get_readonly_transport method.
+        None,
+        workingtree.format_registry._get_all()
+        )
+    return scenarios
+
+
+class TestCaseWithWorkingTree(per_controldir.TestCaseWithControlDir):
+
+    def make_branch_and_tree(self, relpath, format=None):
+        made_control = self.make_bzrdir(relpath, format=format)
+        made_control.create_repository()
+        made_control.create_branch()
+        return self.workingtree_format.initialize(made_control)
+
+    def make_branch_builder(self, relpath, format=None):
+        if format is None:
+            format = self.workingtree_format.get_controldir_for_branch()
+        builder = branchbuilder.BranchBuilder(self.get_transport(relpath),
+                                              format=format)
+        return builder
+
+
+def load_tests(standard_tests, module, loader):
+    test_names = [
+        'add_reference',
+        'add',
+        'annotate_iter',
+        'basis_inventory',
+        'basis_tree',
+        'break_lock',
+        'changes_from',
+        'check',
+        'check_state',
+        'content_filters',
+        'commit',
+        'eol_conversion',
+        'executable',
+        'flush',
+        'get_file_mtime',
+        'get_parent_ids',
+        'inv',
+        'is_control_filename',
+        'is_ignored',
+        'locking',
+        'merge_from_branch',
+        'mkdir',
+        'move',
+        'nested_specifics',
+        'parents',
+        'paths2ids',
+        'pull',
+        'put_file',
+        'readonly',
+        'read_working_inventory',
+        'remove',
+        'rename_one',
+        'revision_tree',
+        'set_root_id',
+        'smart_add',
+        'symlinks',
+        'uncommit',
+        'unversion',
+        'views',
+        'walkdirs',
+        'workingtree',
+        ]
+    test_workingtree_implementations = [
+        'bzrlib.tests.per_workingtree.test_' + name for
+        name in test_names]
+
+    scenarios = wt_scenarios()
+
+    # add the tests for the sub modules
+    return tests.multiply_tests(
+        loader.loadTestsFromModuleNames(test_workingtree_implementations),
+        scenarios, standard_tests)
+
+
+class TestWtScenarios(tests.TestCase):
+
+    def test_protect_wt_scenarios(self):
+        # Just make sure we don't accidentally delete the helper again
+        scenarios = wt_scenarios()