~bzr-pqm/bzr/bzr.dev

Add test for TransportLogDecorator handling of readv

added added

removed removed

# 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

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

# 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