← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_store.py

Merge from integration.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_store.py
20
20
import os
21
21
import gzip
22
22
 
23
 
from bzrlib.errors import BzrError, UnlistableStore
 
23
from bzrlib.errors import BzrError, UnlistableStore, NoSuchFile
24
24
from bzrlib.store import copy_all
25
25
from bzrlib.transport.local import LocalTransport
26
 
from bzrlib.transport import NoSuchFile
27
26
from bzrlib.store.text import TextStore
28
 
from bzrlib.selftest import TestCase, TestCaseInTempDir
 
27
from bzrlib.tests import TestCase, TestCaseInTempDir
29
28
import bzrlib.store as store
30
29
import bzrlib.transport as transport
31
30
from bzrlib.transport.memory import MemoryTransport
80
79
 
81
80
class TestCompressedTextStore(TestCaseInTempDir, TestStores):
82
81
 
83
 
    def get_store(self, path='.'):
 
82
    def get_store(self, path=u'.'):
84
83
        t = LocalTransport(path)
85
84
        return TextStore(t, compressed=True)
86
85
 
87
86
    def test_total_size(self):
88
 
        store = self.get_store('.')
 
87
        store = self.get_store(u'.')
89
88
        store.register_suffix('dsc')
90
89
        store.add(StringIO('goodbye'), '123123')
91
90
        store.add(StringIO('goodbye2'), '123123', 'dsc')
93
92
        self.assertEqual(store.total_size(), (2, 55))
94
93
        
95
94
    def test__relpath_suffixed(self):
96
 
        my_store = TextStore(MockTransport(), True, compressed=True)
 
95
        my_store = TextStore(MockTransport(),
 
96
                             prefixed=True, compressed=True)
97
97
        my_store.register_suffix('dsc')
98
98
        self.assertEqual('45/foo.dsc', my_store._relpath('foo', ['dsc']))
99
99
 
139
139
 
140
140
class TestTextStore(TestCaseInTempDir, TestStores):
141
141
 
142
 
    def get_store(self, path='.'):
 
142
    def get_store(self, path=u'.'):
143
143
        t = LocalTransport(path)
144
144
        return TextStore(t, compressed=False)
145
145
 
157
157
 
158
158
class TestMixedTextStore(TestCaseInTempDir, TestStores):
159
159
 
160
 
    def get_store(self, path='.', compressed=True):
 
160
    def get_store(self, path=u'.', compressed=True):
161
161
        t = LocalTransport(path)
162
162
        return TextStore(t, compressed=compressed)
163
163
 
164
164
    def test_get_mixed(self):
165
 
        cs = self.get_store('.', compressed=True)
166
 
        s = self.get_store('.', compressed=False)
 
165
        cs = self.get_store(u'.', compressed=True)
 
166
        s = self.get_store(u'.', compressed=False)
167
167
        cs.add(StringIO('hello there'), 'a')
168
168
 
169
169
        self.failUnlessExists('a.gz')
358
358
                         my_store.get('missing', 'sig').read())
359
359
 
360
360
    def test___iter__no_suffix(self):
361
 
        my_store = TextStore(MemoryTransport(), False, compressed=False)
 
361
        my_store = TextStore(MemoryTransport(),
 
362
                             prefixed=False, compressed=False)
362
363
        stream = StringIO("content")
363
364
        my_store.add(stream, "foo")
364
365
        self.assertEqual(set(['foo']),
383
384
 
384
385
    def test_copy_suffixes(self):
385
386
        from_store = self.get_populated_store()
386
 
        to_store = TextStore(MemoryTransport(), True, compressed=True)
 
387
        to_store = TextStore(MemoryTransport(),
 
388
                             prefixed=True, compressed=True)
387
389
        to_store.register_suffix('sig')
388
390
        copy_all(from_store, to_store)
389
391
        self.assertEqual(1, len(to_store))