~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2008, 2009, 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 log+ transport decorator."""

import types

from bzrlib.tests import TestCaseWithMemoryTransport

from bzrlib.trace import mutter

from bzrlib.transport import get_transport

from bzrlib.transport.log import TransportLogDecorator

class TestTransportLog(TestCaseWithMemoryTransport):

    def test_log_transport(self):

        base_transport = self.get_transport('')

        logging_transport = get_transport('log+' + base_transport.base)

        # operations such as mkdir are logged

        mutter('where are you?')

        logging_transport.mkdir('subdir')

        log = self.get_log()

        self.assertContainsRe(log, r'mkdir memory\+\d+://.*subdir')

        self.assertContainsRe(log, '  --> None')

        # they have the expected effect

        self.assertTrue(logging_transport.has('subdir'))

        # and they operate on the underlying transport

        self.assertTrue(base_transport.has('subdir'))

    def test_log_readv(self):

        # see <https://bugs.launchpad.net/bzr/+bug/340347>

        # transports are not required to return a generator, but we

        # specifically want to check that those that do cause it to be passed

        # through, for the sake of minimum interference

        base_transport = DummyReadvTransport()

        # construct it directly to avoid needing the dummy transport to be

        # registered etc

        logging_transport = TransportLogDecorator(

            'log+dummy:///', _decorated=base_transport)

        result = base_transport.readv('foo', [(0, 10)])

        # sadly there's no types.IteratorType, and GeneratorType is too

        # specific

        self.assertTrue(getattr(result, 'next'))

        result = logging_transport.readv('foo', [(0, 10)])

        self.assertTrue(getattr(result, 'next'))

        self.assertEquals(list(result),

            [(0, 'abcdefghij')])

class DummyReadvTransport(object):

    base = 'dummy:///'

    def readv(self, filename, offset_length_pairs):

        yield (0, 'abcdefghij')

    def abspath(self, path):

        return self.base + path