~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testtransport.py

  • Committer: Robey Pointer
  • Date: 2005-11-22 01:41:46 UTC
  • mfrom: (1185.40.1)
  • mto: (1185.33.37 bzr.dev)
  • mto: This revision was merged to the branch mainline in revision 1512.
  • Revision ID: robey@lag.net-20051122014146-5186f5e310a15202
make sftp put faster when using paramiko 1.5.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2004, 2005, 2006 by Canonical Ltd
 
1
# Copyright (C) 2004, 2005 by Canonical Ltd
2
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
16
16
 
17
17
 
18
18
import os
19
 
import sys
20
 
import stat
21
19
from cStringIO import StringIO
22
20
 
23
 
import bzrlib
24
 
from bzrlib.errors import (NoSuchFile, FileExists,
25
 
                           TransportNotPossible,
26
 
                           ConnectionError,
27
 
                           DependencyNotPresent,
28
 
                           UnsupportedProtocol,
29
 
                           )
30
 
from bzrlib.tests import TestCase, TestCaseInTempDir
31
 
from bzrlib.transport import (_get_protocol_handlers,
32
 
                              _get_transport_modules,
33
 
                              get_transport,
34
 
                              register_lazy_transport,
35
 
                              _set_protocol_handlers,
36
 
                              Transport,
37
 
                              )
38
 
from bzrlib.transport.memory import MemoryTransport
39
 
from bzrlib.transport.local import LocalTransport
40
 
 
 
21
from bzrlib.errors import (NoSuchFile, FileExists, TransportNotPossible,
 
22
                           ConnectionError)
 
23
from bzrlib.selftest import TestCase, TestCaseInTempDir
 
24
from bzrlib.selftest.HTTPTestUtil import TestCaseWithWebserver
 
25
from bzrlib.transport import memory, urlescape
 
26
 
 
27
 
 
28
def _append(fn, txt):
 
29
    """Append the given text (file-like object) to the supplied filename."""
 
30
    f = open(fn, 'ab')
 
31
    f.write(txt)
 
32
    f.flush()
 
33
    f.close()
 
34
    del f
41
35
 
42
36
class TestTransport(TestCase):
43
37
    """Test the non transport-concrete class functionality."""
44
38
 
45
 
    def test__get_set_protocol_handlers(self):
46
 
        handlers = _get_protocol_handlers()
47
 
        self.assertNotEqual({}, handlers)
48
 
        try:
49
 
            _set_protocol_handlers({})
50
 
            self.assertEqual({}, _get_protocol_handlers())
51
 
        finally:
52
 
            _set_protocol_handlers(handlers)
53
 
 
54
 
    def test_get_transport_modules(self):
55
 
        handlers = _get_protocol_handlers()
56
 
        class SampleHandler(object):
57
 
            """I exist, isnt that enough?"""
58
 
        try:
59
 
            my_handlers = {}
60
 
            _set_protocol_handlers(my_handlers)
61
 
            register_lazy_transport('foo', 'bzrlib.tests.test_transport', 'TestTransport.SampleHandler')
62
 
            register_lazy_transport('bar', 'bzrlib.tests.test_transport', 'TestTransport.SampleHandler')
63
 
            self.assertEqual([SampleHandler.__module__],
64
 
                             _get_transport_modules())
65
 
        finally:
66
 
            _set_protocol_handlers(handlers)
67
 
 
68
 
    def test_transport_dependency(self):
69
 
        """Transport with missing dependency causes no error"""
70
 
        saved_handlers = _get_protocol_handlers()
71
 
        try:
72
 
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
73
 
                    'BadTransportHandler')
74
 
            try:
75
 
                get_transport('foo://fooserver/foo')
76
 
            except UnsupportedProtocol, e:
77
 
                e_str = str(e)
78
 
                self.assertEquals('Unsupported protocol'
79
 
                                  ' for url "foo://fooserver/foo":'
80
 
                                  ' Unable to import library "some_lib":'
81
 
                                  ' testing missing dependency', str(e))
82
 
            else:
83
 
                self.fail('Did not raise UnsupportedProtocol')
84
 
        finally:
85
 
            # restore original values
86
 
            _set_protocol_handlers(saved_handlers)
87
 
            
88
 
    def test_transport_fallback(self):
89
 
        """Transport with missing dependency causes no error"""
90
 
        saved_handlers = _get_protocol_handlers()
91
 
        try:
92
 
            _set_protocol_handlers({})
93
 
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
94
 
                    'BackupTransportHandler')
95
 
            register_lazy_transport('foo', 'bzrlib.tests.test_transport',
96
 
                    'BadTransportHandler')
97
 
            t = get_transport('foo://fooserver/foo')
98
 
            self.assertTrue(isinstance(t, BackupTransportHandler))
99
 
        finally:
100
 
            _set_protocol_handlers(saved_handlers)
101
 
            
 
39
    def test_urlescape(self):
 
40
        self.assertEqual('%25', urlescape('%'))
 
41
 
 
42
 
 
43
class TestTransportMixIn(object):
 
44
    """Subclass this, and it will provide a series of tests for a Transport.
 
45
    It assumes that the Transport object is connected to the 
 
46
    current working directory.  So that whatever is done 
 
47
    through the transport, should show up in the working 
 
48
    directory, and vice-versa.
 
49
 
 
50
    This also tests to make sure that the functions work with both
 
51
    generators and lists (assuming iter(list) is effectively a generator)
 
52
    """
 
53
    readonly = False
 
54
    def get_transport(self):
 
55
        """Children should override this to return the Transport object.
 
56
        """
 
57
        raise NotImplementedError
 
58
 
 
59
    def test_has(self):
 
60
        t = self.get_transport()
 
61
 
 
62
        files = ['a', 'b', 'e', 'g', '%']
 
63
        self.build_tree(files)
 
64
        self.assertEqual(t.has('a'), True)
 
65
        self.assertEqual(t.has('c'), False)
 
66
        self.assertEqual(t.has(urlescape('%')), True)
 
67
        self.assertEqual(list(t.has_multi(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])),
 
68
                [True, True, False, False, True, False, True, False])
 
69
        self.assertEqual(t.has_any(['a', 'b', 'c']), True)
 
70
        self.assertEqual(t.has_any(['c', 'd', 'f', urlescape('%%')]), False)
 
71
        self.assertEqual(list(t.has_multi(iter(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']))),
 
72
                [True, True, False, False, True, False, True, False])
 
73
        self.assertEqual(t.has_any(['c', 'c', 'c']), False)
 
74
        self.assertEqual(t.has_any(['b', 'b', 'b']), True)
 
75
 
 
76
    def test_get(self):
 
77
        t = self.get_transport()
 
78
 
 
79
        files = ['a', 'b', 'e', 'g']
 
80
        self.build_tree(files)
 
81
        self.assertEqual(t.get('a').read(), open('a').read())
 
82
        content_f = t.get_multi(files)
 
83
        for path,f in zip(files, content_f):
 
84
            self.assertEqual(open(path).read(), f.read())
 
85
 
 
86
        content_f = t.get_multi(iter(files))
 
87
        for path,f in zip(files, content_f):
 
88
            self.assertEqual(open(path).read(), f.read())
 
89
 
 
90
        self.assertRaises(NoSuchFile, t.get, 'c')
 
91
        try:
 
92
            files = list(t.get_multi(['a', 'b', 'c']))
 
93
        except NoSuchFile:
 
94
            pass
 
95
        else:
 
96
            self.fail('Failed to raise NoSuchFile for missing file in get_multi')
 
97
        try:
 
98
            files = list(t.get_multi(iter(['a', 'b', 'c', 'e'])))
 
99
        except NoSuchFile:
 
100
            pass
 
101
        else:
 
102
            self.fail('Failed to raise NoSuchFile for missing file in get_multi')
 
103
 
 
104
    def test_put(self):
 
105
        t = self.get_transport()
 
106
 
 
107
        if self.readonly:
 
108
            self.assertRaises(TransportNotPossible,
 
109
                    t.put, 'a', 'some text for a\n')
 
110
            open('a', 'wb').write('some text for a\n')
 
111
        else:
 
112
            t.put('a', 'some text for a\n')
 
113
        self.assert_(os.path.exists('a'))
 
114
        self.check_file_contents('a', 'some text for a\n')
 
115
        self.assertEqual(t.get('a').read(), 'some text for a\n')
 
116
        # Make sure 'has' is updated
 
117
        self.assertEqual(list(t.has_multi(['a', 'b', 'c', 'd', 'e'])),
 
118
                [True, False, False, False, False])
 
119
        if self.readonly:
 
120
            self.assertRaises(TransportNotPossible,
 
121
                    t.put_multi,
 
122
                    [('a', 'new\ncontents for\na\n'),
 
123
                        ('d', 'contents\nfor d\n')])
 
124
            open('a', 'wb').write('new\ncontents for\na\n')
 
125
            open('d', 'wb').write('contents\nfor d\n')
 
126
        else:
 
127
            # Put also replaces contents
 
128
            self.assertEqual(t.put_multi([('a', 'new\ncontents for\na\n'),
 
129
                                          ('d', 'contents\nfor d\n')]),
 
130
                             2)
 
131
        self.assertEqual(list(t.has_multi(['a', 'b', 'c', 'd', 'e'])),
 
132
                [True, False, False, True, False])
 
133
        self.check_file_contents('a', 'new\ncontents for\na\n')
 
134
        self.check_file_contents('d', 'contents\nfor d\n')
 
135
 
 
136
        if self.readonly:
 
137
            self.assertRaises(TransportNotPossible,
 
138
                t.put_multi, iter([('a', 'diff\ncontents for\na\n'),
 
139
                                  ('d', 'another contents\nfor d\n')]))
 
140
            open('a', 'wb').write('diff\ncontents for\na\n')
 
141
            open('d', 'wb').write('another contents\nfor d\n')
 
142
        else:
 
143
            self.assertEqual(
 
144
                t.put_multi(iter([('a', 'diff\ncontents for\na\n'),
 
145
                                  ('d', 'another contents\nfor d\n')]))
 
146
                             , 2)
 
147
        self.check_file_contents('a', 'diff\ncontents for\na\n')
 
148
        self.check_file_contents('d', 'another contents\nfor d\n')
 
149
 
 
150
        if self.readonly:
 
151
            self.assertRaises(TransportNotPossible,
 
152
                    t.put, 'path/doesnt/exist/c', 'contents')
 
153
        else:
 
154
            self.assertRaises(NoSuchFile,
 
155
                    t.put, 'path/doesnt/exist/c', 'contents')
 
156
 
 
157
    def test_put_file(self):
 
158
        t = self.get_transport()
 
159
 
 
160
        # Test that StringIO can be used as a file-like object with put
 
161
        f1 = StringIO('this is a string\nand some more stuff\n')
 
162
        if self.readonly:
 
163
            open('f1', 'wb').write(f1.read())
 
164
        else:
 
165
            t.put('f1', f1)
 
166
 
 
167
        del f1
 
168
 
 
169
        self.check_file_contents('f1', 
 
170
                'this is a string\nand some more stuff\n')
 
171
 
 
172
        f2 = StringIO('here is some text\nand a bit more\n')
 
173
        f3 = StringIO('some text for the\nthird file created\n')
 
174
 
 
175
        if self.readonly:
 
176
            open('f2', 'wb').write(f2.read())
 
177
            open('f3', 'wb').write(f3.read())
 
178
        else:
 
179
            t.put_multi([('f2', f2), ('f3', f3)])
 
180
 
 
181
        del f2, f3
 
182
 
 
183
        self.check_file_contents('f2', 'here is some text\nand a bit more\n')
 
184
        self.check_file_contents('f3', 'some text for the\nthird file created\n')
 
185
 
 
186
        # Test that an actual file object can be used with put
 
187
        f4 = open('f1', 'rb')
 
188
        if self.readonly:
 
189
            open('f4', 'wb').write(f4.read())
 
190
        else:
 
191
            t.put('f4', f4)
 
192
 
 
193
        del f4
 
194
 
 
195
        self.check_file_contents('f4', 
 
196
                'this is a string\nand some more stuff\n')
 
197
 
 
198
        f5 = open('f2', 'rb')
 
199
        f6 = open('f3', 'rb')
 
200
        if self.readonly:
 
201
            open('f5', 'wb').write(f5.read())
 
202
            open('f6', 'wb').write(f6.read())
 
203
        else:
 
204
            t.put_multi([('f5', f5), ('f6', f6)])
 
205
 
 
206
        del f5, f6
 
207
 
 
208
        self.check_file_contents('f5', 'here is some text\nand a bit more\n')
 
209
        self.check_file_contents('f6', 'some text for the\nthird file created\n')
 
210
 
 
211
 
 
212
 
 
213
    def test_mkdir(self):
 
214
        t = self.get_transport()
 
215
 
 
216
        # Test mkdir
 
217
        os.mkdir('dir_a')
 
218
        self.assertEqual(t.has('dir_a'), True)
 
219
        self.assertEqual(t.has('dir_b'), False)
 
220
 
 
221
        if self.readonly:
 
222
            self.assertRaises(TransportNotPossible,
 
223
                    t.mkdir, 'dir_b')
 
224
            os.mkdir('dir_b')
 
225
        else:
 
226
            t.mkdir('dir_b')
 
227
        self.assertEqual(t.has('dir_b'), True)
 
228
        self.assert_(os.path.isdir('dir_b'))
 
229
 
 
230
        if self.readonly:
 
231
            self.assertRaises(TransportNotPossible,
 
232
                    t.mkdir_multi, ['dir_c', 'dir_d'])
 
233
            os.mkdir('dir_c')
 
234
            os.mkdir('dir_d')
 
235
        else:
 
236
            t.mkdir_multi(['dir_c', 'dir_d'])
 
237
 
 
238
        if self.readonly:
 
239
            self.assertRaises(TransportNotPossible,
 
240
                    t.mkdir_multi, iter(['dir_e', 'dir_f']))
 
241
            os.mkdir('dir_e')
 
242
            os.mkdir('dir_f')
 
243
        else:
 
244
            t.mkdir_multi(iter(['dir_e', 'dir_f']))
 
245
        self.assertEqual(list(t.has_multi(
 
246
            ['dir_a', 'dir_b', 'dir_c', 'dir_q',
 
247
             'dir_d', 'dir_e', 'dir_f', 'dir_b'])),
 
248
            [True, True, True, False,
 
249
             True, True, True, True])
 
250
        for d in ['dir_a', 'dir_b', 'dir_c', 'dir_d', 'dir_e', 'dir_f']:
 
251
            self.assert_(os.path.isdir(d))
 
252
 
 
253
        if not self.readonly:
 
254
            self.assertRaises(NoSuchFile, t.mkdir, 'path/doesnt/exist')
 
255
            self.assertRaises(FileExists, t.mkdir, 'dir_a') # Creating a directory again should fail
 
256
 
 
257
        # Make sure the transport recognizes when a
 
258
        # directory is created by other means
 
259
        # Caching Transports will fail, because dir_e was already seen not
 
260
        # to exist. So instead, we will search for a new directory
 
261
        #os.mkdir('dir_e')
 
262
        #if not self.readonly:
 
263
        #    self.assertRaises(FileExists, t.mkdir, 'dir_e')
 
264
 
 
265
        os.mkdir('dir_g')
 
266
        if not self.readonly:
 
267
            self.assertRaises(FileExists, t.mkdir, 'dir_g')
 
268
 
 
269
        # Test get/put in sub-directories
 
270
        if self.readonly:
 
271
            open('dir_a/a', 'wb').write('contents of dir_a/a')
 
272
            open('dir_b/b', 'wb').write('contents of dir_b/b')
 
273
        else:
 
274
            self.assertEqual(
 
275
                t.put_multi([('dir_a/a', 'contents of dir_a/a'),
 
276
                             ('dir_b/b', 'contents of dir_b/b')])
 
277
                          , 2)
 
278
        for f in ('dir_a/a', 'dir_b/b'):
 
279
            self.assertEqual(t.get(f).read(), open(f).read())
 
280
 
 
281
    def test_copy_to(self):
 
282
        import tempfile
 
283
        from bzrlib.transport.local import LocalTransport
 
284
 
 
285
        t = self.get_transport()
 
286
 
 
287
        files = ['a', 'b', 'c', 'd']
 
288
        self.build_tree(files)
 
289
 
 
290
        dtmp = tempfile.mkdtemp(dir='.', prefix='test-transport-')
 
291
        dtmp_base = os.path.basename(dtmp)
 
292
        local_t = LocalTransport(dtmp)
 
293
 
 
294
        t.copy_to(files, local_t)
 
295
        for f in files:
 
296
            self.assertEquals(open(f).read(),
 
297
                    open(os.path.join(dtmp_base, f)).read())
 
298
 
 
299
        # Test that copying into a missing directory raises
 
300
        # NoSuchFile
 
301
        os.mkdir('e')
 
302
        open('e/f', 'wb').write('contents of e')
 
303
        self.assertRaises(NoSuchFile, t.copy_to, ['e/f'], local_t)
 
304
 
 
305
        os.mkdir(os.path.join(dtmp_base, 'e'))
 
306
        t.copy_to(['e/f'], local_t)
 
307
 
 
308
        del dtmp, dtmp_base, local_t
 
309
 
 
310
        dtmp = tempfile.mkdtemp(dir='.', prefix='test-transport-')
 
311
        dtmp_base = os.path.basename(dtmp)
 
312
        local_t = LocalTransport(dtmp)
 
313
 
 
314
        files = ['a', 'b', 'c', 'd']
 
315
        t.copy_to(iter(files), local_t)
 
316
        for f in files:
 
317
            self.assertEquals(open(f).read(),
 
318
                    open(os.path.join(dtmp_base, f)).read())
 
319
 
 
320
        del dtmp, dtmp_base, local_t
 
321
 
 
322
    def test_append(self):
 
323
        t = self.get_transport()
 
324
 
 
325
        if self.readonly:
 
326
            open('a', 'wb').write('diff\ncontents for\na\n')
 
327
            open('b', 'wb').write('contents\nfor b\n')
 
328
        else:
 
329
            t.put_multi([
 
330
                    ('a', 'diff\ncontents for\na\n'),
 
331
                    ('b', 'contents\nfor b\n')
 
332
                    ])
 
333
 
 
334
        if self.readonly:
 
335
            self.assertRaises(TransportNotPossible,
 
336
                    t.append, 'a', 'add\nsome\nmore\ncontents\n')
 
337
            _append('a', 'add\nsome\nmore\ncontents\n')
 
338
        else:
 
339
            t.append('a', 'add\nsome\nmore\ncontents\n')
 
340
 
 
341
        self.check_file_contents('a', 
 
342
            'diff\ncontents for\na\nadd\nsome\nmore\ncontents\n')
 
343
 
 
344
        if self.readonly:
 
345
            self.assertRaises(TransportNotPossible,
 
346
                    t.append_multi,
 
347
                        [('a', 'and\nthen\nsome\nmore\n'),
 
348
                         ('b', 'some\nmore\nfor\nb\n')])
 
349
            _append('a', 'and\nthen\nsome\nmore\n')
 
350
            _append('b', 'some\nmore\nfor\nb\n')
 
351
        else:
 
352
            t.append_multi([('a', 'and\nthen\nsome\nmore\n'),
 
353
                    ('b', 'some\nmore\nfor\nb\n')])
 
354
        self.check_file_contents('a', 
 
355
            'diff\ncontents for\na\n'
 
356
            'add\nsome\nmore\ncontents\n'
 
357
            'and\nthen\nsome\nmore\n')
 
358
        self.check_file_contents('b', 
 
359
                'contents\nfor b\n'
 
360
                'some\nmore\nfor\nb\n')
 
361
 
 
362
        if self.readonly:
 
363
            _append('a', 'a little bit more\n')
 
364
            _append('b', 'from an iterator\n')
 
365
        else:
 
366
            t.append_multi(iter([('a', 'a little bit more\n'),
 
367
                    ('b', 'from an iterator\n')]))
 
368
        self.check_file_contents('a', 
 
369
            'diff\ncontents for\na\n'
 
370
            'add\nsome\nmore\ncontents\n'
 
371
            'and\nthen\nsome\nmore\n'
 
372
            'a little bit more\n')
 
373
        self.check_file_contents('b', 
 
374
                'contents\nfor b\n'
 
375
                'some\nmore\nfor\nb\n'
 
376
                'from an iterator\n')
 
377
 
 
378
    def test_append_file(self):
 
379
        t = self.get_transport()
 
380
 
 
381
        contents = [
 
382
            ('f1', 'this is a string\nand some more stuff\n'),
 
383
            ('f2', 'here is some text\nand a bit more\n'),
 
384
            ('f3', 'some text for the\nthird file created\n'),
 
385
            ('f4', 'this is a string\nand some more stuff\n'),
 
386
            ('f5', 'here is some text\nand a bit more\n'),
 
387
            ('f6', 'some text for the\nthird file created\n')
 
388
        ]
 
389
        
 
390
        if self.readonly:
 
391
            for f, val in contents:
 
392
                open(f, 'wb').write(val)
 
393
        else:
 
394
            t.put_multi(contents)
 
395
 
 
396
        a1 = StringIO('appending to\none\n')
 
397
        if self.readonly:
 
398
            _append('f1', a1.read())
 
399
        else:
 
400
            t.append('f1', a1)
 
401
 
 
402
        del a1
 
403
 
 
404
        self.check_file_contents('f1', 
 
405
                'this is a string\nand some more stuff\n'
 
406
                'appending to\none\n')
 
407
 
 
408
        a2 = StringIO('adding more\ntext to two\n')
 
409
        a3 = StringIO('some garbage\nto put in three\n')
 
410
 
 
411
        if self.readonly:
 
412
            _append('f2', a2.read())
 
413
            _append('f3', a3.read())
 
414
        else:
 
415
            t.append_multi([('f2', a2), ('f3', a3)])
 
416
 
 
417
        del a2, a3
 
418
 
 
419
        self.check_file_contents('f2',
 
420
                'here is some text\nand a bit more\n'
 
421
                'adding more\ntext to two\n')
 
422
        self.check_file_contents('f3', 
 
423
                'some text for the\nthird file created\n'
 
424
                'some garbage\nto put in three\n')
 
425
 
 
426
        # Test that an actual file object can be used with put
 
427
        a4 = open('f1', 'rb')
 
428
        if self.readonly:
 
429
            _append('f4', a4.read())
 
430
        else:
 
431
            t.append('f4', a4)
 
432
 
 
433
        del a4
 
434
 
 
435
        self.check_file_contents('f4', 
 
436
                'this is a string\nand some more stuff\n'
 
437
                'this is a string\nand some more stuff\n'
 
438
                'appending to\none\n')
 
439
 
 
440
        a5 = open('f2', 'rb')
 
441
        a6 = open('f3', 'rb')
 
442
        if self.readonly:
 
443
            _append('f5', a5.read())
 
444
            _append('f6', a6.read())
 
445
        else:
 
446
            t.append_multi([('f5', a5), ('f6', a6)])
 
447
 
 
448
        del a5, a6
 
449
 
 
450
        self.check_file_contents('f5',
 
451
                'here is some text\nand a bit more\n'
 
452
                'here is some text\nand a bit more\n'
 
453
                'adding more\ntext to two\n')
 
454
        self.check_file_contents('f6',
 
455
                'some text for the\nthird file created\n'
 
456
                'some text for the\nthird file created\n'
 
457
                'some garbage\nto put in three\n')
 
458
 
 
459
    def test_delete(self):
 
460
        # TODO: Test Transport.delete
 
461
        pass
 
462
 
 
463
    def test_move(self):
 
464
        # TODO: Test Transport.move
 
465
        pass
 
466
 
 
467
    def test_connection_error(self):
 
468
        """ConnectionError is raised when connection is impossible"""
 
469
        if not hasattr(self, "get_bogus_transport"):
 
470
            return
 
471
        t = self.get_bogus_transport()
 
472
        self.assertRaises(ConnectionError, t.get, '.bzr/branch')
 
473
 
 
474
        
 
475
class LocalTransportTest(TestCaseInTempDir, TestTransportMixIn):
 
476
    def get_transport(self):
 
477
        from bzrlib.transport.local import LocalTransport
 
478
        return LocalTransport('.')
 
479
 
 
480
 
 
481
class HttpTransportTest(TestCaseWithWebserver, TestTransportMixIn):
 
482
 
 
483
    readonly = True
 
484
 
 
485
    def get_transport(self):
 
486
        from bzrlib.transport.http import HttpTransport
 
487
        url = self.get_remote_url('.')
 
488
        return HttpTransport(url)
 
489
 
 
490
    def get_bogus_transport(self):
 
491
        from bzrlib.transport.http import HttpTransport
 
492
        return HttpTransport('http://jasldkjsalkdjalksjdkljasd')
 
493
 
102
494
 
103
495
class TestMemoryTransport(TestCase):
104
496
 
105
497
    def test_get_transport(self):
106
 
        MemoryTransport()
 
498
        memory.MemoryTransport()
107
499
 
108
500
    def test_clone(self):
109
 
        transport = MemoryTransport()
110
 
        self.assertTrue(isinstance(transport, MemoryTransport))
 
501
        transport = memory.MemoryTransport()
 
502
        self.failUnless(transport.clone() is transport)
111
503
 
112
504
    def test_abspath(self):
113
 
        transport = MemoryTransport()
114
 
        self.assertEqual("memory:///relpath", transport.abspath('relpath'))
 
505
        transport = memory.MemoryTransport()
 
506
        self.assertEqual("in-memory:relpath", transport.abspath('relpath'))
115
507
 
116
508
    def test_relpath(self):
117
 
        transport = MemoryTransport()
 
509
        transport = memory.MemoryTransport()
118
510
 
119
511
    def test_append_and_get(self):
120
 
        transport = MemoryTransport()
 
512
        transport = memory.MemoryTransport()
121
513
        transport.append('path', StringIO('content'))
122
514
        self.assertEqual(transport.get('path').read(), 'content')
123
515
        transport.append('path', StringIO('content'))
124
516
        self.assertEqual(transport.get('path').read(), 'contentcontent')
125
517
 
126
518
    def test_put_and_get(self):
127
 
        transport = MemoryTransport()
 
519
        transport = memory.MemoryTransport()
128
520
        transport.put('path', StringIO('content'))
129
521
        self.assertEqual(transport.get('path').read(), 'content')
130
522
        transport.put('path', StringIO('content'))
131
523
        self.assertEqual(transport.get('path').read(), 'content')
132
524
 
133
525
    def test_append_without_dir_fails(self):
134
 
        transport = MemoryTransport()
 
526
        transport = memory.MemoryTransport()
135
527
        self.assertRaises(NoSuchFile,
136
528
                          transport.append, 'dir/path', StringIO('content'))
137
529
 
138
530
    def test_put_without_dir_fails(self):
139
 
        transport = MemoryTransport()
 
531
        transport = memory.MemoryTransport()
140
532
        self.assertRaises(NoSuchFile,
141
533
                          transport.put, 'dir/path', StringIO('content'))
142
534
 
143
535
    def test_get_missing(self):
144
 
        transport = MemoryTransport()
 
536
        transport = memory.MemoryTransport()
145
537
        self.assertRaises(NoSuchFile, transport.get, 'foo')
146
538
 
147
539
    def test_has_missing(self):
148
 
        transport = MemoryTransport()
 
540
        transport = memory.MemoryTransport()
149
541
        self.assertEquals(False, transport.has('foo'))
150
542
 
151
543
    def test_has_present(self):
152
 
        transport = MemoryTransport()
 
544
        transport = memory.MemoryTransport()
153
545
        transport.append('foo', StringIO('content'))
154
546
        self.assertEquals(True, transport.has('foo'))
155
547
 
156
548
    def test_mkdir(self):
157
 
        transport = MemoryTransport()
 
549
        transport = memory.MemoryTransport()
158
550
        transport.mkdir('dir')
159
551
        transport.append('dir/path', StringIO('content'))
160
552
        self.assertEqual(transport.get('dir/path').read(), 'content')
161
553
 
162
554
    def test_mkdir_missing_parent(self):
163
 
        transport = MemoryTransport()
 
555
        transport = memory.MemoryTransport()
164
556
        self.assertRaises(NoSuchFile,
165
557
                          transport.mkdir, 'dir/dir')
166
558
 
167
559
    def test_mkdir_twice(self):
168
 
        transport = MemoryTransport()
 
560
        transport = memory.MemoryTransport()
169
561
        transport.mkdir('dir')
170
562
        self.assertRaises(FileExists, transport.mkdir, 'dir')
171
563
 
172
564
    def test_parameters(self):
173
 
        transport = MemoryTransport()
 
565
        transport = memory.MemoryTransport()
174
566
        self.assertEqual(True, transport.listable())
175
567
        self.assertEqual(False, transport.should_cache())
176
 
        self.assertEqual(False, transport.is_readonly())
177
568
 
178
569
    def test_iter_files_recursive(self):
179
 
        transport = MemoryTransport()
 
570
        transport = memory.MemoryTransport()
180
571
        transport.mkdir('dir')
181
572
        transport.put('dir/foo', StringIO('content'))
182
573
        transport.put('dir/bar', StringIO('content'))
185
576
        self.assertEqual(set(['dir/foo', 'dir/bar', 'bar']), paths)
186
577
 
187
578
    def test_stat(self):
188
 
        transport = MemoryTransport()
 
579
        transport = memory.MemoryTransport()
189
580
        transport.put('foo', StringIO('content'))
190
581
        transport.put('bar', StringIO('phowar'))
191
582
        self.assertEqual(7, transport.stat('foo').st_size)
192
583
        self.assertEqual(6, transport.stat('bar').st_size)
193
584
 
194
 
        
195
 
class ReadonlyDecoratorTransportTest(TestCase):
196
 
    """Readonly decoration specific tests."""
197
 
 
198
 
    def test_local_parameters(self):
199
 
        import bzrlib.transport.readonly as readonly
200
 
        # connect to . in readonly mode
201
 
        transport = readonly.ReadonlyTransportDecorator('readonly+.')
202
 
        self.assertEqual(True, transport.listable())
203
 
        self.assertEqual(False, transport.should_cache())
204
 
        self.assertEqual(True, transport.is_readonly())
205
 
 
206
 
    def test_http_parameters(self):
207
 
        import bzrlib.transport.readonly as readonly
208
 
        from bzrlib.transport.http import HttpServer
209
 
        # connect to . via http which is not listable
210
 
        server = HttpServer()
211
 
        server.setUp()
212
 
        try:
213
 
            transport = get_transport('readonly+' + server.get_url())
214
 
            self.failUnless(isinstance(transport,
215
 
                                       readonly.ReadonlyTransportDecorator))
216
 
            self.assertEqual(False, transport.listable())
217
 
            self.assertEqual(True, transport.should_cache())
218
 
            self.assertEqual(True, transport.is_readonly())
219
 
        finally:
220
 
            server.tearDown()
221
 
 
222
 
 
223
 
class FakeNFSDecoratorTests(TestCaseInTempDir):
224
 
    """NFS decorator specific tests."""
225
 
 
226
 
    def get_nfs_transport(self, url):
227
 
        import bzrlib.transport.fakenfs as fakenfs
228
 
        # connect to url with nfs decoration
229
 
        return fakenfs.FakeNFSTransportDecorator('fakenfs+' + url)
230
 
 
231
 
    def test_local_parameters(self):
232
 
        # the listable, should_cache and is_readonly parameters
233
 
        # are not changed by the fakenfs decorator
234
 
        transport = self.get_nfs_transport('.')
235
 
        self.assertEqual(True, transport.listable())
236
 
        self.assertEqual(False, transport.should_cache())
237
 
        self.assertEqual(False, transport.is_readonly())
238
 
 
239
 
    def test_http_parameters(self):
240
 
        # the listable, should_cache and is_readonly parameters
241
 
        # are not changed by the fakenfs decorator
242
 
        from bzrlib.transport.http import HttpServer
243
 
        # connect to . via http which is not listable
244
 
        server = HttpServer()
245
 
        server.setUp()
246
 
        try:
247
 
            transport = self.get_nfs_transport(server.get_url())
248
 
            self.assertIsInstance(
249
 
                transport, bzrlib.transport.fakenfs.FakeNFSTransportDecorator)
250
 
            self.assertEqual(False, transport.listable())
251
 
            self.assertEqual(True, transport.should_cache())
252
 
            self.assertEqual(True, transport.is_readonly())
253
 
        finally:
254
 
            server.tearDown()
255
 
 
256
 
    def test_fakenfs_server_default(self):
257
 
        # a FakeNFSServer() should bring up a local relpath server for itself
258
 
        import bzrlib.transport.fakenfs as fakenfs
259
 
        server = fakenfs.FakeNFSServer()
260
 
        server.setUp()
261
 
        try:
262
 
            # the server should be a relpath localhost server
263
 
            self.assertEqual(server.get_url(), 'fakenfs+.')
264
 
            # and we should be able to get a transport for it
265
 
            transport = get_transport(server.get_url())
266
 
            # which must be a FakeNFSTransportDecorator instance.
267
 
            self.assertIsInstance(
268
 
                transport, fakenfs.FakeNFSTransportDecorator)
269
 
        finally:
270
 
            server.tearDown()
271
 
 
272
 
    def test_fakenfs_rename_semantics(self):
273
 
        # a FakeNFS transport must mangle the way rename errors occur to
274
 
        # look like NFS problems.
275
 
        transport = self.get_nfs_transport('.')
276
 
        self.build_tree(['from/', 'from/foo', 'to/', 'to/bar'],
277
 
                        transport=transport)
278
 
        self.assertRaises(bzrlib.errors.ResourceBusy,
279
 
                          transport.rename, 'from', 'to')
280
 
 
281
 
 
282
 
class FakeVFATDecoratorTests(TestCaseInTempDir):
283
 
    """Tests for simulation of VFAT restrictions"""
284
 
 
285
 
    def get_vfat_transport(self, url):
286
 
        """Return vfat-backed transport for test directory"""
287
 
        from bzrlib.transport.fakevfat import FakeVFATTransportDecorator
288
 
        return FakeVFATTransportDecorator('vfat+' + url)
289
 
 
290
 
    def test_transport_creation(self):
291
 
        from bzrlib.transport.fakevfat import FakeVFATTransportDecorator
292
 
        transport = self.get_vfat_transport('.')
293
 
        self.assertIsInstance(transport, FakeVFATTransportDecorator)
294
 
 
295
 
    def test_transport_mkdir(self):
296
 
        transport = self.get_vfat_transport('.')
297
 
        transport.mkdir('HELLO')
298
 
        self.assertTrue(transport.has('hello'))
299
 
        self.assertTrue(transport.has('Hello'))
300
 
 
301
 
    def test_forbidden_chars(self):
302
 
        transport = self.get_vfat_transport('.')
303
 
        self.assertRaises(ValueError, transport.has, "<NU>")
304
 
 
305
 
 
306
 
class BadTransportHandler(Transport):
307
 
    def __init__(self, base_url):
308
 
        raise DependencyNotPresent('some_lib', 'testing missing dependency')
309
 
 
310
 
 
311
 
class BackupTransportHandler(Transport):
312
 
    """Test transport that works as a backup for the BadTransportHandler"""
313
 
    pass
314
 
 
315
 
 
316
 
class TestTransportImplementation(TestCaseInTempDir):
317
 
    """Implementation verification for transports.
318
 
    
319
 
    To verify a transport we need a server factory, which is a callable
320
 
    that accepts no parameters and returns an implementation of
321
 
    bzrlib.transport.Server.
322
 
    
323
 
    That Server is then used to construct transport instances and test
324
 
    the transport via loopback activity.
325
 
 
326
 
    Currently this assumes that the Transport object is connected to the 
327
 
    current working directory.  So that whatever is done 
328
 
    through the transport, should show up in the working 
329
 
    directory, and vice-versa. This is a bug, because its possible to have
330
 
    URL schemes which provide access to something that may not be 
331
 
    result in storage on the local disk, i.e. due to file system limits, or 
332
 
    due to it being a database or some other non-filesystem tool.
333
 
 
334
 
    This also tests to make sure that the functions work with both
335
 
    generators and lists (assuming iter(list) is effectively a generator)
336
 
    """
337
 
    
338
 
    def setUp(self):
339
 
        super(TestTransportImplementation, self).setUp()
340
 
        self._server = self.transport_server()
341
 
        self._server.setUp()
342
 
 
343
 
    def tearDown(self):
344
 
        super(TestTransportImplementation, self).tearDown()
345
 
        self._server.tearDown()
346
 
        
347
 
    def get_transport(self):
348
 
        """Return a connected transport to the local directory."""
349
 
        base_url = self._server.get_url()
350
 
        # try getting the transport via the regular interface:
351
 
        t = get_transport(base_url)
352
 
        if not isinstance(t, self.transport_class): 
353
 
            # we did not get the correct transport class type. Override the
354
 
            # regular connection behaviour by direct construction.
355
 
            t = self.transport_class(base_url)
356
 
        return t