← 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: Martin Pool
  • Date: 2006-03-03 08:55:34 UTC
  • mto: This revision was merged to the branch mainline in revision 1593.
  • Revision ID: mbp@sourcefrog.net-20060303085534-d24a8118f4ce571a
Add some tests that format7 repo creates the right lock type

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_store.py
1
 
# Copyright (C) 2005, 2007 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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20
20
import os
21
21
import gzip
22
22
 
23
 
import bzrlib.errors as errors
24
23
from bzrlib.errors import BzrError, UnlistableStore, NoSuchFile
 
24
from bzrlib.store import copy_all
25
25
from bzrlib.transport.local import LocalTransport
26
26
from bzrlib.store.text import TextStore
27
 
from bzrlib.tests import TestCase, TestCaseInTempDir, TestCaseWithTransport
 
27
from bzrlib.tests import TestCase, TestCaseInTempDir
28
28
import bzrlib.store as store
29
 
import bzrlib.transactions as transactions
30
29
import bzrlib.transport as transport
31
30
from bzrlib.transport.memory import MemoryTransport
32
31
 
48
47
        """Test copying"""
49
48
        os.mkdir('a')
50
49
        store_a = self.get_store('a')
51
 
        store_a.add(StringIO('foo'), '1')
 
50
        store_a.add('foo', '1')
52
51
        os.mkdir('b')
53
52
        store_b = self.get_store('b')
54
 
        store_b.copy_all_ids(store_a)
 
53
        copy_all(store_a, store_b)
55
54
        self.assertEqual(store_a.get('1').read(), 'foo')
56
55
        self.assertEqual(store_b.get('1').read(), 'foo')
57
56
        # TODO: Switch the exception form UnlistableStore to
81
80
class TestCompressedTextStore(TestCaseInTempDir, TestStores):
82
81
 
83
82
    def get_store(self, path=u'.'):
84
 
        t = transport.get_transport(path)
 
83
        t = LocalTransport(path)
85
84
        return TextStore(t, compressed=True)
86
85
 
87
86
    def test_total_size(self):
102
101
class TestMemoryStore(TestCase):
103
102
    
104
103
    def get_store(self):
105
 
        return TextStore(MemoryTransport())
 
104
        return store.ImmutableMemoryStore()
106
105
    
 
106
    def test_imports(self):
 
107
        from bzrlib.store import ImmutableMemoryStore
 
108
 
107
109
    def test_add_and_retrieve(self):
108
110
        store = self.get_store()
109
111
        store.add(StringIO('hello'), 'aa')
138
140
class TestTextStore(TestCaseInTempDir, TestStores):
139
141
 
140
142
    def get_store(self, path=u'.'):
141
 
        t = transport.get_transport(path)
 
143
        t = LocalTransport(path)
142
144
        return TextStore(t, compressed=False)
143
145
 
144
146
    def test_total_size(self):
156
158
class TestMixedTextStore(TestCaseInTempDir, TestStores):
157
159
 
158
160
    def get_store(self, path=u'.', compressed=True):
159
 
        t = transport.get_transport(path)
 
161
        t = LocalTransport(path)
160
162
        return TextStore(t, compressed=compressed)
161
163
 
162
164
    def test_get_mixed(self):
385
387
        to_store = TextStore(MemoryTransport(),
386
388
                             prefixed=True, compressed=True)
387
389
        to_store.register_suffix('sig')
388
 
        to_store.copy_all_ids(from_store)
 
390
        copy_all(from_store, to_store)
389
391
        self.assertEqual(1, len(to_store))
390
392
        self.assertEqual(set(['foo']), set(to_store.__iter__()))
391
393
        self.assertEqual('content', to_store.get('foo').read())
395
397
    def test_relpath_escaped(self):
396
398
        my_store = store.TransportStore(MemoryTransport())
397
399
        self.assertEqual('%25', my_store._relpath('%'))
398
 
 
399
 
    def test_escaped_uppercase(self):
400
 
        """Uppercase letters are escaped for safety on Windows"""
401
 
        my_store = store.TransportStore(MemoryTransport(), prefixed=True,
402
 
            escaped=True)
403
 
        # a particularly perverse file-id! :-)
404
 
        self.assertEquals(my_store._relpath('C:<>'), 'be/%2543%253a%253c%253e')
405
 
 
406
 
 
407
 
class TestVersionFileStore(TestCaseWithTransport):
408
 
 
409
 
    def get_scope(self):
410
 
        return self._transaction
411
 
 
412
 
    def setUp(self):
413
 
        super(TestVersionFileStore, self).setUp()
414
 
        self.vfstore = store.versioned.VersionedFileStore(MemoryTransport())
415
 
        self.vfstore.get_scope = self.get_scope
416
 
        self._transaction = None
417
 
 
418
 
    def test_get_weave_registers_dirty_in_write(self):
419
 
        self._transaction = transactions.WriteTransaction()
420
 
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
421
 
        self._transaction.finish()
422
 
        self._transaction = None
423
 
        self.assertRaises(errors.OutSideTransaction, vf.add_lines, 'b', [], [])
424
 
        self._transaction = transactions.WriteTransaction()
425
 
        vf = self.vfstore.get_weave('id', self._transaction)
426
 
        self._transaction.finish()
427
 
        self._transaction = None
428
 
        self.assertRaises(errors.OutSideTransaction, vf.add_lines, 'b', [], [])
429
 
 
430
 
    def test_get_weave_readonly_cant_write(self):
431
 
        self._transaction = transactions.WriteTransaction()
432
 
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
433
 
        self._transaction.finish()
434
 
        self._transaction = transactions.ReadOnlyTransaction()
435
 
        vf = self.vfstore.get_weave_or_empty('id', self._transaction)
436
 
        self.assertRaises(errors.ReadOnlyError, vf.add_lines, 'b', [], [])
437
 
 
438
 
    def test___iter__escaped(self):
439
 
        self.vfstore = store.versioned.VersionedFileStore(MemoryTransport(),
440
 
            prefixed=True, escaped=True)
441
 
        self.vfstore.get_scope = self.get_scope
442
 
        self._transaction = transactions.WriteTransaction()
443
 
        vf = self.vfstore.get_weave_or_empty(' ', self._transaction)
444
 
        vf.add_lines('a', [], [])
445
 
        del vf
446
 
        self._transaction.finish()
447
 
        self.assertEqual([' '], list(self.vfstore))