57
75
raise NotImplementedError
77
def assertListRaises(self, excClass, func, *args, **kwargs):
78
"""Many transport functions can return generators this makes sure
79
to wrap them in a list() call to make sure the whole generator
80
is run, and that the proper exception is raised.
83
list(func(*args, **kwargs))
87
if hasattr(excClass,'__name__'): excName = excClass.__name__
88
else: excName = str(excClass)
89
raise self.failureException, "%s not raised" % excName
59
91
def test_has(self):
60
92
t = self.get_transport()
62
94
files = ['a', 'b', 'e', 'g', '%']
63
95
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)
96
self.assertEqual(True, t.has('a'))
97
self.assertEqual(False, t.has('c'))
98
self.assertEqual(True, t.has(urlescape('%')))
67
99
self.assertEqual(list(t.has_multi(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])),
68
100
[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)
101
self.assertEqual(True, t.has_any(['a', 'b', 'c']))
102
self.assertEqual(False, t.has_any(['c', 'd', 'f', urlescape('%%')]))
71
103
self.assertEqual(list(t.has_multi(iter(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']))),
72
104
[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)
105
self.assertEqual(False, t.has_any(['c', 'c', 'c']))
106
self.assertEqual(True, t.has_any(['b', 'b', 'b']))
76
108
def test_get(self):
77
109
t = self.get_transport()
79
111
files = ['a', 'b', 'e', 'g']
80
112
self.build_tree(files)
81
self.assertEqual(t.get('a').read(), open('a').read())
113
self.assertEqual(open('a', 'rb').read(), t.get('a').read())
82
114
content_f = t.get_multi(files)
83
115
for path,f in zip(files, content_f):
84
self.assertEqual(open(path).read(), f.read())
116
self.assertEqual(f.read(), open(path, 'rb').read())
86
118
content_f = t.get_multi(iter(files))
87
119
for path,f in zip(files, content_f):
88
self.assertEqual(open(path).read(), f.read())
120
self.assertEqual(f.read(), open(path, 'rb').read())
90
122
self.assertRaises(NoSuchFile, t.get, 'c')
92
files = list(t.get_multi(['a', 'b', 'c']))
96
self.fail('Failed to raise NoSuchFile for missing file in get_multi')
98
files = list(t.get_multi(iter(['a', 'b', 'c', 'e'])))
102
self.fail('Failed to raise NoSuchFile for missing file in get_multi')
123
self.assertListRaises(NoSuchFile, t.get_multi, ['a', 'b', 'c'])
124
self.assertListRaises(NoSuchFile, t.get_multi, iter(['a', 'b', 'c']))
104
126
def test_put(self):
105
127
t = self.get_transport()
129
# TODO: jam 20051215 No need to do anything if the test is readonly
130
# origininally it was thought that it would give
131
# more of a workout to readonly tests. By now the
132
# suite is probably thorough enough without testing
133
# readonly protocols in write sections
134
# The only thing that needs to be tested is that the
135
# right error is raised
107
137
if self.readonly:
108
138
self.assertRaises(TransportNotPossible,
109
139
t.put, 'a', 'some text for a\n')
302
389
open('e/f', 'wb').write('contents of e')
303
390
self.assertRaises(NoSuchFile, t.copy_to, ['e/f'], local_t)
305
os.mkdir(os.path.join(dtmp_base, 'e'))
392
os.mkdir(pathjoin(dtmp_base, 'e'))
306
393
t.copy_to(['e/f'], local_t)
308
del dtmp, dtmp_base, local_t
395
del dtmp_base, local_t
310
dtmp = tempfile.mkdtemp(dir='.', prefix='test-transport-')
311
dtmp_base = os.path.basename(dtmp)
312
local_t = LocalTransport(dtmp)
397
dtmp_base, local_t = get_temp_local()
314
399
files = ['a', 'b', 'c', 'd']
315
400
t.copy_to(iter(files), local_t)
317
self.assertEquals(open(f).read(),
318
open(os.path.join(dtmp_base, f)).read())
320
del dtmp, dtmp_base, local_t
402
self.assertEquals(open(f, 'rb').read(),
403
open(pathjoin(dtmp_base, f), 'rb').read())
405
del dtmp_base, local_t
407
for mode in (0666, 0644, 0600, 0400):
408
dtmp_base, local_t = get_temp_local()
409
t.copy_to(files, local_t, mode=mode)
411
check_mode(self, os.path.join(dtmp_base, f), mode)
322
413
def test_append(self):
323
414
t = self.get_transport()
456
564
'some text for the\nthird file created\n'
457
565
'some garbage\nto put in three\n')
567
a5 = open('f2', 'rb')
568
a6 = open('f2', 'rb')
569
a7 = open('f3', 'rb')
571
_append('c', a5.read())
572
_append('a', a6.read())
573
_append('d', a7.read())
576
t.append_multi([('a', a6), ('d', a7)])
578
self.check_file_contents('c', open('f2', 'rb').read())
579
self.check_file_contents('d', open('f3', 'rb').read())
459
582
def test_delete(self):
460
583
# TODO: Test Transport.delete
584
t = self.get_transport()
586
# Not much to do with a readonly transport
590
open('a', 'wb').write('a little bit of text\n')
591
self.failUnless(t.has('a'))
592
self.failUnlessExists('a')
594
self.failIf(os.path.lexists('a'))
596
self.assertRaises(NoSuchFile, t.delete, 'a')
598
open('a', 'wb').write('a text\n')
599
open('b', 'wb').write('b text\n')
600
open('c', 'wb').write('c text\n')
601
self.assertEqual([True, True, True],
602
list(t.has_multi(['a', 'b', 'c'])))
603
t.delete_multi(['a', 'c'])
604
self.assertEqual([False, True, False],
605
list(t.has_multi(['a', 'b', 'c'])))
606
self.failIf(os.path.lexists('a'))
607
self.failUnlessExists('b')
608
self.failIf(os.path.lexists('c'))
610
self.assertRaises(NoSuchFile,
611
t.delete_multi, ['a', 'b', 'c'])
613
self.assertRaises(NoSuchFile,
614
t.delete_multi, iter(['a', 'b', 'c']))
616
open('a', 'wb').write('another a text\n')
617
open('c', 'wb').write('another c text\n')
618
t.delete_multi(iter(['a', 'b', 'c']))
620
# We should have deleted everything
621
# SftpServer creates control files in the
622
# working directory, so we can just do a
624
# self.assertEqual([], os.listdir('.'))
463
626
def test_move(self):
464
# TODO: Test Transport.move
627
t = self.get_transport()
632
# TODO: I would like to use os.listdir() to
633
# make sure there are no extra files, but SftpServer
634
# creates control files in the working directory
635
# perhaps all of this could be done in a subdirectory
637
open('a', 'wb').write('a first file\n')
638
self.assertEquals([True, False], list(t.has_multi(['a', 'b'])))
641
self.failUnlessExists('b')
642
self.failIf(os.path.lexists('a'))
644
self.check_file_contents('b', 'a first file\n')
645
self.assertEquals([False, True], list(t.has_multi(['a', 'b'])))
648
open('c', 'wb').write('c this file\n')
650
self.failIf(os.path.lexists('c'))
651
self.check_file_contents('b', 'c this file\n')
653
# TODO: Try to write a test for atomicity
654
# TODO: Test moving into a non-existant subdirectory
655
# TODO: Test Transport.move_multi
658
t = self.get_transport()
663
open('a', 'wb').write('a file\n')
665
self.check_file_contents('b', 'a file\n')
667
self.assertRaises(NoSuchFile, t.copy, 'c', 'd')
669
# What should the assert be if you try to copy a
670
# file over a directory?
671
#self.assertRaises(Something, t.copy, 'a', 'c')
672
open('d', 'wb').write('text in d\n')
674
self.check_file_contents('b', 'text in d\n')
676
# TODO: test copy_multi
467
678
def test_connection_error(self):
468
679
"""ConnectionError is raised when connection is impossible"""
469
680
if not hasattr(self, "get_bogus_transport"):
471
682
t = self.get_bogus_transport()
472
self.assertRaises(ConnectionError, t.get, '.bzr/branch')
685
except (ConnectionError, NoSuchFile), e:
687
except (Exception), e:
688
self.failIf(True, 'Wrong exception thrown: %s' % e)
690
self.failIf(True, 'Did not get the expected exception.')
693
# TODO: Test stat, just try once, and if it throws, stop testing
694
from stat import S_ISDIR, S_ISREG
696
t = self.get_transport()
700
except TransportNotPossible, e:
701
# This transport cannot stat
704
paths = ['a', 'b/', 'b/c', 'b/d/', 'b/d/e']
705
self.build_tree(paths)
711
local_st = os.stat(p)
713
self.failUnless(S_ISDIR(st.st_mode))
715
self.failUnless(S_ISREG(st.st_mode))
716
self.assertEqual(local_st.st_size, st.st_size)
717
self.assertEqual(local_st.st_mode, st.st_mode)
718
local_stats.append(local_st)
720
remote_stats = list(t.stat_multi(paths))
721
remote_iter_stats = list(t.stat_multi(iter(paths)))
723
for local, remote, remote_iter in \
724
zip(local_stats, remote_stats, remote_iter_stats):
725
self.assertEqual(local.st_mode, remote.st_mode)
726
self.assertEqual(local.st_mode, remote_iter.st_mode)
728
self.assertEqual(local.st_size, remote.st_size)
729
self.assertEqual(local.st_size, remote_iter.st_size)
730
# Should we test UID/GID?
732
self.assertRaises(NoSuchFile, t.stat, 'q')
733
self.assertRaises(NoSuchFile, t.stat, 'b/a')
735
self.assertListRaises(NoSuchFile, t.stat_multi, ['a', 'c', 'd'])
736
self.assertListRaises(NoSuchFile, t.stat_multi, iter(['a', 'c', 'd']))
738
def test_list_dir(self):
739
# TODO: Test list_dir, just try once, and if it throws, stop testing
740
t = self.get_transport()
743
self.assertRaises(TransportNotPossible, t.list_dir, '.')
747
l = list(t.list_dir(d))
751
# SftpServer creates control files in the working directory
752
# so lets move down a directory to be safe
757
self.assertEqual([], sorted_list(u'.'))
758
self.build_tree(['a', 'b', 'c/', 'c/d', 'c/e'])
760
self.assertEqual([u'a', u'b', u'c'], sorted_list(u'.'))
761
self.assertEqual([u'd', u'e'], sorted_list(u'c'))
765
self.assertEqual([u'a', u'c'], sorted_list('.'))
766
self.assertEqual([u'e'], sorted_list(u'c'))
768
self.assertListRaises(NoSuchFile, t.list_dir, 'q')
769
self.assertListRaises(NoSuchFile, t.list_dir, 'c/f')
770
self.assertListRaises(NoSuchFile, t.list_dir, 'a')
772
def test_clone(self):
773
# TODO: Test that clone moves up and down the filesystem
774
t1 = self.get_transport()
776
self.build_tree(['a', 'b/', 'b/c'])
778
self.failUnless(t1.has('a'))
779
self.failUnless(t1.has('b/c'))
780
self.failIf(t1.has('c'))
783
self.failUnless(t2.has('c'))
784
self.failIf(t2.has('a'))
787
self.failUnless(t3.has('a'))
788
self.failIf(t3.has('c'))
790
self.failIf(t1.has('b/d'))
791
self.failIf(t2.has('d'))
792
self.failIf(t3.has('b/d'))
795
open('b/d', 'wb').write('newfile\n')
797
t2.put('d', 'newfile\n')
799
self.failUnless(t1.has('b/d'))
800
self.failUnless(t2.has('d'))
801
self.failUnless(t3.has('b/d'))
475
804
class LocalTransportTest(TestCaseInTempDir, TestTransportMixIn):
476
805
def get_transport(self):
477
806
from bzrlib.transport.local import LocalTransport
478
return LocalTransport('.')
807
return LocalTransport(u'.')
481
810
class HttpTransportTest(TestCaseWithWebserver, TestTransportMixIn):