← 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
27
from bzrlib.tests import TestCase, TestCaseInTempDir
29
28
import bzrlib.store as store
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')
139
138
 
140
139
class TestTextStore(TestCaseInTempDir, TestStores):
141
140
 
142
 
    def get_store(self, path='.'):
 
141
    def get_store(self, path=u'.'):
143
142
        t = LocalTransport(path)
144
143
        return TextStore(t, compressed=False)
145
144
 
157
156
 
158
157
class TestMixedTextStore(TestCaseInTempDir, TestStores):
159
158
 
160
 
    def get_store(self, path='.', compressed=True):
 
159
    def get_store(self, path=u'.', compressed=True):
161
160
        t = LocalTransport(path)
162
161
        return TextStore(t, compressed=compressed)
163
162
 
164
163
    def test_get_mixed(self):
165
 
        cs = self.get_store('.', compressed=True)
166
 
        s = self.get_store('.', compressed=False)
 
164
        cs = self.get_store(u'.', compressed=True)
 
165
        s = self.get_store(u'.', compressed=False)
167
166
        cs.add(StringIO('hello there'), 'a')
168
167
 
169
168
        self.failUnlessExists('a.gz')