← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_store.py

  • Committer: John Arbash Meinel
  • Date: 2013-05-19 14:29:37 UTC
  • mfrom: (6437.63.9 2.5)
  • mto: (6437.63.10 2.5)
  • mto: This revision was merged to the branch mainline in revision 6575.
  • Revision ID: john@arbash-meinel.com-20130519142937-21ykz2n2y2f22za9
Merge in the actual 2.5 branch. It seems I failed before

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_store.py
1
 
# Copyright (C) 2005 Canonical Ltd
 
1
# Copyright (C) 2005-2009, 2011 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 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, UnlistableStore, NoSuchFile
25
 
from bzrlib.transport.local import LocalTransport
 
24
from bzrlib.errors import BzrError
 
25
from bzrlib.store import TransportStore
26
26
from bzrlib.store.text import TextStore
 
27
from bzrlib.store.versioned import VersionedFileStore
27
28
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
32
33
 
33
34
 
34
35
class TestStores(object):
38
39
        f = store.get(fileid)
39
40
        self.assertEqual(f.read(), value)
40
41
 
41
 
    def test_add_str_deprecated(self):
42
 
        os.mkdir('a')
43
 
        store = self.get_store('a')
44
 
        self.callDeprecated(['Passing a string to Store.add'
45
 
                             ' was deprecated in version 0.11.'],
46
 
                            store.add, 'foo', '1')
47
 
        self.assertEqual('foo', store.get('1').read())
48
 
 
49
42
    def fill_store(self, store):
50
43
        store.add(StringIO('hello'), 'a')
51
44
        store.add(StringIO('other'), 'b')
71
64
    def test_get(self):
72
65
        store = self.get_store()
73
66
        self.fill_store(store)
74
 
    
 
67
 
75
68
        self.check_content(store, 'a', 'hello')
76
69
        self.check_content(store, 'b', 'other')
77
70
        self.check_content(store, 'c', 'something')
78
 
    
 
71
 
79
72
        # Make sure that requesting a non-existing file fails
80
73
        self.assertRaises(KeyError, self.check_content, store, 'd', None)
81
74
 
89
82
class TestCompressedTextStore(TestCaseInTempDir, TestStores):
90
83
 
91
84
    def get_store(self, path=u'.'):
92
 
        t = transport.get_transport(path)
 
85
        t = transport.get_transport_from_path(path)
93
86
        return TextStore(t, compressed=True)
94
87
 
95
88
    def test_total_size(self):
99
92
        store.add(StringIO('goodbye2'), '123123', 'dsc')
100
93
        # these get gzipped - content should be stable
101
94
        self.assertEqual(store.total_size(), (2, 55))
102
 
        
 
95
 
103
96
    def test__relpath_suffixed(self):
104
97
        my_store = TextStore(MockTransport(),
105
98
                             prefixed=True, compressed=True)
108
101
 
109
102
 
110
103
class TestMemoryStore(TestCase):
111
 
    
 
104
 
112
105
    def get_store(self):
113
106
        return TextStore(MemoryTransport())
114
 
    
 
107
 
115
108
    def test_add_and_retrieve(self):
116
109
        store = self.get_store()
117
110
        store.add(StringIO('hello'), 'aa')
123
116
 
124
117
    def test_missing_is_absent(self):
125
118
        store = self.get_store()
126
 
        self.failIf('aa' in store)
 
119
        self.assertFalse('aa' in store)
127
120
 
128
121
    def test_adding_fails_when_present(self):
129
122
        my_store = self.get_store()
146
139
class TestTextStore(TestCaseInTempDir, TestStores):
147
140
 
148
141
    def get_store(self, path=u'.'):
149
 
        t = transport.get_transport(path)
 
142
        t = transport.get_transport_from_path(path)
150
143
        return TextStore(t, compressed=False)
151
144
 
152
145
    def test_total_size(self):
164
157
class TestMixedTextStore(TestCaseInTempDir, TestStores):
165
158
 
166
159
    def get_store(self, path=u'.', compressed=True):
167
 
        t = transport.get_transport(path)
 
160
        t = transport.get_transport_from_path(path)
168
161
        return TextStore(t, compressed=compressed)
169
162
 
170
163
    def test_get_mixed(self):
172
165
        s = self.get_store(u'.', compressed=False)
173
166
        cs.add(StringIO('hello there'), 'a')
174
167
 
175
 
        self.failUnlessExists('a.gz')
176
 
        self.failIf(os.path.lexists('a'))
 
168
        self.assertPathExists('a.gz')
 
169
        self.assertFalse(os.path.lexists('a'))
177
170
 
178
171
        self.assertEquals(gzip.GzipFile('a.gz').read(), 'hello there')
179
172
 
181
174
        self.assertEquals(s.has_id('a'), True)
182
175
        self.assertEquals(cs.get('a').read(), 'hello there')
183
176
        self.assertEquals(s.get('a').read(), 'hello there')
184
 
        
 
177
 
185
178
        self.assertRaises(BzrError, s.add, StringIO('goodbye'), 'a')
186
179
 
187
180
        s.add(StringIO('goodbye'), 'b')
188
 
        self.failUnlessExists('b')
189
 
        self.failIf(os.path.lexists('b.gz'))
 
181
        self.assertPathExists('b')
 
182
        self.assertFalse(os.path.lexists('b.gz'))
190
183
        self.assertEquals(open('b').read(), 'goodbye')
191
184
 
192
185
        self.assertEquals(cs.has_id('b'), True)
193
186
        self.assertEquals(s.has_id('b'), True)
194
187
        self.assertEquals(cs.get('b').read(), 'goodbye')
195
188
        self.assertEquals(s.get('b').read(), 'goodbye')
196
 
        
 
189
 
197
190
        self.assertRaises(BzrError, cs.add, StringIO('again'), 'b')
198
191
 
199
192
class MockTransport(transport.Transport):
211
204
        return
212
205
 
213
206
 
214
 
class InstrumentedTransportStore(store.TransportStore):
 
207
class InstrumentedTransportStore(TransportStore):
215
208
    """An instrumented TransportStore.
216
209
 
217
210
    Here we replace template method worker methods with calls that record the
237
230
class TestMockTransport(TestCase):
238
231
 
239
232
    def test_isinstance(self):
240
 
        self.failUnless(isinstance(MockTransport(), transport.Transport))
 
233
        self.assertIsInstance(MockTransport(), transport.Transport)
241
234
 
242
235
    def test_has(self):
243
236
        self.assertEqual(False, MockTransport().has('foo'))
247
240
 
248
241
 
249
242
class TestTransportStore(TestCase):
250
 
    
 
243
 
251
244
    def test__relpath_invalid(self):
252
 
        my_store = store.TransportStore(MockTransport())
 
245
        my_store = TransportStore(MockTransport())
253
246
        self.assertRaises(ValueError, my_store._relpath, '/foo')
254
247
        self.assertRaises(ValueError, my_store._relpath, 'foo/')
255
248
 
256
249
    def test_register_invalid_suffixes(self):
257
 
        my_store = store.TransportStore(MockTransport())
 
250
        my_store = TransportStore(MockTransport())
258
251
        self.assertRaises(ValueError, my_store.register_suffix, '/')
259
252
        self.assertRaises(ValueError, my_store.register_suffix, '.gz/bar')
260
253
 
261
254
    def test__relpath_unregister_suffixes(self):
262
 
        my_store = store.TransportStore(MockTransport())
 
255
        my_store = TransportStore(MockTransport())
263
256
        self.assertRaises(ValueError, my_store._relpath, 'foo', ['gz'])
264
257
        self.assertRaises(ValueError, my_store._relpath, 'foo', ['dsc', 'gz'])
265
258
 
266
259
    def test__relpath_simple(self):
267
 
        my_store = store.TransportStore(MockTransport())
 
260
        my_store = TransportStore(MockTransport())
268
261
        self.assertEqual("foo", my_store._relpath('foo'))
269
262
 
270
263
    def test__relpath_prefixed(self):
271
 
        my_store = store.TransportStore(MockTransport(), True)
 
264
        my_store = TransportStore(MockTransport(), True)
272
265
        self.assertEqual('45/foo', my_store._relpath('foo'))
273
266
 
274
267
    def test__relpath_simple_suffixed(self):
275
 
        my_store = store.TransportStore(MockTransport())
 
268
        my_store = TransportStore(MockTransport())
276
269
        my_store.register_suffix('bar')
277
270
        my_store.register_suffix('baz')
278
271
        self.assertEqual('foo.baz', my_store._relpath('foo', ['baz']))
279
272
        self.assertEqual('foo.bar.baz', my_store._relpath('foo', ['bar', 'baz']))
280
273
 
281
274
    def test__relpath_prefixed_suffixed(self):
282
 
        my_store = store.TransportStore(MockTransport(), True)
 
275
        my_store = TransportStore(MockTransport(), True)
283
276
        my_store.register_suffix('bar')
284
277
        my_store.register_suffix('baz')
285
278
        self.assertEqual('45/foo.baz', my_store._relpath('foo', ['baz']))
304
297
        my_store.register_suffix('dsc')
305
298
        my_store.add(stream, "foo", 'dsc')
306
299
        self.assertEqual([("_add", "foo.dsc", stream)], my_store._calls)
307
 
        
 
300
 
308
301
    def test_add_simple_suffixed(self):
309
302
        stream = StringIO("content")
310
303
        my_store = InstrumentedTransportStore(MockTransport(), True)
324
317
        stream = StringIO("signature for missing base")
325
318
        my_store.add(stream, "missing", 'sig')
326
319
        return my_store
327
 
        
 
320
 
328
321
    def test_has_simple(self):
329
322
        my_store = self.get_populated_store()
330
323
        self.assertEqual(True, my_store.has_id('foo'))
401
394
        self.assertRaises(KeyError, to_store.get, 'missing', 'sig')
402
395
 
403
396
    def test_relpath_escaped(self):
404
 
        my_store = store.TransportStore(MemoryTransport())
 
397
        my_store = TransportStore(MemoryTransport())
405
398
        self.assertEqual('%25', my_store._relpath('%'))
406
399
 
407
400
    def test_escaped_uppercase(self):
408
401
        """Uppercase letters are escaped for safety on Windows"""
409
 
        my_store = store.TransportStore(MemoryTransport(), escaped=True)
 
402
        my_store = TransportStore(MemoryTransport(), prefixed=True,
 
403
            escaped=True)
410
404
        # a particularly perverse file-id! :-)
411
 
        self.assertEquals(my_store._escape_file_id('C:<>'), '%43%3a%3c%3e')
 
405
        self.assertEquals(my_store._relpath('C:<>'), 'be/%2543%253a%253c%253e')
412
406
 
413
407
 
414
408
class TestVersionFileStore(TestCaseWithTransport):
415
409
 
 
410
    def get_scope(self):
 
411
        return self._transaction
 
412
 
416
413
    def setUp(self):
417
414
        super(TestVersionFileStore, self).setUp()
418
 
        self.vfstore = store.versioned.VersionedFileStore(MemoryTransport())
 
415
        self.vfstore = VersionedFileStore(MemoryTransport(),
 
416
            versionedfile_class=WeaveFile)
 
417
        self.vfstore.get_scope = self.get_scope
 
418
        self._transaction = None
419
419
 
420
420
    def test_get_weave_registers_dirty_in_write(self):
421
 
        transaction = transactions.WriteTransaction()
422
 
        vf = self.vfstore.get_weave_or_empty('id', transaction)
423
 
        transaction.finish()
424
 
        self.assertRaises(errors.OutSideTransaction, vf.add_lines, 'b', [], [])
425
 
        transaction = transactions.WriteTransaction()
426
 
        vf = self.vfstore.get_weave('id', transaction)
427
 
        transaction.finish()
428
 
        self.assertRaises(errors.OutSideTransaction, vf.add_lines, 'b', [], [])
429
 
 
430
 
    def test_get_weave_or_empty_readonly_fails(self):
431
 
        transaction = transactions.ReadOnlyTransaction()
432
 
        vf = self.assertRaises(errors.ReadOnlyError,
433
 
                               self.vfstore.get_weave_or_empty,
434
 
                               'id',
435
 
                               transaction)
 
421
        self._transaction = transactions.WriteTransaction()
 
422
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
 
423
        self._transaction.finish()
 
424
        self._transaction = None
 
425
        self.assertRaises(errors.OutSideTransaction, vf.add_lines, 'b', [], [])
 
426
        self._transaction = transactions.WriteTransaction()
 
427
        vf = self.vfstore.get_weave('id', self._transaction)
 
428
        self._transaction.finish()
 
429
        self._transaction = None
 
430
        self.assertRaises(errors.OutSideTransaction, vf.add_lines, 'b', [], [])
436
431
 
437
432
    def test_get_weave_readonly_cant_write(self):
438
 
        transaction = transactions.WriteTransaction()
439
 
        vf = self.vfstore.get_weave_or_empty('id', transaction)
440
 
        transaction.finish()
441
 
        transaction = transactions.ReadOnlyTransaction()
442
 
        vf = self.vfstore.get_weave_or_empty('id', transaction)
 
433
        self._transaction = transactions.WriteTransaction()
 
434
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
 
435
        self._transaction.finish()
 
436
        self._transaction = transactions.ReadOnlyTransaction()
 
437
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
443
438
        self.assertRaises(errors.ReadOnlyError, vf.add_lines, 'b', [], [])
444
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))