← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_store.py

Implement version 3 of the network protocol. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_store.py
1
 
# Copyright (C) 2005-2011, 2016 Canonical Ltd
 
1
# Copyright (C) 2005, 2007 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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Test Store implementations."""
18
18
 
21
21
import gzip
22
22
 
23
23
import bzrlib.errors as errors
24
 
from bzrlib.errors import BzrError
25
 
from bzrlib.store import TransportStore
 
24
from bzrlib.errors import BzrError, UnlistableStore, NoSuchFile
 
25
from bzrlib.transport.local import LocalTransport
26
26
from bzrlib.store.text import TextStore
27
 
from bzrlib.store.versioned import VersionedFileStore
28
27
from bzrlib.tests import TestCase, TestCaseInTempDir, TestCaseWithTransport
 
28
import bzrlib.store as store
29
29
import bzrlib.transactions as transactions
30
30
import bzrlib.transport as transport
31
31
from bzrlib.transport.memory import MemoryTransport
32
 
from bzrlib.weave import WeaveFile
33
32
 
34
33
 
35
34
class TestStores(object):
64
63
    def test_get(self):
65
64
        store = self.get_store()
66
65
        self.fill_store(store)
67
 
 
 
66
    
68
67
        self.check_content(store, 'a', 'hello')
69
68
        self.check_content(store, 'b', 'other')
70
69
        self.check_content(store, 'c', 'something')
71
 
 
 
70
    
72
71
        # Make sure that requesting a non-existing file fails
73
72
        self.assertRaises(KeyError, self.check_content, store, 'd', None)
74
73
 
82
81
class TestCompressedTextStore(TestCaseInTempDir, TestStores):
83
82
 
84
83
    def get_store(self, path=u'.'):
85
 
        t = transport.get_transport_from_path(path)
 
84
        t = transport.get_transport(path)
86
85
        return TextStore(t, compressed=True)
87
86
 
88
87
    def test_total_size(self):
92
91
        store.add(StringIO('goodbye2'), '123123', 'dsc')
93
92
        # these get gzipped - content should be stable
94
93
        self.assertEqual(store.total_size(), (2, 55))
95
 
 
 
94
        
96
95
    def test__relpath_suffixed(self):
97
96
        my_store = TextStore(MockTransport(),
98
97
                             prefixed=True, compressed=True)
101
100
 
102
101
 
103
102
class TestMemoryStore(TestCase):
104
 
 
 
103
    
105
104
    def get_store(self):
106
105
        return TextStore(MemoryTransport())
107
 
 
 
106
    
108
107
    def test_add_and_retrieve(self):
109
108
        store = self.get_store()
110
109
        store.add(StringIO('hello'), 'aa')
116
115
 
117
116
    def test_missing_is_absent(self):
118
117
        store = self.get_store()
119
 
        self.assertFalse('aa' in store)
 
118
        self.failIf('aa' in store)
120
119
 
121
120
    def test_adding_fails_when_present(self):
122
121
        my_store = self.get_store()
139
138
class TestTextStore(TestCaseInTempDir, TestStores):
140
139
 
141
140
    def get_store(self, path=u'.'):
142
 
        t = transport.get_transport_from_path(path)
 
141
        t = transport.get_transport(path)
143
142
        return TextStore(t, compressed=False)
144
143
 
145
144
    def test_total_size(self):
157
156
class TestMixedTextStore(TestCaseInTempDir, TestStores):
158
157
 
159
158
    def get_store(self, path=u'.', compressed=True):
160
 
        t = transport.get_transport_from_path(path)
 
159
        t = transport.get_transport(path)
161
160
        return TextStore(t, compressed=compressed)
162
161
 
163
162
    def test_get_mixed(self):
165
164
        s = self.get_store(u'.', compressed=False)
166
165
        cs.add(StringIO('hello there'), 'a')
167
166
 
168
 
        self.assertPathExists('a.gz')
169
 
        self.assertFalse(os.path.lexists('a'))
170
 
 
171
 
        self.assertEqual(gzip.GzipFile('a.gz').read(), 'hello there')
172
 
 
173
 
        self.assertEqual(cs.has_id('a'), True)
174
 
        self.assertEqual(s.has_id('a'), True)
175
 
        self.assertEqual(cs.get('a').read(), 'hello there')
176
 
        self.assertEqual(s.get('a').read(), 'hello there')
177
 
 
 
167
        self.failUnlessExists('a.gz')
 
168
        self.failIf(os.path.lexists('a'))
 
169
 
 
170
        self.assertEquals(gzip.GzipFile('a.gz').read(), 'hello there')
 
171
 
 
172
        self.assertEquals(cs.has_id('a'), True)
 
173
        self.assertEquals(s.has_id('a'), True)
 
174
        self.assertEquals(cs.get('a').read(), 'hello there')
 
175
        self.assertEquals(s.get('a').read(), 'hello there')
 
176
        
178
177
        self.assertRaises(BzrError, s.add, StringIO('goodbye'), 'a')
179
178
 
180
179
        s.add(StringIO('goodbye'), 'b')
181
 
        self.assertPathExists('b')
182
 
        self.assertFalse(os.path.lexists('b.gz'))
183
 
        self.assertEqual(open('b').read(), 'goodbye')
184
 
 
185
 
        self.assertEqual(cs.has_id('b'), True)
186
 
        self.assertEqual(s.has_id('b'), True)
187
 
        self.assertEqual(cs.get('b').read(), 'goodbye')
188
 
        self.assertEqual(s.get('b').read(), 'goodbye')
189
 
 
 
180
        self.failUnlessExists('b')
 
181
        self.failIf(os.path.lexists('b.gz'))
 
182
        self.assertEquals(open('b').read(), 'goodbye')
 
183
 
 
184
        self.assertEquals(cs.has_id('b'), True)
 
185
        self.assertEquals(s.has_id('b'), True)
 
186
        self.assertEquals(cs.get('b').read(), 'goodbye')
 
187
        self.assertEquals(s.get('b').read(), 'goodbye')
 
188
        
190
189
        self.assertRaises(BzrError, cs.add, StringIO('again'), 'b')
191
190
 
192
191
class MockTransport(transport.Transport):
204
203
        return
205
204
 
206
205
 
207
 
class InstrumentedTransportStore(TransportStore):
 
206
class InstrumentedTransportStore(store.TransportStore):
208
207
    """An instrumented TransportStore.
209
208
 
210
209
    Here we replace template method worker methods with calls that record the
230
229
class TestMockTransport(TestCase):
231
230
 
232
231
    def test_isinstance(self):
233
 
        self.assertIsInstance(MockTransport(), transport.Transport)
 
232
        self.failUnless(isinstance(MockTransport(), transport.Transport))
234
233
 
235
234
    def test_has(self):
236
235
        self.assertEqual(False, MockTransport().has('foo'))
240
239
 
241
240
 
242
241
class TestTransportStore(TestCase):
243
 
 
 
242
    
244
243
    def test__relpath_invalid(self):
245
 
        my_store = TransportStore(MockTransport())
 
244
        my_store = store.TransportStore(MockTransport())
246
245
        self.assertRaises(ValueError, my_store._relpath, '/foo')
247
246
        self.assertRaises(ValueError, my_store._relpath, 'foo/')
248
247
 
249
248
    def test_register_invalid_suffixes(self):
250
 
        my_store = TransportStore(MockTransport())
 
249
        my_store = store.TransportStore(MockTransport())
251
250
        self.assertRaises(ValueError, my_store.register_suffix, '/')
252
251
        self.assertRaises(ValueError, my_store.register_suffix, '.gz/bar')
253
252
 
254
253
    def test__relpath_unregister_suffixes(self):
255
 
        my_store = TransportStore(MockTransport())
 
254
        my_store = store.TransportStore(MockTransport())
256
255
        self.assertRaises(ValueError, my_store._relpath, 'foo', ['gz'])
257
256
        self.assertRaises(ValueError, my_store._relpath, 'foo', ['dsc', 'gz'])
258
257
 
259
258
    def test__relpath_simple(self):
260
 
        my_store = TransportStore(MockTransport())
 
259
        my_store = store.TransportStore(MockTransport())
261
260
        self.assertEqual("foo", my_store._relpath('foo'))
262
261
 
263
262
    def test__relpath_prefixed(self):
264
 
        my_store = TransportStore(MockTransport(), True)
 
263
        my_store = store.TransportStore(MockTransport(), True)
265
264
        self.assertEqual('45/foo', my_store._relpath('foo'))
266
265
 
267
266
    def test__relpath_simple_suffixed(self):
268
 
        my_store = TransportStore(MockTransport())
 
267
        my_store = store.TransportStore(MockTransport())
269
268
        my_store.register_suffix('bar')
270
269
        my_store.register_suffix('baz')
271
270
        self.assertEqual('foo.baz', my_store._relpath('foo', ['baz']))
272
271
        self.assertEqual('foo.bar.baz', my_store._relpath('foo', ['bar', 'baz']))
273
272
 
274
273
    def test__relpath_prefixed_suffixed(self):
275
 
        my_store = TransportStore(MockTransport(), True)
 
274
        my_store = store.TransportStore(MockTransport(), True)
276
275
        my_store.register_suffix('bar')
277
276
        my_store.register_suffix('baz')
278
277
        self.assertEqual('45/foo.baz', my_store._relpath('foo', ['baz']))
297
296
        my_store.register_suffix('dsc')
298
297
        my_store.add(stream, "foo", 'dsc')
299
298
        self.assertEqual([("_add", "foo.dsc", stream)], my_store._calls)
300
 
 
 
299
        
301
300
    def test_add_simple_suffixed(self):
302
301
        stream = StringIO("content")
303
302
        my_store = InstrumentedTransportStore(MockTransport(), True)
317
316
        stream = StringIO("signature for missing base")
318
317
        my_store.add(stream, "missing", 'sig')
319
318
        return my_store
320
 
 
 
319
        
321
320
    def test_has_simple(self):
322
321
        my_store = self.get_populated_store()
323
322
        self.assertEqual(True, my_store.has_id('foo'))
394
393
        self.assertRaises(KeyError, to_store.get, 'missing', 'sig')
395
394
 
396
395
    def test_relpath_escaped(self):
397
 
        my_store = TransportStore(MemoryTransport())
 
396
        my_store = store.TransportStore(MemoryTransport())
398
397
        self.assertEqual('%25', my_store._relpath('%'))
399
398
 
400
399
    def test_escaped_uppercase(self):
401
400
        """Uppercase letters are escaped for safety on Windows"""
402
 
        my_store = TransportStore(MemoryTransport(), prefixed=True,
403
 
            escaped=True)
 
401
        my_store = store.TransportStore(MemoryTransport(), escaped=True)
404
402
        # a particularly perverse file-id! :-)
405
 
        self.assertEqual(my_store._relpath('C:<>'), 'be/%2543%253a%253c%253e')
 
403
        self.assertEquals(my_store._escape_file_id('C:<>'), '%43%3a%3c%3e')
406
404
 
407
405
 
408
406
class TestVersionFileStore(TestCaseWithTransport):
412
410
 
413
411
    def setUp(self):
414
412
        super(TestVersionFileStore, self).setUp()
415
 
        self.vfstore = VersionedFileStore(MemoryTransport(),
416
 
            versionedfile_class=WeaveFile)
 
413
        self.vfstore = store.versioned.VersionedFileStore(MemoryTransport())
417
414
        self.vfstore.get_scope = self.get_scope
418
415
        self._transaction = None
419
416
 
436
433
        self._transaction = transactions.ReadOnlyTransaction()
437
434
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
438
435
        self.assertRaises(errors.ReadOnlyError, vf.add_lines, 'b', [], [])
439
 
 
440
 
    def test___iter__escaped(self):
441
 
        self.vfstore = VersionedFileStore(MemoryTransport(),
442
 
            prefixed=True, escaped=True, versionedfile_class=WeaveFile)
443
 
        self.vfstore.get_scope = self.get_scope
444
 
        self._transaction = transactions.WriteTransaction()
445
 
        vf = self.vfstore.get_weave_or_empty(' ', self._transaction)
446
 
        vf.add_lines('a', [], [])
447
 
        del vf
448
 
        self._transaction.finish()
449
 
        self.assertEqual([' '], list(self.vfstore))