~bzr-pqm/bzr/bzr.dev

# (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):

        """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)

        # 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())