1
# Copyright (C) 2004, 2005, 2006, 2007 Canonical Ltd
 
 
3
# This program is free software; you can redistribute it and/or modify
 
 
4
# it under the terms of the GNU General Public License as published by
 
 
5
# the Free Software Foundation; either version 2 of the License, or
 
 
6
# (at your option) any later version.
 
 
8
# This program is distributed in the hope that it will be useful,
 
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
 
11
# GNU General Public License for more details.
 
 
13
# You should have received a copy of the GNU General Public License
 
 
14
# along with this program; if not, write to the Free Software
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 
18
from cStringIO import StringIO
 
 
26
from bzrlib.errors import (DependencyNotPresent,
 
 
34
from bzrlib.tests import TestCase, TestCaseInTempDir
 
 
35
from bzrlib.transport import (_clear_protocol_handlers,
 
 
38
                              _get_protocol_handlers,
 
 
39
                              _set_protocol_handlers,
 
 
40
                              _get_transport_modules,
 
 
43
                              register_lazy_transport,
 
 
44
                              register_transport_proto,
 
 
47
from bzrlib.transport.chroot import ChrootServer
 
 
48
from bzrlib.transport.memory import MemoryTransport
 
 
49
from bzrlib.transport.local import (LocalTransport,
 
 
50
                                    EmulatedWin32LocalTransport)
 
 
53
# TODO: Should possibly split transport-specific tests into their own files.
 
 
56
class TestTransport(TestCase):
 
 
57
    """Test the non transport-concrete class functionality."""
 
 
59
    def test__get_set_protocol_handlers(self):
 
 
60
        handlers = _get_protocol_handlers()
 
 
61
        self.assertNotEqual([], handlers.keys( ))
 
 
63
            _clear_protocol_handlers()
 
 
64
            self.assertEqual([], _get_protocol_handlers().keys())
 
 
66
            _set_protocol_handlers(handlers)
 
 
68
    def test_get_transport_modules(self):
 
 
69
        handlers = _get_protocol_handlers()
 
 
70
        # don't pollute the current handlers
 
 
71
        _clear_protocol_handlers()
 
 
72
        class SampleHandler(object):
 
 
73
            """I exist, isnt that enough?"""
 
 
75
            _clear_protocol_handlers()
 
 
76
            register_transport_proto('foo')
 
 
77
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
 
 
78
                                    'TestTransport.SampleHandler')
 
 
79
            register_transport_proto('bar')
 
 
80
            register_lazy_transport('bar', 'bzrlib.tests.test_transport',
 
 
81
                                    'TestTransport.SampleHandler')
 
 
82
            self.assertEqual([SampleHandler.__module__,
 
 
83
                              'bzrlib.transport.chroot'],
 
 
84
                             _get_transport_modules())
 
 
86
            _set_protocol_handlers(handlers)
 
 
88
    def test_transport_dependency(self):
 
 
89
        """Transport with missing dependency causes no error"""
 
 
90
        saved_handlers = _get_protocol_handlers()
 
 
91
        # don't pollute the current handlers
 
 
92
        _clear_protocol_handlers()
 
 
94
            register_transport_proto('foo')
 
 
95
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
 
 
96
                    'BadTransportHandler')
 
 
98
                get_transport('foo://fooserver/foo')
 
 
99
            except UnsupportedProtocol, e:
 
 
101
                self.assertEquals('Unsupported protocol'
 
 
102
                                  ' for url "foo://fooserver/foo":'
 
 
103
                                  ' Unable to import library "some_lib":'
 
 
104
                                  ' testing missing dependency', str(e))
 
 
106
                self.fail('Did not raise UnsupportedProtocol')
 
 
108
            # restore original values
 
 
109
            _set_protocol_handlers(saved_handlers)
 
 
111
    def test_transport_fallback(self):
 
 
112
        """Transport with missing dependency causes no error"""
 
 
113
        saved_handlers = _get_protocol_handlers()
 
 
115
            _clear_protocol_handlers()
 
 
116
            register_transport_proto('foo')
 
 
117
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
 
 
118
                    'BackupTransportHandler')
 
 
119
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
 
 
120
                    'BadTransportHandler')
 
 
121
            t = get_transport('foo://fooserver/foo')
 
 
122
            self.assertTrue(isinstance(t, BackupTransportHandler))
 
 
124
            _set_protocol_handlers(saved_handlers)
 
 
126
    def test_LateReadError(self):
 
 
127
        """The LateReadError helper should raise on read()."""
 
 
128
        a_file = LateReadError('a path')
 
 
131
        except ReadError, error:
 
 
132
            self.assertEqual('a path', error.path)
 
 
133
        self.assertRaises(ReadError, a_file.read, 40)
 
 
136
    def test__combine_paths(self):
 
 
138
        self.assertEqual('/home/sarah/project/foo',
 
 
139
                         t._combine_paths('/home/sarah', 'project/foo'))
 
 
140
        self.assertEqual('/etc',
 
 
141
                         t._combine_paths('/home/sarah', '../../etc'))
 
 
142
        self.assertEqual('/etc',
 
 
143
                         t._combine_paths('/home/sarah', '../../../etc'))
 
 
144
        self.assertEqual('/etc',
 
 
145
                         t._combine_paths('/home/sarah', '/etc'))
 
 
147
    def test_local_abspath_non_local_transport(self):
 
 
148
        # the base implementation should throw
 
 
149
        t = MemoryTransport()
 
 
150
        e = self.assertRaises(errors.NotLocalUrl, t.local_abspath, 't')
 
 
151
        self.assertEqual('memory:///t is not a local path.', str(e))
 
 
154
class TestCoalesceOffsets(TestCase):
 
 
156
    def check(self, expected, offsets, limit=0, max_size=0, fudge=0):
 
 
157
        coalesce = Transport._coalesce_offsets
 
 
158
        exp = [_CoalescedOffset(*x) for x in expected]
 
 
159
        out = list(coalesce(offsets, limit=limit, fudge_factor=fudge,
 
 
161
        self.assertEqual(exp, out)
 
 
163
    def test_coalesce_empty(self):
 
 
166
    def test_coalesce_simple(self):
 
 
167
        self.check([(0, 10, [(0, 10)])], [(0, 10)])
 
 
169
    def test_coalesce_unrelated(self):
 
 
170
        self.check([(0, 10, [(0, 10)]),
 
 
172
                   ], [(0, 10), (20, 10)])
 
 
174
    def test_coalesce_unsorted(self):
 
 
175
        self.check([(20, 10, [(0, 10)]),
 
 
177
                   ], [(20, 10), (0, 10)])
 
 
179
    def test_coalesce_nearby(self):
 
 
180
        self.check([(0, 20, [(0, 10), (10, 10)])],
 
 
183
    def test_coalesce_overlapped(self):
 
 
184
        self.assertRaises(ValueError,
 
 
185
            self.check, [(0, 15, [(0, 10), (5, 10)])],
 
 
188
    def test_coalesce_limit(self):
 
 
189
        self.check([(10, 50, [(0, 10), (10, 10), (20, 10),
 
 
190
                              (30, 10), (40, 10)]),
 
 
191
                    (60, 50, [(0, 10), (10, 10), (20, 10),
 
 
192
                              (30, 10), (40, 10)]),
 
 
193
                   ], [(10, 10), (20, 10), (30, 10), (40, 10),
 
 
194
                       (50, 10), (60, 10), (70, 10), (80, 10),
 
 
195
                       (90, 10), (100, 10)],
 
 
198
    def test_coalesce_no_limit(self):
 
 
199
        self.check([(10, 100, [(0, 10), (10, 10), (20, 10),
 
 
200
                               (30, 10), (40, 10), (50, 10),
 
 
201
                               (60, 10), (70, 10), (80, 10),
 
 
203
                   ], [(10, 10), (20, 10), (30, 10), (40, 10),
 
 
204
                       (50, 10), (60, 10), (70, 10), (80, 10),
 
 
205
                       (90, 10), (100, 10)])
 
 
207
    def test_coalesce_fudge(self):
 
 
208
        self.check([(10, 30, [(0, 10), (20, 10)]),
 
 
209
                    (100, 10, [(0, 10),]),
 
 
210
                   ], [(10, 10), (30, 10), (100, 10)],
 
 
213
    def test_coalesce_max_size(self):
 
 
214
        self.check([(10, 20, [(0, 10), (10, 10)]),
 
 
216
                    # If one range is above max_size, it gets its own coalesced
 
 
218
                    (100, 80, [(0, 80),]),],
 
 
219
                   [(10, 10), (20, 10), (30, 50), (100, 80)],
 
 
223
    def test_coalesce_no_max_size(self):
 
 
224
        self.check([(10, 170, [(0, 10), (10, 10), (20, 50), (70, 100)]),],
 
 
225
                   [(10, 10), (20, 10), (30, 50), (80, 100)],
 
 
229
class TestMemoryTransport(TestCase):
 
 
231
    def test_get_transport(self):
 
 
234
    def test_clone(self):
 
 
235
        transport = MemoryTransport()
 
 
236
        self.assertTrue(isinstance(transport, MemoryTransport))
 
 
237
        self.assertEqual("memory:///", transport.clone("/").base)
 
 
239
    def test_abspath(self):
 
 
240
        transport = MemoryTransport()
 
 
241
        self.assertEqual("memory:///relpath", transport.abspath('relpath'))
 
 
243
    def test_abspath_of_root(self):
 
 
244
        transport = MemoryTransport()
 
 
245
        self.assertEqual("memory:///", transport.base)
 
 
246
        self.assertEqual("memory:///", transport.abspath('/'))
 
 
248
    def test_abspath_of_relpath_starting_at_root(self):
 
 
249
        transport = MemoryTransport()
 
 
250
        self.assertEqual("memory:///foo", transport.abspath('/foo'))
 
 
252
    def test_append_and_get(self):
 
 
253
        transport = MemoryTransport()
 
 
254
        transport.append_bytes('path', 'content')
 
 
255
        self.assertEqual(transport.get('path').read(), 'content')
 
 
256
        transport.append_file('path', StringIO('content'))
 
 
257
        self.assertEqual(transport.get('path').read(), 'contentcontent')
 
 
259
    def test_put_and_get(self):
 
 
260
        transport = MemoryTransport()
 
 
261
        transport.put_file('path', StringIO('content'))
 
 
262
        self.assertEqual(transport.get('path').read(), 'content')
 
 
263
        transport.put_bytes('path', 'content')
 
 
264
        self.assertEqual(transport.get('path').read(), 'content')
 
 
266
    def test_append_without_dir_fails(self):
 
 
267
        transport = MemoryTransport()
 
 
268
        self.assertRaises(NoSuchFile,
 
 
269
                          transport.append_bytes, 'dir/path', 'content')
 
 
271
    def test_put_without_dir_fails(self):
 
 
272
        transport = MemoryTransport()
 
 
273
        self.assertRaises(NoSuchFile,
 
 
274
                          transport.put_file, 'dir/path', StringIO('content'))
 
 
276
    def test_get_missing(self):
 
 
277
        transport = MemoryTransport()
 
 
278
        self.assertRaises(NoSuchFile, transport.get, 'foo')
 
 
280
    def test_has_missing(self):
 
 
281
        transport = MemoryTransport()
 
 
282
        self.assertEquals(False, transport.has('foo'))
 
 
284
    def test_has_present(self):
 
 
285
        transport = MemoryTransport()
 
 
286
        transport.append_bytes('foo', 'content')
 
 
287
        self.assertEquals(True, transport.has('foo'))
 
 
289
    def test_list_dir(self):
 
 
290
        transport = MemoryTransport()
 
 
291
        transport.put_bytes('foo', 'content')
 
 
292
        transport.mkdir('dir')
 
 
293
        transport.put_bytes('dir/subfoo', 'content')
 
 
294
        transport.put_bytes('dirlike', 'content')
 
 
296
        self.assertEquals(['dir', 'dirlike', 'foo'], sorted(transport.list_dir('.')))
 
 
297
        self.assertEquals(['subfoo'], sorted(transport.list_dir('dir')))
 
 
299
    def test_mkdir(self):
 
 
300
        transport = MemoryTransport()
 
 
301
        transport.mkdir('dir')
 
 
302
        transport.append_bytes('dir/path', 'content')
 
 
303
        self.assertEqual(transport.get('dir/path').read(), 'content')
 
 
305
    def test_mkdir_missing_parent(self):
 
 
306
        transport = MemoryTransport()
 
 
307
        self.assertRaises(NoSuchFile,
 
 
308
                          transport.mkdir, 'dir/dir')
 
 
310
    def test_mkdir_twice(self):
 
 
311
        transport = MemoryTransport()
 
 
312
        transport.mkdir('dir')
 
 
313
        self.assertRaises(FileExists, transport.mkdir, 'dir')
 
 
315
    def test_parameters(self):
 
 
316
        transport = MemoryTransport()
 
 
317
        self.assertEqual(True, transport.listable())
 
 
318
        self.assertEqual(False, transport.is_readonly())
 
 
320
    def test_iter_files_recursive(self):
 
 
321
        transport = MemoryTransport()
 
 
322
        transport.mkdir('dir')
 
 
323
        transport.put_bytes('dir/foo', 'content')
 
 
324
        transport.put_bytes('dir/bar', 'content')
 
 
325
        transport.put_bytes('bar', 'content')
 
 
326
        paths = set(transport.iter_files_recursive())
 
 
327
        self.assertEqual(set(['dir/foo', 'dir/bar', 'bar']), paths)
 
 
330
        transport = MemoryTransport()
 
 
331
        transport.put_bytes('foo', 'content')
 
 
332
        transport.put_bytes('bar', 'phowar')
 
 
333
        self.assertEqual(7, transport.stat('foo').st_size)
 
 
334
        self.assertEqual(6, transport.stat('bar').st_size)
 
 
337
class ChrootDecoratorTransportTest(TestCase):
 
 
338
    """Chroot decoration specific tests."""
 
 
340
    def test_abspath(self):
 
 
341
        # The abspath is always relative to the chroot_url.
 
 
342
        server = ChrootServer(get_transport('memory:///foo/bar/'))
 
 
344
        transport = get_transport(server.get_url())
 
 
345
        self.assertEqual(server.get_url(), transport.abspath('/'))
 
 
347
        subdir_transport = transport.clone('subdir')
 
 
348
        self.assertEqual(server.get_url(), subdir_transport.abspath('/'))
 
 
351
    def test_clone(self):
 
 
352
        server = ChrootServer(get_transport('memory:///foo/bar/'))
 
 
354
        transport = get_transport(server.get_url())
 
 
355
        # relpath from root and root path are the same
 
 
356
        relpath_cloned = transport.clone('foo')
 
 
357
        abspath_cloned = transport.clone('/foo')
 
 
358
        self.assertEqual(server, relpath_cloned.server)
 
 
359
        self.assertEqual(server, abspath_cloned.server)
 
 
362
    def test_chroot_url_preserves_chroot(self):
 
 
363
        """Calling get_transport on a chroot transport's base should produce a
 
 
364
        transport with exactly the same behaviour as the original chroot
 
 
367
        This is so that it is not possible to escape a chroot by doing::
 
 
368
            url = chroot_transport.base
 
 
369
            parent_url = urlutils.join(url, '..')
 
 
370
            new_transport = get_transport(parent_url)
 
 
372
        server = ChrootServer(get_transport('memory:///path/subpath'))
 
 
374
        transport = get_transport(server.get_url())
 
 
375
        new_transport = get_transport(transport.base)
 
 
376
        self.assertEqual(transport.server, new_transport.server)
 
 
377
        self.assertEqual(transport.base, new_transport.base)
 
 
380
    def test_urljoin_preserves_chroot(self):
 
 
381
        """Using urlutils.join(url, '..') on a chroot URL should not produce a
 
 
382
        URL that escapes the intended chroot.
 
 
384
        This is so that it is not possible to escape a chroot by doing::
 
 
385
            url = chroot_transport.base
 
 
386
            parent_url = urlutils.join(url, '..')
 
 
387
            new_transport = get_transport(parent_url)
 
 
389
        server = ChrootServer(get_transport('memory:///path/'))
 
 
391
        transport = get_transport(server.get_url())
 
 
393
            InvalidURLJoin, urlutils.join, transport.base, '..')
 
 
397
class ChrootServerTest(TestCase):
 
 
399
    def test_construct(self):
 
 
400
        backing_transport = MemoryTransport()
 
 
401
        server = ChrootServer(backing_transport)
 
 
402
        self.assertEqual(backing_transport, server.backing_transport)
 
 
404
    def test_setUp(self):
 
 
405
        backing_transport = MemoryTransport()
 
 
406
        server = ChrootServer(backing_transport)
 
 
408
        self.assertTrue(server.scheme in _get_protocol_handlers().keys())
 
 
410
    def test_tearDown(self):
 
 
411
        backing_transport = MemoryTransport()
 
 
412
        server = ChrootServer(backing_transport)
 
 
415
        self.assertFalse(server.scheme in _get_protocol_handlers().keys())
 
 
417
    def test_get_url(self):
 
 
418
        backing_transport = MemoryTransport()
 
 
419
        server = ChrootServer(backing_transport)
 
 
421
        self.assertEqual('chroot-%d:///' % id(server), server.get_url())
 
 
425
class ReadonlyDecoratorTransportTest(TestCase):
 
 
426
    """Readonly decoration specific tests."""
 
 
428
    def test_local_parameters(self):
 
 
429
        import bzrlib.transport.readonly as readonly
 
 
430
        # connect to . in readonly mode
 
 
431
        transport = readonly.ReadonlyTransportDecorator('readonly+.')
 
 
432
        self.assertEqual(True, transport.listable())
 
 
433
        self.assertEqual(True, transport.is_readonly())
 
 
435
    def test_http_parameters(self):
 
 
436
        from bzrlib.tests.http_server import HttpServer
 
 
437
        import bzrlib.transport.readonly as readonly
 
 
438
        # connect to '.' via http which is not listable
 
 
439
        server = HttpServer()
 
 
442
            transport = get_transport('readonly+' + server.get_url())
 
 
443
            self.failUnless(isinstance(transport,
 
 
444
                                       readonly.ReadonlyTransportDecorator))
 
 
445
            self.assertEqual(False, transport.listable())
 
 
446
            self.assertEqual(True, transport.is_readonly())
 
 
451
class FakeNFSDecoratorTests(TestCaseInTempDir):
 
 
452
    """NFS decorator specific tests."""
 
 
454
    def get_nfs_transport(self, url):
 
 
455
        import bzrlib.transport.fakenfs as fakenfs
 
 
456
        # connect to url with nfs decoration
 
 
457
        return fakenfs.FakeNFSTransportDecorator('fakenfs+' + url)
 
 
459
    def test_local_parameters(self):
 
 
460
        # the listable and is_readonly parameters
 
 
461
        # are not changed by the fakenfs decorator
 
 
462
        transport = self.get_nfs_transport('.')
 
 
463
        self.assertEqual(True, transport.listable())
 
 
464
        self.assertEqual(False, transport.is_readonly())
 
 
466
    def test_http_parameters(self):
 
 
467
        # the listable and is_readonly parameters
 
 
468
        # are not changed by the fakenfs decorator
 
 
469
        from bzrlib.tests.http_server import HttpServer
 
 
470
        # connect to '.' via http which is not listable
 
 
471
        server = HttpServer()
 
 
474
            transport = self.get_nfs_transport(server.get_url())
 
 
475
            self.assertIsInstance(
 
 
476
                transport, bzrlib.transport.fakenfs.FakeNFSTransportDecorator)
 
 
477
            self.assertEqual(False, transport.listable())
 
 
478
            self.assertEqual(True, transport.is_readonly())
 
 
482
    def test_fakenfs_server_default(self):
 
 
483
        # a FakeNFSServer() should bring up a local relpath server for itself
 
 
484
        import bzrlib.transport.fakenfs as fakenfs
 
 
485
        server = fakenfs.FakeNFSServer()
 
 
488
            # the url should be decorated appropriately
 
 
489
            self.assertStartsWith(server.get_url(), 'fakenfs+')
 
 
490
            # and we should be able to get a transport for it
 
 
491
            transport = get_transport(server.get_url())
 
 
492
            # which must be a FakeNFSTransportDecorator instance.
 
 
493
            self.assertIsInstance(
 
 
494
                transport, fakenfs.FakeNFSTransportDecorator)
 
 
498
    def test_fakenfs_rename_semantics(self):
 
 
499
        # a FakeNFS transport must mangle the way rename errors occur to
 
 
500
        # look like NFS problems.
 
 
501
        transport = self.get_nfs_transport('.')
 
 
502
        self.build_tree(['from/', 'from/foo', 'to/', 'to/bar'],
 
 
504
        self.assertRaises(errors.ResourceBusy,
 
 
505
                          transport.rename, 'from', 'to')
 
 
508
class FakeVFATDecoratorTests(TestCaseInTempDir):
 
 
509
    """Tests for simulation of VFAT restrictions"""
 
 
511
    def get_vfat_transport(self, url):
 
 
512
        """Return vfat-backed transport for test directory"""
 
 
513
        from bzrlib.transport.fakevfat import FakeVFATTransportDecorator
 
 
514
        return FakeVFATTransportDecorator('vfat+' + url)
 
 
516
    def test_transport_creation(self):
 
 
517
        from bzrlib.transport.fakevfat import FakeVFATTransportDecorator
 
 
518
        transport = self.get_vfat_transport('.')
 
 
519
        self.assertIsInstance(transport, FakeVFATTransportDecorator)
 
 
521
    def test_transport_mkdir(self):
 
 
522
        transport = self.get_vfat_transport('.')
 
 
523
        transport.mkdir('HELLO')
 
 
524
        self.assertTrue(transport.has('hello'))
 
 
525
        self.assertTrue(transport.has('Hello'))
 
 
527
    def test_forbidden_chars(self):
 
 
528
        transport = self.get_vfat_transport('.')
 
 
529
        self.assertRaises(ValueError, transport.has, "<NU>")
 
 
532
class BadTransportHandler(Transport):
 
 
533
    def __init__(self, base_url):
 
 
534
        raise DependencyNotPresent('some_lib', 'testing missing dependency')
 
 
537
class BackupTransportHandler(Transport):
 
 
538
    """Test transport that works as a backup for the BadTransportHandler"""
 
 
542
class TestTransportImplementation(TestCaseInTempDir):
 
 
543
    """Implementation verification for transports.
 
 
545
    To verify a transport we need a server factory, which is a callable
 
 
546
    that accepts no parameters and returns an implementation of
 
 
547
    bzrlib.transport.Server.
 
 
549
    That Server is then used to construct transport instances and test
 
 
550
    the transport via loopback activity.
 
 
552
    Currently this assumes that the Transport object is connected to the 
 
 
553
    current working directory.  So that whatever is done 
 
 
554
    through the transport, should show up in the working 
 
 
555
    directory, and vice-versa. This is a bug, because its possible to have
 
 
556
    URL schemes which provide access to something that may not be 
 
 
557
    result in storage on the local disk, i.e. due to file system limits, or 
 
 
558
    due to it being a database or some other non-filesystem tool.
 
 
560
    This also tests to make sure that the functions work with both
 
 
561
    generators and lists (assuming iter(list) is effectively a generator)
 
 
565
        super(TestTransportImplementation, self).setUp()
 
 
566
        self._server = self.transport_server()
 
 
568
        self.addCleanup(self._server.tearDown)
 
 
570
    def get_transport(self, relpath=None):
 
 
571
        """Return a connected transport to the local directory.
 
 
573
        :param relpath: a path relative to the base url.
 
 
575
        base_url = self._server.get_url()
 
 
576
        url = self._adjust_url(base_url, relpath)
 
 
577
        # try getting the transport via the regular interface:
 
 
578
        t = get_transport(url)
 
 
579
        # vila--20070607 if the following are commented out the test suite
 
 
580
        # still pass. Is this really still needed or was it a forgotten
 
 
582
        if not isinstance(t, self.transport_class):
 
 
583
            # we did not get the correct transport class type. Override the
 
 
584
            # regular connection behaviour by direct construction.
 
 
585
            t = self.transport_class(url)
 
 
589
class TestLocalTransports(TestCase):
 
 
591
    def test_get_transport_from_abspath(self):
 
 
592
        here = osutils.abspath('.')
 
 
593
        t = get_transport(here)
 
 
594
        self.assertIsInstance(t, LocalTransport)
 
 
595
        self.assertEquals(t.base, urlutils.local_path_to_url(here) + '/')
 
 
597
    def test_get_transport_from_relpath(self):
 
 
598
        here = osutils.abspath('.')
 
 
599
        t = get_transport('.')
 
 
600
        self.assertIsInstance(t, LocalTransport)
 
 
601
        self.assertEquals(t.base, urlutils.local_path_to_url('.') + '/')
 
 
603
    def test_get_transport_from_local_url(self):
 
 
604
        here = osutils.abspath('.')
 
 
605
        here_url = urlutils.local_path_to_url(here) + '/'
 
 
606
        t = get_transport(here_url)
 
 
607
        self.assertIsInstance(t, LocalTransport)
 
 
608
        self.assertEquals(t.base, here_url)
 
 
610
    def test_local_abspath(self):
 
 
611
        here = osutils.abspath('.')
 
 
612
        t = get_transport(here)
 
 
613
        self.assertEquals(t.local_abspath(''), here)
 
 
616
class TestWin32LocalTransport(TestCase):
 
 
618
    def test_unc_clone_to_root(self):
 
 
619
        # Win32 UNC path like \\HOST\path
 
 
620
        # clone to root should stop at least at \\HOST part
 
 
622
        t = EmulatedWin32LocalTransport('file://HOST/path/to/some/dir/')
 
 
625
        self.assertEquals(t.base, 'file://HOST/')
 
 
626
        # make sure we reach the root
 
 
628
        self.assertEquals(t.base, 'file://HOST/')
 
 
631
class TestConnectedTransport(TestCase):
 
 
632
    """Tests for connected to remote server transports"""
 
 
634
    def test_parse_url(self):
 
 
635
        t = ConnectedTransport('http://simple.example.com/home/source')
 
 
636
        self.assertEquals(t._host, 'simple.example.com')
 
 
637
        self.assertEquals(t._port, None)
 
 
638
        self.assertEquals(t._path, '/home/source/')
 
 
639
        self.failUnless(t._user is None)
 
 
640
        self.failUnless(t._password is None)
 
 
642
        self.assertEquals(t.base, 'http://simple.example.com/home/source/')
 
 
644
    def test_parse_url_with_at_in_user(self):
 
 
646
        t = ConnectedTransport('ftp://user@host.com@www.host.com/')
 
 
647
        self.assertEquals(t._user, 'user@host.com')
 
 
649
    def test_parse_quoted_url(self):
 
 
650
        t = ConnectedTransport('http://ro%62ey:h%40t@ex%41mple.com:2222/path')
 
 
651
        self.assertEquals(t._host, 'exAmple.com')
 
 
652
        self.assertEquals(t._port, 2222)
 
 
653
        self.assertEquals(t._user, 'robey')
 
 
654
        self.assertEquals(t._password, 'h@t')
 
 
655
        self.assertEquals(t._path, '/path/')
 
 
657
        # Base should not keep track of the password
 
 
658
        self.assertEquals(t.base, 'http://robey@exAmple.com:2222/path/')
 
 
660
    def test_parse_invalid_url(self):
 
 
661
        self.assertRaises(errors.InvalidURL,
 
 
663
                          'sftp://lily.org:~janneke/public/bzr/gub')
 
 
665
    def test_relpath(self):
 
 
666
        t = ConnectedTransport('sftp://user@host.com/abs/path')
 
 
668
        self.assertEquals(t.relpath('sftp://user@host.com/abs/path/sub'), 'sub')
 
 
669
        self.assertRaises(errors.PathNotChild, t.relpath,
 
 
670
                          'http://user@host.com/abs/path/sub')
 
 
671
        self.assertRaises(errors.PathNotChild, t.relpath,
 
 
672
                          'sftp://user2@host.com/abs/path/sub')
 
 
673
        self.assertRaises(errors.PathNotChild, t.relpath,
 
 
674
                          'sftp://user@otherhost.com/abs/path/sub')
 
 
675
        self.assertRaises(errors.PathNotChild, t.relpath,
 
 
676
                          'sftp://user@host.com:33/abs/path/sub')
 
 
677
        # Make sure it works when we don't supply a username
 
 
678
        t = ConnectedTransport('sftp://host.com/abs/path')
 
 
679
        self.assertEquals(t.relpath('sftp://host.com/abs/path/sub'), 'sub')
 
 
681
        # Make sure it works when parts of the path will be url encoded
 
 
682
        t = ConnectedTransport('sftp://host.com/dev/%path')
 
 
683
        self.assertEquals(t.relpath('sftp://host.com/dev/%path/sub'), 'sub')
 
 
685
    def test_connection_sharing_propagate_credentials(self):
 
 
686
        t = ConnectedTransport('ftp://user@host.com/abs/path')
 
 
687
        self.assertEquals('user', t._user)
 
 
688
        self.assertEquals('host.com', t._host)
 
 
689
        self.assertIs(None, t._get_connection())
 
 
690
        self.assertIs(None, t._password)
 
 
691
        c = t.clone('subdir')
 
 
692
        self.assertIs(None, c._get_connection())
 
 
693
        self.assertIs(None, t._password)
 
 
695
        # Simulate the user entering a password
 
 
697
        connection = object()
 
 
698
        t._set_connection(connection, password)
 
 
699
        self.assertIs(connection, t._get_connection())
 
 
700
        self.assertIs(password, t._get_credentials())
 
 
701
        self.assertIs(connection, c._get_connection())
 
 
702
        self.assertIs(password, c._get_credentials())
 
 
704
        # credentials can be updated
 
 
705
        new_password = 'even more secret'
 
 
706
        c._update_credentials(new_password)
 
 
707
        self.assertIs(connection, t._get_connection())
 
 
708
        self.assertIs(new_password, t._get_credentials())
 
 
709
        self.assertIs(connection, c._get_connection())
 
 
710
        self.assertIs(new_password, c._get_credentials())
 
 
713
class TestReusedTransports(TestCase):
 
 
714
    """Tests for transport reuse"""
 
 
716
    def test_reuse_same_transport(self):
 
 
717
        possible_transports = []
 
 
718
        t1 = get_transport('http://foo/',
 
 
719
                           possible_transports=possible_transports)
 
 
720
        self.assertEqual([t1], possible_transports)
 
 
721
        t2 = get_transport('http://foo/', possible_transports=[t1])
 
 
722
        self.assertIs(t1, t2)
 
 
724
        # Also check that final '/' are handled correctly
 
 
725
        t3 = get_transport('http://foo/path/')
 
 
726
        t4 = get_transport('http://foo/path', possible_transports=[t3])
 
 
727
        self.assertIs(t3, t4)
 
 
729
        t5 = get_transport('http://foo/path')
 
 
730
        t6 = get_transport('http://foo/path/', possible_transports=[t5])
 
 
731
        self.assertIs(t5, t6)
 
 
733
    def test_don_t_reuse_different_transport(self):
 
 
734
        t1 = get_transport('http://foo/path')
 
 
735
        t2 = get_transport('http://bar/path', possible_transports=[t1])
 
 
736
        self.assertIsNot(t1, t2)
 
 
739
class TestTransportTrace(TestCase):
 
 
742
        transport = get_transport('trace+memory://')
 
 
743
        self.assertIsInstance(
 
 
744
            transport, bzrlib.transport.trace.TransportTraceDecorator)
 
 
746
    def test_clone_preserves_activity(self):
 
 
747
        transport = get_transport('trace+memory://')
 
 
748
        transport2 = transport.clone('.')
 
 
749
        self.assertTrue(transport is not transport2)
 
 
750
        self.assertTrue(transport._activity is transport2._activity)
 
 
752
    # the following specific tests are for the operations that have made use of
 
 
753
    # logging in tests; we could test every single operation but doing that
 
 
754
    # still won't cause a test failure when the top level Transport API
 
 
755
    # changes; so there is little return doing that.
 
 
757
        transport = get_transport('trace+memory:///')
 
 
758
        transport.put_bytes('foo', 'barish')
 
 
761
        # put_bytes records the bytes, not the content to avoid memory
 
 
763
        expected_result.append(('put_bytes', 'foo', 6, None))
 
 
764
        # get records the file name only.
 
 
765
        expected_result.append(('get', 'foo'))
 
 
766
        self.assertEqual(expected_result, transport._activity)
 
 
768
    def test_readv(self):
 
 
769
        transport = get_transport('trace+memory:///')
 
 
770
        transport.put_bytes('foo', 'barish')
 
 
771
        list(transport.readv('foo', [(0, 1), (3, 2)], adjust_for_latency=True,
 
 
774
        # put_bytes records the bytes, not the content to avoid memory
 
 
776
        expected_result.append(('put_bytes', 'foo', 6, None))
 
 
777
        # readv records the supplied offset request
 
 
778
        expected_result.append(('readv', 'foo', [(0, 1), (3, 2)], True, 6))
 
 
779
        self.assertEqual(expected_result, transport._activity)