diff options
Diffstat (limited to 'bzrlib/plugins/weave_fmt/test_repository.py')
| -rw-r--r-- | bzrlib/plugins/weave_fmt/test_repository.py | 331 |
1 files changed, 331 insertions, 0 deletions
diff --git a/bzrlib/plugins/weave_fmt/test_repository.py b/bzrlib/plugins/weave_fmt/test_repository.py new file mode 100644 index 0000000..7453296 --- /dev/null +++ b/bzrlib/plugins/weave_fmt/test_repository.py @@ -0,0 +1,331 @@ +# Copyright (C) 2006-2010 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 + +"""Tests for weave repositories. + +For interface tests see tests/per_repository/*.py. + +""" + +from __future__ import absolute_import + +from cStringIO import StringIO +from stat import S_ISDIR +import sys + +from bzrlib.bzrdir import ( + BzrDirMetaFormat1, + ) +from bzrlib.errors import ( + IllegalPath, + NoSuchFile, + ) +from bzrlib.repository import ( + InterRepository, + Repository, + ) +from bzrlib.serializer import ( + format_registry as serializer_format_registry, + ) +from bzrlib.tests import ( + TestCase, + TestCaseWithTransport, + ) + +from bzrlib.plugins.weave_fmt import xml4 +from bzrlib.plugins.weave_fmt.bzrdir import ( + BzrDirFormat6, + ) +from bzrlib.plugins.weave_fmt.repository import ( + InterWeaveRepo, + RepositoryFormat4, + RepositoryFormat5, + RepositoryFormat6, + RepositoryFormat7, + ) + + +class TestFormat6(TestCaseWithTransport): + + def test_attribute__fetch_order(self): + """Weaves need topological data insertion.""" + control = BzrDirFormat6().initialize(self.get_url()) + repo = RepositoryFormat6().initialize(control) + self.assertEqual('topological', repo._format._fetch_order) + + def test_attribute__fetch_uses_deltas(self): + """Weaves do not reuse deltas.""" + control = BzrDirFormat6().initialize(self.get_url()) + repo = RepositoryFormat6().initialize(control) + self.assertEqual(False, repo._format._fetch_uses_deltas) + + def test_attribute__fetch_reconcile(self): + """Weave repositories need a reconcile after fetch.""" + control = BzrDirFormat6().initialize(self.get_url()) + repo = RepositoryFormat6().initialize(control) + self.assertEqual(True, repo._format._fetch_reconcile) + + def test_no_ancestry_weave(self): + control = BzrDirFormat6().initialize(self.get_url()) + repo = RepositoryFormat6().initialize(control) + # We no longer need to create the ancestry.weave file + # since it is *never* used. + self.assertRaises(NoSuchFile, + control.transport.get, + 'ancestry.weave') + + def test_supports_external_lookups(self): + control = BzrDirFormat6().initialize(self.get_url()) + repo = RepositoryFormat6().initialize(control) + self.assertFalse(repo._format.supports_external_lookups) + + + + +class TestFormat7(TestCaseWithTransport): + + def test_attribute__fetch_order(self): + """Weaves need topological data insertion.""" + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control) + self.assertEqual('topological', repo._format._fetch_order) + + def test_attribute__fetch_uses_deltas(self): + """Weaves do not reuse deltas.""" + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control) + self.assertEqual(False, repo._format._fetch_uses_deltas) + + def test_attribute__fetch_reconcile(self): + """Weave repositories need a reconcile after fetch.""" + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control) + self.assertEqual(True, repo._format._fetch_reconcile) + + def test_disk_layout(self): + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control) + # in case of side effects of locking. + repo.lock_write() + repo.unlock() + # we want: + # format 'Bazaar-NG Repository format 7' + # lock '' + # inventory.weave == empty_weave + # empty revision-store directory + # empty weaves directory + t = control.get_repository_transport(None) + self.assertEqualDiff('Bazaar-NG Repository format 7', + t.get('format').read()) + self.assertTrue(S_ISDIR(t.stat('revision-store').st_mode)) + self.assertTrue(S_ISDIR(t.stat('weaves').st_mode)) + self.assertEqualDiff('# bzr weave file v5\n' + 'w\n' + 'W\n', + t.get('inventory.weave').read()) + # Creating a file with id Foo:Bar results in a non-escaped file name on + # disk. + control.create_branch() + tree = control.create_workingtree() + tree.add(['foo'], ['Foo:Bar'], ['file']) + tree.put_file_bytes_non_atomic('Foo:Bar', 'content\n') + try: + tree.commit('first post', rev_id='first') + except IllegalPath: + if sys.platform != 'win32': + raise + self.knownFailure('Foo:Bar cannot be used as a file-id on windows' + ' in repo format 7') + return + self.assertEqualDiff( + '# bzr weave file v5\n' + 'i\n' + '1 7fe70820e08a1aac0ef224d9c66ab66831cc4ab1\n' + 'n first\n' + '\n' + 'w\n' + '{ 0\n' + '. content\n' + '}\n' + 'W\n', + t.get('weaves/74/Foo%3ABar.weave').read()) + + def test_shared_disk_layout(self): + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control, shared=True) + # we want: + # format 'Bazaar-NG Repository format 7' + # inventory.weave == empty_weave + # empty revision-store directory + # empty weaves directory + # a 'shared-storage' marker file. + # lock is not present when unlocked + t = control.get_repository_transport(None) + self.assertEqualDiff('Bazaar-NG Repository format 7', + t.get('format').read()) + self.assertEqualDiff('', t.get('shared-storage').read()) + self.assertTrue(S_ISDIR(t.stat('revision-store').st_mode)) + self.assertTrue(S_ISDIR(t.stat('weaves').st_mode)) + self.assertEqualDiff('# bzr weave file v5\n' + 'w\n' + 'W\n', + t.get('inventory.weave').read()) + self.assertFalse(t.has('branch-lock')) + + def test_creates_lockdir(self): + """Make sure it appears to be controlled by a LockDir existence""" + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control, shared=True) + t = control.get_repository_transport(None) + # TODO: Should check there is a 'lock' toplevel directory, + # regardless of contents + self.assertFalse(t.has('lock/held/info')) + repo.lock_write() + try: + self.assertTrue(t.has('lock/held/info')) + finally: + # unlock so we don't get a warning about failing to do so + repo.unlock() + + def test_uses_lockdir(self): + """repo format 7 actually locks on lockdir""" + base_url = self.get_url() + control = BzrDirMetaFormat1().initialize(base_url) + repo = RepositoryFormat7().initialize(control, shared=True) + t = control.get_repository_transport(None) + repo.lock_write() + repo.unlock() + del repo + # make sure the same lock is created by opening it + repo = Repository.open(base_url) + repo.lock_write() + self.assertTrue(t.has('lock/held/info')) + repo.unlock() + self.assertFalse(t.has('lock/held/info')) + + def test_shared_no_tree_disk_layout(self): + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control, shared=True) + repo.set_make_working_trees(False) + # we want: + # format 'Bazaar-NG Repository format 7' + # lock '' + # inventory.weave == empty_weave + # empty revision-store directory + # empty weaves directory + # a 'shared-storage' marker file. + t = control.get_repository_transport(None) + self.assertEqualDiff('Bazaar-NG Repository format 7', + t.get('format').read()) + ## self.assertEqualDiff('', t.get('lock').read()) + self.assertEqualDiff('', t.get('shared-storage').read()) + self.assertEqualDiff('', t.get('no-working-trees').read()) + repo.set_make_working_trees(True) + self.assertFalse(t.has('no-working-trees')) + self.assertTrue(S_ISDIR(t.stat('revision-store').st_mode)) + self.assertTrue(S_ISDIR(t.stat('weaves').st_mode)) + self.assertEqualDiff('# bzr weave file v5\n' + 'w\n' + 'W\n', + t.get('inventory.weave').read()) + + def test_supports_external_lookups(self): + control = BzrDirMetaFormat1().initialize(self.get_url()) + repo = RepositoryFormat7().initialize(control) + self.assertFalse(repo._format.supports_external_lookups) + + +class TestInterWeaveRepo(TestCaseWithTransport): + + def test_is_compatible_and_registered(self): + # InterWeaveRepo is compatible when either side + # is a format 5/6/7 branch + from bzrlib.repofmt import knitrepo + formats = [RepositoryFormat5(), + RepositoryFormat6(), + RepositoryFormat7()] + incompatible_formats = [RepositoryFormat4(), + knitrepo.RepositoryFormatKnit1(), + ] + repo_a = self.make_repository('a') + repo_b = self.make_repository('b') + is_compatible = InterWeaveRepo.is_compatible + for source in incompatible_formats: + # force incompatible left then right + repo_a._format = source + repo_b._format = formats[0] + self.assertFalse(is_compatible(repo_a, repo_b)) + self.assertFalse(is_compatible(repo_b, repo_a)) + for source in formats: + repo_a._format = source + for target in formats: + repo_b._format = target + self.assertTrue(is_compatible(repo_a, repo_b)) + self.assertEqual(InterWeaveRepo, + InterRepository.get(repo_a, repo_b).__class__) + + +_working_inventory_v4 = """<inventory file_id="TREE_ROOT"> +<entry file_id="bar-20050901064931-73b4b1138abc9cd2" kind="file" name="bar" parent_id="TREE_ROOT" /> +<entry file_id="foo-20050801201819-4139aa4a272f4250" kind="directory" name="foo" parent_id="TREE_ROOT" /> +<entry file_id="bar-20050824000535-6bc48cfad47ed134" kind="file" name="bar" parent_id="foo-20050801201819-4139aa4a272f4250" /> +</inventory>""" + + +_revision_v4 = """<revision committer="Martin Pool <mbp@sourcefrog.net>" + inventory_id="mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9" + inventory_sha1="e79c31c1deb64c163cf660fdedd476dd579ffd41" + revision_id="mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9" + timestamp="1125907235.212" + timezone="36000"> +<message>- start splitting code for xml (de)serialization away from objects + preparatory to supporting multiple formats by a single library +</message> +<parents> +<revision_ref revision_id="mbp@sourcefrog.net-20050905063503-43948f59fa127d92" revision_sha1="7bdf4cc8c5bdac739f8cf9b10b78cf4b68f915ff" /> +</parents> +</revision> +""" + + +class TestSerializer(TestCase): + """Test serializer""" + + def test_registry(self): + self.assertIs(xml4.serializer_v4, + serializer_format_registry.get('4')) + + def test_canned_inventory(self): + """Test unpacked a canned inventory v4 file.""" + inp = StringIO(_working_inventory_v4) + inv = xml4.serializer_v4.read_inventory(inp) + self.assertEqual(len(inv), 4) + self.assert_(inv.has_id('bar-20050901064931-73b4b1138abc9cd2')) + + def test_unpack_revision(self): + """Test unpacking a canned revision v4""" + inp = StringIO(_revision_v4) + rev = xml4.serializer_v4.read_revision(inp) + eq = self.assertEqual + eq(rev.committer, + "Martin Pool <mbp@sourcefrog.net>") + eq(rev.inventory_id, + "mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9") + eq(len(rev.parent_ids), 1) + eq(rev.parent_ids[0], + "mbp@sourcefrog.net-20050905063503-43948f59fa127d92") + + |