1
# Copyright (C) 2005, 2007 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
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
17
"""Test Escaped Stores."""
19
from cStringIO import StringIO
23
from bzrlib import osutils
24
from bzrlib.errors import BzrError, UnlistableStore, NoSuchFile
25
from bzrlib.store.text import TextStore
26
from bzrlib.tests import TestCaseWithTransport
27
import bzrlib.transport
30
class TestEscaped(TestCaseWithTransport):
31
"""Mixin template class that provides some common tests for stores"""
33
def get_store(self, prefixed=False, escaped=True):
34
t = bzrlib.transport.get_transport(self.get_url())
35
return TextStore(t, prefixed=prefixed, escaped=escaped)
38
text_store = self.get_store()
40
self.assertEqual('a', text_store._relpath('a'))
41
self.assertEqual('a', text_store._relpath(u'a'))
42
self.assertEqual('%2520', text_store._relpath(' '))
43
self.assertEqual('%40%253a%253c%253e', text_store._relpath('@:<>'))
44
self.assertEqual('%25c3%25a5', text_store._relpath(u'\xe5'))
46
def test_prefixed(self):
47
# Prefix should be determined by unescaped string
48
text_store = self.get_store(prefixed=True)
50
# hash_prefix() is not defined for unicode characters
51
# it is only defined for byte streams.
52
# so hash_prefix() needs to operate on *at most* utf-8
53
# encoded. However urlutils.escape() does both encoding to utf-8
54
# and urllib quoting, so we will use the escaped form
55
# as the path passed to hash_prefix
57
self.assertEqual('62/a', text_store._relpath('a'))
58
self.assertEqual('88/%2520', text_store._relpath(' '))
59
self.assertEqual('72/%40%253a%253c%253e',
60
text_store._relpath('@:<>'))
61
self.assertEqual('77/%25c3%25a5', text_store._relpath(u'\xe5'))
64
text_store = self.get_store(prefixed=True)
66
text_store.add(StringIO('a'), 'a')
67
self.failUnlessExists('62/a')
69
text_store.add(StringIO('space'), ' ')
70
self.failUnlessExists('88/%20')
71
self.assertEquals('space', text_store.get(' ').read())
73
text_store.add(StringIO('surprise'), '@:<>')
74
self.failUnlessExists('72/@%3a%3c%3e')
75
self.assertEquals('surprise', text_store.get('@:<>').read())
77
text_store.add(StringIO('utf8'), '\xc2\xb5')
78
self.failUnlessExists('77/%c2%b5')
79
self.assertEquals('utf8', text_store.get('\xc2\xb5').read())
82
from bzrlib.store.versioned import WeaveStore
83
from bzrlib.transactions import PassThroughTransaction
85
trans = PassThroughTransaction()
87
t = bzrlib.transport.get_transport(self.get_url())
88
weave_store = WeaveStore(t, prefixed=True, escaped=True)
89
def add_text(file_id, rev_id, contents, parents, transaction):
90
vfile = weave_store.get_weave_or_empty(file_id, transaction)
91
vfile.add_lines(rev_id, parents, contents)
93
def check_text(file_id, revision_id, contents):
94
vfile = weave_store.get_weave(file_id, trans)
95
self.assertEqual(contents, vfile.get_lines(revision_id))
97
add_text('a', 'r', ['a'], [], trans)
98
self.failUnlessExists('62/a.weave')
99
check_text('a', 'r', ['a'])
101
add_text(' ', 'r', ['space'], [], trans)
102
self.failIfExists('21/ .weave')
103
self.failUnlessExists('88/%20.weave')
104
check_text(' ', 'r', ['space'])
106
add_text('@:<>', 'r', ['surprise'], [], trans)
107
self.failUnlessExists('72/@%3a%3c%3e.weave')
108
check_text('@:<>', 'r', ['surprise'])
110
add_text('\xc2\xb5', 'r', ['utf8'], [], trans)
111
self.failUnlessExists('77/%c2%b5.weave')
112
check_text('\xc2\xb5', 'r', ['utf8'])
added
removed
bzrlib/tests/test_escaped_store.py
