~bzr-pqm/bzr/bzr.dev

# 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 the smart wire/domain protocol.

This module contains tests for the domain-level smart requests and responses,

such as the 'Branch.lock_write' request. Many of these use specific disk

formats to exercise calls that only make sense for formats with specific

properties.

Tests for low-level protocol encoding are found in test_smart_transport.

"""

import bz2

from cStringIO import StringIO

import tarfile

from bzrlib import (

    bencode,

    branch as _mod_branch,

    bzrdir,

    errors,

    pack,

    tests,

    transport,

    urlutils,

    versionedfile,

    )

from bzrlib.smart import (

    branch as smart_branch,

    bzrdir as smart_dir,

    repository as smart_repo,

    packrepository as smart_packrepo,

    request as smart_req,

    server,

    vfs,

    )

from bzrlib.tests import test_server

from bzrlib.transport import (

    chroot,

    memory,

    )

def load_tests(standard_tests, module, loader):

    """Multiply tests version and protocol consistency."""

    # FindRepository tests.

    scenarios = [

        ("find_repository", {

            "_request_class": smart_dir.SmartServerRequestFindRepositoryV1}),

        ("find_repositoryV2", {

            "_request_class": smart_dir.SmartServerRequestFindRepositoryV2}),

        ("find_repositoryV3", {

            "_request_class": smart_dir.SmartServerRequestFindRepositoryV3}),

        ]

    to_adapt, result = tests.split_suite_by_re(standard_tests,

        "TestSmartServerRequestFindRepository")

    v2_only, v1_and_2 = tests.split_suite_by_re(to_adapt,

        "_v2")

    tests.multiply_tests(v1_and_2, scenarios, result)

    # The first scenario is only applicable to v1 protocols, it is deleted

    # since.

    tests.multiply_tests(v2_only, scenarios[1:], result)

    return result

class TestCaseWithChrootedTransport(tests.TestCaseWithTransport):

    def setUp(self):

        self.vfs_transport_factory = memory.MemoryServer

        tests.TestCaseWithTransport.setUp(self)

        self._chroot_server = None

    def get_transport(self, relpath=None):

        if self._chroot_server is None:

            backing_transport = tests.TestCaseWithTransport.get_transport(self)

            self._chroot_server = chroot.ChrootServer(backing_transport)

            self.start_server(self._chroot_server)

        t = transport.get_transport(self._chroot_server.get_url())

        if relpath is not None:

            t = t.clone(relpath)

        return t

class TestCaseWithSmartMedium(tests.TestCaseWithMemoryTransport):

    def setUp(self):

        super(TestCaseWithSmartMedium, self).setUp()

        # We're allowed to set  the transport class here, so that we don't use

        # the default or a parameterized class, but rather use the

        # TestCaseWithTransport infrastructure to set up a smart server and

        # transport.

        self.transport_server = self.make_transport_server

    def make_transport_server(self):

        return test_server.SmartTCPServer_for_testing('-' + self.id())

    def get_smart_medium(self):

        """Get a smart medium to use in tests."""

        return self.get_transport().get_smart_medium()

class TestByteStreamToStream(tests.TestCase):

    def test_repeated_substreams_same_kind_are_one_stream(self):

        # Make a stream - an iterable of bytestrings.

        stream = [('text', [versionedfile.FulltextContentFactory(('k1',), None,

            None, 'foo')]),('text', [

            versionedfile.FulltextContentFactory(('k2',), None, None, 'bar')])]

        fmt = bzrdir.format_registry.get('pack-0.92')().repository_format

        bytes = smart_repo._stream_to_byte_stream(stream, fmt)

        streams = []

        # Iterate the resulting iterable; checking that we get only one stream

        # out.

        fmt, stream = smart_repo._byte_stream_to_stream(bytes)

        for kind, substream in stream:

            streams.append((kind, list(substream)))

        self.assertLength(1, streams)

        self.assertLength(2, streams[0][1])

class TestSmartServerResponse(tests.TestCase):

    def test__eq__(self):

        self.assertEqual(smart_req.SmartServerResponse(('ok', )),

            smart_req.SmartServerResponse(('ok', )))

        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'body'),

            smart_req.SmartServerResponse(('ok', ), 'body'))

        self.assertNotEqual(smart_req.SmartServerResponse(('ok', )),

            smart_req.SmartServerResponse(('notok', )))

        self.assertNotEqual(smart_req.SmartServerResponse(('ok', ), 'body'),

            smart_req.SmartServerResponse(('ok', )))

        self.assertNotEqual(None,

            smart_req.SmartServerResponse(('ok', )))

    def test__str__(self):

        """SmartServerResponses can be stringified."""

        self.assertEqual(

            "<SuccessfulSmartServerResponse args=('args',) body='body'>",

            str(smart_req.SuccessfulSmartServerResponse(('args',), 'body')))

        self.assertEqual(

            "<FailedSmartServerResponse args=('args',) body='body'>",

            str(smart_req.FailedSmartServerResponse(('args',), 'body')))

class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):

    def test_translate_client_path(self):

        transport = self.get_transport()

        request = smart_req.SmartServerRequest(transport, 'foo/')

        self.assertEqual('./', request.translate_client_path('foo/'))

        self.assertRaises(

            errors.InvalidURLJoin, request.translate_client_path, 'foo/..')

        self.assertRaises(

            errors.PathNotChild, request.translate_client_path, '/')

        self.assertRaises(

            errors.PathNotChild, request.translate_client_path, 'bar/')

        self.assertEqual('./baz', request.translate_client_path('foo/baz'))

        e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')

        self.assertEqual('./' + urlutils.escape(e_acute),

                         request.translate_client_path('foo/' + e_acute))

    def test_translate_client_path_vfs(self):

        """VfsRequests receive escaped paths rather than raw UTF-8."""

        transport = self.get_transport()

        request = vfs.VfsRequest(transport, 'foo/')

        e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')

        escaped = urlutils.escape('foo/' + e_acute)

        self.assertEqual('./' + urlutils.escape(e_acute),

                         request.translate_client_path(escaped))

    def test_transport_from_client_path(self):

        transport = self.get_transport()

        request = smart_req.SmartServerRequest(transport, 'foo/')

        self.assertEqual(

            transport.base,

            request.transport_from_client_path('foo/').base)

class TestSmartServerBzrDirRequestCloningMetaDir(

    tests.TestCaseWithMemoryTransport):

    """Tests for BzrDir.cloning_metadir."""

    def test_cloning_metadir(self):

        """When there is a bzrdir present, the call succeeds."""

        backing = self.get_transport()

        dir = self.make_bzrdir('.')

        local_result = dir.cloning_metadir()

        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir

        request = request_class(backing)

        expected = smart_req.SuccessfulSmartServerResponse(

            (local_result.network_name(),

            local_result.repository_format.network_name(),

            ('branch', local_result.get_branch_format().network_name())))

        self.assertEqual(expected, request.execute('', 'False'))

    def test_cloning_metadir_reference(self):

        """The request fails when bzrdir contains a branch reference."""

        backing = self.get_transport()

        referenced_branch = self.make_branch('referenced')

        dir = self.make_bzrdir('.')

        local_result = dir.cloning_metadir()

        reference = _mod_branch.BranchReferenceFormat().initialize(

            dir, target_branch=referenced_branch)

        reference_url = _mod_branch.BranchReferenceFormat().get_reference(dir)

        # The server shouldn't try to follow the branch reference, so it's fine

        # if the referenced branch isn't reachable.

        backing.rename('referenced', 'moved')

        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir

        request = request_class(backing)

        expected = smart_req.FailedSmartServerResponse(('BranchReference',))

        self.assertEqual(expected, request.execute('', 'False'))

class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):

    """Tests for BzrDir.create_repository."""

    def test_makes_repository(self):

        """When there is a bzrdir present, the call succeeds."""

        backing = self.get_transport()

        self.make_bzrdir('.')

        request_class = smart_dir.SmartServerRequestCreateRepository

        request = request_class(backing)

        reference_bzrdir_format = bzrdir.format_registry.get('pack-0.92')()

        reference_format = reference_bzrdir_format.repository_format

        network_name = reference_format.network_name()

        expected = smart_req.SuccessfulSmartServerResponse(

            ('ok', 'no', 'no', 'no', network_name))

        self.assertEqual(expected, request.execute('', network_name, 'True'))

class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):

    """Tests for BzrDir.find_repository."""

    def test_no_repository(self):

        """When there is no repository to be found, ('norepository', ) is returned."""

        backing = self.get_transport()

        request = self._request_class(backing)

        self.make_bzrdir('.')

        self.assertEqual(smart_req.SmartServerResponse(('norepository', )),

            request.execute(''))

    def test_nonshared_repository(self):

        # nonshared repositorys only allow 'find' to return a handle when the

        # path the repository is being searched on is the same as that that

        # the repository is at.

        backing = self.get_transport()

        request = self._request_class(backing)

        result = self._make_repository_and_result()

        self.assertEqual(result, request.execute(''))

        self.make_bzrdir('subdir')

        self.assertEqual(smart_req.SmartServerResponse(('norepository', )),

            request.execute('subdir'))

    def _make_repository_and_result(self, shared=False, format=None):

        """Convenience function to setup a repository.

        :result: The SmartServerResponse to expect when opening it.

        """

        repo = self.make_repository('.', shared=shared, format=format)

        if repo.supports_rich_root():

            rich_root = 'yes'

        else:

            rich_root = 'no'

        if repo._format.supports_tree_reference:

            subtrees = 'yes'

        else:

            subtrees = 'no'

        if repo._format.supports_external_lookups:

            external = 'yes'

        else:

            external = 'no'

        if (smart_dir.SmartServerRequestFindRepositoryV3 ==

            self._request_class):

            return smart_req.SuccessfulSmartServerResponse(

                ('ok', '', rich_root, subtrees, external,

                 repo._format.network_name()))

        elif (smart_dir.SmartServerRequestFindRepositoryV2 ==

            self._request_class):

            # All tests so far are on formats, and for non-external

            # repositories.

            return smart_req.SuccessfulSmartServerResponse(

                ('ok', '', rich_root, subtrees, external))

        else:

            return smart_req.SuccessfulSmartServerResponse(

                ('ok', '', rich_root, subtrees))

    def test_shared_repository(self):

        """When there is a shared repository, we get 'ok', 'relpath-to-repo'."""

        backing = self.get_transport()

        request = self._request_class(backing)

        result = self._make_repository_and_result(shared=True)

        self.assertEqual(result, request.execute(''))

        self.make_bzrdir('subdir')

        result2 = smart_req.SmartServerResponse(

            result.args[0:1] + ('..', ) + result.args[2:])

        self.assertEqual(result2,

            request.execute('subdir'))

        self.make_bzrdir('subdir/deeper')

        result3 = smart_req.SmartServerResponse(

            result.args[0:1] + ('../..', ) + result.args[2:])

        self.assertEqual(result3,

            request.execute('subdir/deeper'))

    def test_rich_root_and_subtree_encoding(self):

        """Test for the format attributes for rich root and subtree support."""

        backing = self.get_transport()

        request = self._request_class(backing)

        result = self._make_repository_and_result(

            format='dirstate-with-subtree')

        # check the test will be valid

        self.assertEqual('yes', result.args[2])

        self.assertEqual('yes', result.args[3])

        self.assertEqual(result, request.execute(''))

    def test_supports_external_lookups_no_v2(self):

        """Test for the supports_external_lookups attribute."""

        backing = self.get_transport()

        request = self._request_class(backing)

        result = self._make_repository_and_result(

            format='dirstate-with-subtree')

        # check the test will be valid

        self.assertEqual('no', result.args[4])

        self.assertEqual(result, request.execute(''))

class TestSmartServerBzrDirRequestGetConfigFile(

    tests.TestCaseWithMemoryTransport):

    """Tests for BzrDir.get_config_file."""

    def test_present(self):

        backing = self.get_transport()

        dir = self.make_bzrdir('.')

        dir.get_config().set_default_stack_on("/")

        local_result = dir._get_config()._get_config_file().read()

        request_class = smart_dir.SmartServerBzrDirRequestConfigFile

        request = request_class(backing)

        expected = smart_req.SuccessfulSmartServerResponse((), local_result)

        self.assertEqual(expected, request.execute(''))

    def test_missing(self):

        backing = self.get_transport()

        dir = self.make_bzrdir('.')

        request_class = smart_dir.SmartServerBzrDirRequestConfigFile

        request = request_class(backing)

        expected = smart_req.SuccessfulSmartServerResponse((), '')

        self.assertEqual(expected, request.execute(''))

class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):

    def test_empty_dir(self):

        """Initializing an empty dir should succeed and do it."""

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)

        self.assertEqual(smart_req.SmartServerResponse(('ok', )),

            request.execute(''))

        made_dir = bzrdir.BzrDir.open_from_transport(backing)

        # no branch, tree or repository is expected with the current

        # default formart.

        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)

        self.assertRaises(errors.NotBranchError, made_dir.open_branch)

        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)

    def test_missing_dir(self):

        """Initializing a missing directory should fail like the bzrdir api."""

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)

        self.assertRaises(errors.NoSuchFile,

            request.execute, 'subdir')

    def test_initialized_dir(self):

        """Initializing an extant bzrdir should fail like the bzrdir api."""

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)

        self.make_bzrdir('subdir')

        self.assertRaises(errors.FileExists,

            request.execute, 'subdir')

class TestSmartServerRequestBzrDirInitializeEx(

    tests.TestCaseWithMemoryTransport):

    """Basic tests for BzrDir.initialize_ex_1.16 in the smart server.

    The main unit tests in test_bzrdir exercise the API comprehensively.

    """

    def test_empty_dir(self):

        """Initializing an empty dir should succeed and do it."""

        backing = self.get_transport()

        name = self.make_bzrdir('reference')._format.network_name()

        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)

        self.assertEqual(

            smart_req.SmartServerResponse(('', '', '', '', '', '', name,

                                           'False', '', '', '')),

            request.execute(name, '', 'True', 'False', 'False', '', '', '', '',

                            'False'))

        made_dir = bzrdir.BzrDir.open_from_transport(backing)

        # no branch, tree or repository is expected with the current

        # default format.

        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)

        self.assertRaises(errors.NotBranchError, made_dir.open_branch)

        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)

    def test_missing_dir(self):

        """Initializing a missing directory should fail like the bzrdir api."""

        backing = self.get_transport()

        name = self.make_bzrdir('reference')._format.network_name()

        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)

        self.assertRaises(errors.NoSuchFile, request.execute, name,

            'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')

    def test_initialized_dir(self):

        """Initializing an extant directory should fail like the bzrdir api."""

        backing = self.get_transport()

        name = self.make_bzrdir('reference')._format.network_name()

        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)

        self.make_bzrdir('subdir')

        self.assertRaises(errors.FileExists, request.execute, name, 'subdir',

            'False', 'False', 'False', '', '', '', '', 'False')

class TestSmartServerRequestOpenBzrDir(tests.TestCaseWithMemoryTransport):

    def test_no_directory(self):

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBzrDir(backing)

        self.assertEqual(smart_req.SmartServerResponse(('no', )),

            request.execute('does-not-exist'))

    def test_empty_directory(self):

        backing = self.get_transport()

        backing.mkdir('empty')

        request = smart_dir.SmartServerRequestOpenBzrDir(backing)

        self.assertEqual(smart_req.SmartServerResponse(('no', )),

            request.execute('empty'))

    def test_outside_root_client_path(self):

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBzrDir(backing,

            root_client_path='root')

        self.assertEqual(smart_req.SmartServerResponse(('no', )),

            request.execute('not-root'))

class TestSmartServerRequestOpenBzrDir_2_1(tests.TestCaseWithMemoryTransport):

    def test_no_directory(self):

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)

        self.assertEqual(smart_req.SmartServerResponse(('no', )),

            request.execute('does-not-exist'))

    def test_empty_directory(self):

        backing = self.get_transport()

        backing.mkdir('empty')

        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)

        self.assertEqual(smart_req.SmartServerResponse(('no', )),

            request.execute('empty'))

    def test_present_without_workingtree(self):

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)

        self.make_bzrdir('.')

        self.assertEqual(smart_req.SmartServerResponse(('yes', 'no')),

            request.execute(''))

    def test_outside_root_client_path(self):

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing,

            root_client_path='root')

        self.assertEqual(smart_req.SmartServerResponse(('no',)),

            request.execute('not-root'))

class TestSmartServerRequestOpenBzrDir_2_1_disk(TestCaseWithChrootedTransport):

    def test_present_with_workingtree(self):

        self.vfs_transport_factory = test_server.LocalURLServer

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)

        bd = self.make_bzrdir('.')

        bd.create_repository()

        bd.create_branch()

        bd.create_workingtree()

        self.assertEqual(smart_req.SmartServerResponse(('yes', 'yes')),

            request.execute(''))

class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):

    def test_no_branch(self):

        """When there is no branch, ('nobranch', ) is returned."""

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBranch(backing)

        self.make_bzrdir('.')

        self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),

            request.execute(''))

    def test_branch(self):

        """When there is a branch, 'ok' is returned."""

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBranch(backing)

        self.make_branch('.')

        self.assertEqual(smart_req.SmartServerResponse(('ok', '')),

            request.execute(''))

    def test_branch_reference(self):

        """When there is a branch reference, the reference URL is returned."""

        self.vfs_transport_factory = test_server.LocalURLServer

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBranch(backing)

        branch = self.make_branch('branch')

        checkout = branch.create_checkout('reference',lightweight=True)

        reference_url = _mod_branch.BranchReferenceFormat().get_reference(

            checkout.bzrdir)

        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')

        self.assertEqual(smart_req.SmartServerResponse(('ok', reference_url)),

            request.execute('reference'))

    def test_notification_on_branch_from_repository(self):

        """When there is a repository, the error should return details."""

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBranch(backing)

        repo = self.make_repository('.')

        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),

            request.execute(''))

class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):

    def test_no_branch(self):

        """When there is no branch, ('nobranch', ) is returned."""

        backing = self.get_transport()

        self.make_bzrdir('.')

        request = smart_dir.SmartServerRequestOpenBranchV2(backing)

        self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),

            request.execute(''))

    def test_branch(self):

        """When there is a branch, 'ok' is returned."""

        backing = self.get_transport()

        expected = self.make_branch('.')._format.network_name()

        request = smart_dir.SmartServerRequestOpenBranchV2(backing)

        self.assertEqual(smart_req.SuccessfulSmartServerResponse(

                ('branch', expected)),

                         request.execute(''))

    def test_branch_reference(self):

        """When there is a branch reference, the reference URL is returned."""

        self.vfs_transport_factory = test_server.LocalURLServer

        backing = self.get_transport()

        request = smart_dir.SmartServerRequestOpenBranchV2(backing)

        branch = self.make_branch('branch')

        checkout = branch.create_checkout('reference',lightweight=True)

        reference_url = _mod_branch.BranchReferenceFormat().get_reference(

            checkout.bzrdir)

        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')

        self.assertEqual(smart_req.SuccessfulSmartServerResponse(

                ('ref', reference_url)),

                         request.execute('reference'))

    def test_stacked_branch(self):

        """Opening a stacked branch does not open the stacked-on branch."""

        trunk = self.make_branch('trunk')

        feature = self.make_branch('feature')

        feature.set_stacked_on_url(trunk.base)

        opened_branches = []

        _mod_branch.Branch.hooks.install_named_hook(

            'open', opened_branches.append, None)