# (C) 2006 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""Tests for the Repository facility that are not interface tests.
For interface tests see tests/repository_implementations/*.py.
For concrete class tests see this file, and for storage formats tests
also see this file.
"""
from stat import *
from StringIO import StringIO
import bzrlib.bzrdir as bzrdir
import bzrlib.errors as errors
from bzrlib.errors import (NotBranchError,
NoSuchFile,
UnknownFormatError,
UnsupportedFormatError,
)
import bzrlib.repository as repository
from bzrlib.tests import TestCase, TestCaseWithTransport
from bzrlib.transport import get_transport
from bzrlib.transport.http import HttpServer
from bzrlib.transport.memory import MemoryServer
class TestDefaultFormat(TestCase):
def test_get_set_default_format(self):
old_format = repository.RepositoryFormat.get_default_format()
# default is None - we cannot create a Repository independently yet
self.assertTrue(isinstance(old_format, repository.RepositoryFormat7))
repository.RepositoryFormat.set_default_format(SampleRepositoryFormat())
# creating a repository should now create an instrumented dir.
try:
# the default branch format is used by the meta dir format
# which is not the default bzrdir format at this point
dir = bzrdir.BzrDirMetaFormat1().initialize('memory:/')
result = dir.create_repository()
self.assertEqual(result, 'A bzr repository dir')
finally:
repository.RepositoryFormat.set_default_format(old_format)
self.assertEqual(old_format, repository.RepositoryFormat.get_default_format())
class SampleRepositoryFormat(repository.RepositoryFormat):
"""A sample format
this format is initializable, unsupported to aid in testing the
open and open(unsupported=True) routines.
"""
def get_format_string(self):
"""See RepositoryFormat.get_format_string()."""
return "Sample .bzr repository format."
def initialize(self, a_bzrdir, shared=False):
"""Initialize a repository in a BzrDir"""
t = a_bzrdir.get_repository_transport(self)
t.put('format', StringIO(self.get_format_string()))
return 'A bzr repository dir'
def is_supported(self):
return False
def open(self, a_bzrdir, _found=False):
return "opened repository."
class TestRepositoryFormat(TestCaseWithTransport):
"""Tests for the Repository format detection used by the bzr meta dir facility.BzrBranchFormat facility."""
def test_find_format(self):
# is the right format object found for a repository?
# create a branch with a few known format objects.
# this is not quite the same as
self.build_tree(["foo/", "bar/"])
def check_format(format, url):
dir = format._matchingbzrdir.initialize(url)
format.initialize(dir)
t = get_transport(url)
found_format = repository.RepositoryFormat.find_format(dir)
self.failUnless(isinstance(found_format, format.__class__))
check_format(repository.RepositoryFormat7(), "bar")
def test_find_format_no_repository(self):
dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
self.assertRaises(errors.NoRepositoryPresent,
repository.RepositoryFormat.find_format,
dir)
def test_find_format_unknown_format(self):
dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
SampleRepositoryFormat().initialize(dir)
self.assertRaises(UnknownFormatError,
repository.RepositoryFormat.find_format,
dir)
def test_register_unregister_format(self):
format = SampleRepositoryFormat()
# make a control dir
dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
# make a repo
format.initialize(dir)
# register a format for it.
repository.RepositoryFormat.register_format(format)
# which repository.Open will refuse (not supported)
self.assertRaises(UnsupportedFormatError, repository.Repository.open, self.get_url())
# but open(unsupported) will work
self.assertEqual(format.open(dir), "opened repository.")
# unregister the format
repository.RepositoryFormat.unregister_format(format)
class TestFormat6(TestCaseWithTransport):
def test_no_ancestry_weave(self):
control = bzrdir.BzrDirFormat6().initialize(self.get_url())
repo = repository.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')
class TestFormat7(TestCaseWithTransport):
def test_disk_layout(self):
control = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
repo = repository.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.assertEqualDiff('', t.get('lock').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())
def test_shared_disk_layout(self):
control = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
repo = repository.RepositoryFormat7().initialize(control, shared=True)
# 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.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_shared_no_tree_disk_layout(self):
control = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
repo = repository.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())