← 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: Robert Collins
  • Date: 2006-03-04 02:30:17 UTC
  • mto: (1594.2.4 integration)
  • mto: This revision was merged to the branch mainline in revision 1596.
  • Revision ID: robertc@robertcollins.net-20060304023017-7f5f460efff7d2e8
Make join cheaper for compatibly inconsistent parents.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_store.py
1
 
# Copyright (C) 2005-2009, 2011 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 by Canonical Development Ltd
 
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
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
 
20
20
import os
21
21
import gzip
22
22
 
23
 
import bzrlib.errors as errors
24
 
from bzrlib.errors import BzrError
25
 
from bzrlib.store import TransportStore
 
23
from bzrlib.errors import BzrError, UnlistableStore, NoSuchFile
 
24
from bzrlib.store import copy_all
 
25
from bzrlib.transport.local import LocalTransport
26
26
from bzrlib.store.text import TextStore
27
 
from bzrlib.store.versioned import VersionedFileStore
28
 
from bzrlib.tests import TestCase, TestCaseInTempDir, TestCaseWithTransport
29
 
import bzrlib.transactions as transactions
 
27
from bzrlib.tests import TestCase, TestCaseInTempDir
 
28
import bzrlib.store as store
30
29
import bzrlib.transport as transport
31
30
from bzrlib.transport.memory import MemoryTransport
32
 
from bzrlib.weave import WeaveFile
33
31
 
34
32
 
35
33
class TestStores(object):
49
47
        """Test copying"""
50
48
        os.mkdir('a')
51
49
        store_a = self.get_store('a')
52
 
        store_a.add(StringIO('foo'), '1')
 
50
        store_a.add('foo', '1')
53
51
        os.mkdir('b')
54
52
        store_b = self.get_store('b')
55
 
        store_b.copy_all_ids(store_a)
 
53
        copy_all(store_a, store_b)
56
54
        self.assertEqual(store_a.get('1').read(), 'foo')
57
55
        self.assertEqual(store_b.get('1').read(), 'foo')
58
56
        # TODO: Switch the exception form UnlistableStore to
64
62
    def test_get(self):
65
63
        store = self.get_store()
66
64
        self.fill_store(store)
67
 
 
 
65
    
68
66
        self.check_content(store, 'a', 'hello')
69
67
        self.check_content(store, 'b', 'other')
70
68
        self.check_content(store, 'c', 'something')
71
 
 
 
69
    
72
70
        # Make sure that requesting a non-existing file fails
73
71
        self.assertRaises(KeyError, self.check_content, store, 'd', None)
74
72
 
82
80
class TestCompressedTextStore(TestCaseInTempDir, TestStores):
83
81
 
84
82
    def get_store(self, path=u'.'):
85
 
        t = transport.get_transport_from_path(path)
 
83
        t = LocalTransport(path)
86
84
        return TextStore(t, compressed=True)
87
85
 
88
86
    def test_total_size(self):
92
90
        store.add(StringIO('goodbye2'), '123123', 'dsc')
93
91
        # these get gzipped - content should be stable
94
92
        self.assertEqual(store.total_size(), (2, 55))
95
 
 
 
93
        
96
94
    def test__relpath_suffixed(self):
97
95
        my_store = TextStore(MockTransport(),
98
96
                             prefixed=True, compressed=True)
101
99
 
102
100
 
103
101
class TestMemoryStore(TestCase):
104
 
 
 
102
    
105
103
    def get_store(self):
106
 
        return TextStore(MemoryTransport())
 
104
        return store.ImmutableMemoryStore()
 
105
    
 
106
    def test_imports(self):
 
107
        from bzrlib.store import ImmutableMemoryStore
107
108
 
108
109
    def test_add_and_retrieve(self):
109
110
        store = self.get_store()
116
117
 
117
118
    def test_missing_is_absent(self):
118
119
        store = self.get_store()
119
 
        self.assertFalse('aa' in store)
 
120
        self.failIf('aa' in store)
120
121
 
121
122
    def test_adding_fails_when_present(self):
122
123
        my_store = self.get_store()
139
140
class TestTextStore(TestCaseInTempDir, TestStores):
140
141
 
141
142
    def get_store(self, path=u'.'):
142
 
        t = transport.get_transport_from_path(path)
 
143
        t = LocalTransport(path)
143
144
        return TextStore(t, compressed=False)
144
145
 
145
146
    def test_total_size(self):
157
158
class TestMixedTextStore(TestCaseInTempDir, TestStores):
158
159
 
159
160
    def get_store(self, path=u'.', compressed=True):
160
 
        t = transport.get_transport_from_path(path)
 
161
        t = LocalTransport(path)
161
162
        return TextStore(t, compressed=compressed)
162
163
 
163
164
    def test_get_mixed(self):
165
166
        s = self.get_store(u'.', compressed=False)
166
167
        cs.add(StringIO('hello there'), 'a')
167
168
 
168
 
        self.assertPathExists('a.gz')
169
 
        self.assertFalse(os.path.lexists('a'))
 
169
        self.failUnlessExists('a.gz')
 
170
        self.failIf(os.path.lexists('a'))
170
171
 
171
172
        self.assertEquals(gzip.GzipFile('a.gz').read(), 'hello there')
172
173
 
174
175
        self.assertEquals(s.has_id('a'), True)
175
176
        self.assertEquals(cs.get('a').read(), 'hello there')
176
177
        self.assertEquals(s.get('a').read(), 'hello there')
177
 
 
 
178
        
178
179
        self.assertRaises(BzrError, s.add, StringIO('goodbye'), 'a')
179
180
 
180
181
        s.add(StringIO('goodbye'), 'b')
181
 
        self.assertPathExists('b')
182
 
        self.assertFalse(os.path.lexists('b.gz'))
 
182
        self.failUnlessExists('b')
 
183
        self.failIf(os.path.lexists('b.gz'))
183
184
        self.assertEquals(open('b').read(), 'goodbye')
184
185
 
185
186
        self.assertEquals(cs.has_id('b'), True)
186
187
        self.assertEquals(s.has_id('b'), True)
187
188
        self.assertEquals(cs.get('b').read(), 'goodbye')
188
189
        self.assertEquals(s.get('b').read(), 'goodbye')
189
 
 
 
190
        
190
191
        self.assertRaises(BzrError, cs.add, StringIO('again'), 'b')
191
192
 
192
193
class MockTransport(transport.Transport):
204
205
        return
205
206
 
206
207
 
207
 
class InstrumentedTransportStore(TransportStore):
 
208
class InstrumentedTransportStore(store.TransportStore):
208
209
    """An instrumented TransportStore.
209
210
 
210
211
    Here we replace template method worker methods with calls that record the
230
231
class TestMockTransport(TestCase):
231
232
 
232
233
    def test_isinstance(self):
233
 
        self.assertIsInstance(MockTransport(), transport.Transport)
 
234
        self.failUnless(isinstance(MockTransport(), transport.Transport))
234
235
 
235
236
    def test_has(self):
236
237
        self.assertEqual(False, MockTransport().has('foo'))
240
241
 
241
242
 
242
243
class TestTransportStore(TestCase):
243
 
 
 
244
    
244
245
    def test__relpath_invalid(self):
245
 
        my_store = TransportStore(MockTransport())
 
246
        my_store = store.TransportStore(MockTransport())
246
247
        self.assertRaises(ValueError, my_store._relpath, '/foo')
247
248
        self.assertRaises(ValueError, my_store._relpath, 'foo/')
248
249
 
249
250
    def test_register_invalid_suffixes(self):
250
 
        my_store = TransportStore(MockTransport())
 
251
        my_store = store.TransportStore(MockTransport())
251
252
        self.assertRaises(ValueError, my_store.register_suffix, '/')
252
253
        self.assertRaises(ValueError, my_store.register_suffix, '.gz/bar')
253
254
 
254
255
    def test__relpath_unregister_suffixes(self):
255
 
        my_store = TransportStore(MockTransport())
 
256
        my_store = store.TransportStore(MockTransport())
256
257
        self.assertRaises(ValueError, my_store._relpath, 'foo', ['gz'])
257
258
        self.assertRaises(ValueError, my_store._relpath, 'foo', ['dsc', 'gz'])
258
259
 
259
260
    def test__relpath_simple(self):
260
 
        my_store = TransportStore(MockTransport())
 
261
        my_store = store.TransportStore(MockTransport())
261
262
        self.assertEqual("foo", my_store._relpath('foo'))
262
263
 
263
264
    def test__relpath_prefixed(self):
264
 
        my_store = TransportStore(MockTransport(), True)
 
265
        my_store = store.TransportStore(MockTransport(), True)
265
266
        self.assertEqual('45/foo', my_store._relpath('foo'))
266
267
 
267
268
    def test__relpath_simple_suffixed(self):
268
 
        my_store = TransportStore(MockTransport())
 
269
        my_store = store.TransportStore(MockTransport())
269
270
        my_store.register_suffix('bar')
270
271
        my_store.register_suffix('baz')
271
272
        self.assertEqual('foo.baz', my_store._relpath('foo', ['baz']))
272
273
        self.assertEqual('foo.bar.baz', my_store._relpath('foo', ['bar', 'baz']))
273
274
 
274
275
    def test__relpath_prefixed_suffixed(self):
275
 
        my_store = TransportStore(MockTransport(), True)
 
276
        my_store = store.TransportStore(MockTransport(), True)
276
277
        my_store.register_suffix('bar')
277
278
        my_store.register_suffix('baz')
278
279
        self.assertEqual('45/foo.baz', my_store._relpath('foo', ['baz']))
297
298
        my_store.register_suffix('dsc')
298
299
        my_store.add(stream, "foo", 'dsc')
299
300
        self.assertEqual([("_add", "foo.dsc", stream)], my_store._calls)
300
 
 
 
301
        
301
302
    def test_add_simple_suffixed(self):
302
303
        stream = StringIO("content")
303
304
        my_store = InstrumentedTransportStore(MockTransport(), True)
317
318
        stream = StringIO("signature for missing base")
318
319
        my_store.add(stream, "missing", 'sig')
319
320
        return my_store
320
 
 
 
321
        
321
322
    def test_has_simple(self):
322
323
        my_store = self.get_populated_store()
323
324
        self.assertEqual(True, my_store.has_id('foo'))
386
387
        to_store = TextStore(MemoryTransport(),
387
388
                             prefixed=True, compressed=True)
388
389
        to_store.register_suffix('sig')
389
 
        to_store.copy_all_ids(from_store)
 
390
        copy_all(from_store, to_store)
390
391
        self.assertEqual(1, len(to_store))
391
392
        self.assertEqual(set(['foo']), set(to_store.__iter__()))
392
393
        self.assertEqual('content', to_store.get('foo').read())
394
395
        self.assertRaises(KeyError, to_store.get, 'missing', 'sig')
395
396
 
396
397
    def test_relpath_escaped(self):
397
 
        my_store = TransportStore(MemoryTransport())
 
398
        my_store = store.TransportStore(MemoryTransport())
398
399
        self.assertEqual('%25', my_store._relpath('%'))
399
 
 
400
 
    def test_escaped_uppercase(self):
401
 
        """Uppercase letters are escaped for safety on Windows"""
402
 
        my_store = TransportStore(MemoryTransport(), prefixed=True,
403
 
            escaped=True)
404
 
        # a particularly perverse file-id! :-)
405
 
        self.assertEquals(my_store._relpath('C:<>'), 'be/%2543%253a%253c%253e')
406
 
 
407
 
 
408
 
class TestVersionFileStore(TestCaseWithTransport):
409
 
 
410
 
    def get_scope(self):
411
 
        return self._transaction
412
 
 
413
 
    def setUp(self):
414
 
        super(TestVersionFileStore, self).setUp()
415
 
        self.vfstore = VersionedFileStore(MemoryTransport(),
416
 
            versionedfile_class=WeaveFile)
417
 
        self.vfstore.get_scope = self.get_scope
418
 
        self._transaction = None
419
 
 
420
 
    def test_get_weave_registers_dirty_in_write(self):
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', [], [])
431
 
 
432
 
    def test_get_weave_readonly_cant_write(self):
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)
438
 
        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))