~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_escaped_store.py

  • Committer: Martin Pool
  • Date: 2006-04-12 04:45:32 UTC
  • mfrom: (1608.2.13 bzr.mbp.escape-stores)
  • mto: This revision was merged to the branch mainline in revision 1657.
  • Revision ID: mbp@sourcefrog.net-20060412044532-fc8c5c9408aae88b
[merge][wip] Storage escaping

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005 by Canonical Development Ltd
 
2
 
 
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.
 
7
 
 
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.
 
12
 
 
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
 
16
 
 
17
"""Test Escaped Stores."""
 
18
 
 
19
from cStringIO import StringIO
 
20
import os
 
21
import gzip
 
22
 
 
23
from bzrlib.errors import BzrError, UnlistableStore, NoSuchFile
 
24
from bzrlib.store import copy_all
 
25
from bzrlib.store.text import TextStore
 
26
from bzrlib.tests import TestCaseWithTransport
 
27
import bzrlib.transport
 
28
 
 
29
 
 
30
class TestEscaped(TestCaseWithTransport):
 
31
    """Mixin template class that provides some common tests for stores"""
 
32
 
 
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)
 
36
 
 
37
    def test_paths(self):
 
38
        text_store = self.get_store()
 
39
 
 
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'))
 
45
 
 
46
    def test_prefixed(self):
 
47
        # Prefix should be determined by unescaped string
 
48
        text_store = self.get_store(prefixed=True)
 
49
 
 
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 urlescape() 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
 
56
 
 
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'))
 
62
 
 
63
    def test_files(self):
 
64
        text_store = self.get_store(prefixed=True)
 
65
 
 
66
        text_store.add(StringIO('a'), 'a')
 
67
        self.failUnlessExists('62/a')
 
68
 
 
69
        text_store.add(StringIO('space'), ' ')
 
70
        self.failUnlessExists('88/%20')
 
71
        self.assertEquals('space', text_store.get(' ').read())
 
72
 
 
73
        text_store.add(StringIO('surprise'), '@:<>')
 
74
        self.failUnlessExists('72/@%3a%3c%3e')
 
75
        self.assertEquals('surprise', text_store.get('@:<>').read())
 
76
 
 
77
        text_store.add(StringIO('unicode'), u'\xe5')
 
78
        self.failUnlessExists('77/%c3%a5')
 
79
        self.assertEquals('unicode', text_store.get(u'\xe5').read())
 
80
 
 
81
    def test_weave(self):
 
82
        from bzrlib.store.versioned import WeaveStore
 
83
        from bzrlib.transactions import PassThroughTransaction
 
84
 
 
85
        trans = PassThroughTransaction()
 
86
 
 
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)
 
92
 
 
93
        add_text('a', 'r', ['a'], [], trans)
 
94
        self.failUnlessExists('62/a.weave')
 
95
        self.assertEqual(['a'], weave_store.get_lines('a', 'r', trans))
 
96
 
 
97
        add_text(' ', 'r', ['space'], [], trans)
 
98
        self.failIfExists('21/ .weave')
 
99
        self.failUnlessExists('88/%20.weave')
 
100
        self.assertEquals(['space'], weave_store.get_lines(' ', 'r', trans))
 
101
 
 
102
        add_text('@:<>', 'r', ['surprise'], [], trans)
 
103
        self.failUnlessExists('72/@%3a%3c%3e.weave')
 
104
        self.assertEquals(['surprise'], weave_store.get_lines('@:<>', 'r', trans))
 
105
 
 
106
        add_text(u'\xe5', 'r', ['unicode'], [], trans)
 
107
        self.failUnlessExists('77/%c3%a5.weave')
 
108
        self.assertEquals(['unicode'], weave_store.get_lines(u'\xe5', 'r', trans))
 
109