/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_bencode.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-18 18:18:36 UTC
  • mto: This revision was merged to the branch mainline in revision 4461.
  • Revision ID: john@arbash-meinel.com-20090618181836-biodfkat9a8eyzjz
The new add_inventory_by_delta is returning a CHKInventory when mapping from NULL
Which is completely valid, but 'broke' one of the tests.
So to fix it, changed the test to use CHKInventories on both sides, and add an __eq__
member. The nice thing is that CHKInventory.__eq__ is fairly cheap, since it only
has to check the root keys.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_bencode.py
 
1
# Copyright (C) 2007 Canonical 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
"""Tests for bencode structured encoding"""
 
18
 
 
19
from bzrlib import tests
 
20
 
 
21
def load_tests(standard_tests, module, loader):
 
22
    # parameterize all tests in this module
 
23
    suite = loader.suiteClass()
 
24
    import bzrlib.util._bencode_py as py_module
 
25
    scenarios = [('python', {'bencode': py_module})]
 
26
    if CompiledBencodeFeature.available():
 
27
        import bzrlib._bencode_pyx as c_module
 
28
        scenarios.append(('C', {'bencode': c_module}))
 
29
    else:
 
30
        # the compiled module isn't available, so we add a failing test
 
31
        class FailWithoutFeature(tests.TestCase):
 
32
            def test_fail(self):
 
33
                self.requireFeature(CompiledBencodeFeature)
 
34
        suite.addTest(loader.loadTestsFromTestCase(FailWithoutFeature))
 
35
    tests.multiply_tests(standard_tests, scenarios, suite)
 
36
    return suite
 
37
 
 
38
 
 
39
class _CompiledBencodeFeature(tests.Feature):
 
40
 
 
41
    def _probe(self):
 
42
        try:
 
43
            import bzrlib._bencode_pyx
 
44
        except ImportError:
 
45
            return False
 
46
        return True
 
47
 
 
48
    def feature_name(self):
 
49
        return 'bzrlib._bencode_pyx'
 
50
 
 
51
CompiledBencodeFeature = _CompiledBencodeFeature()
 
52
 
 
53
 
 
54
class TestBencodeDecode(tests.TestCase):
 
55
 
 
56
    bencode = None
 
57
 
 
58
    def _check(self, expected, source):
 
59
        self.assertEquals(expected, self.bencode.bdecode(source))
 
60
 
 
61
    def _run_check_error(self, exc, bad):
 
62
        """Check that bdecoding a string raises a particular exception."""
 
63
        self.assertRaises(exc, self.bencode.bdecode, bad)
 
64
 
 
65
    def test_int(self):
 
66
        self._check(0, 'i0e')
 
67
        self._check(4, 'i4e')
 
68
        self._check(123456789, 'i123456789e')
 
69
        self._check(-10, 'i-10e')
 
70
        self._check(int('1' * 1000), 'i' + ('1' * 1000) + 'e')
 
71
 
 
72
    def test_long(self):
 
73
        self._check(12345678901234567890L, 'i12345678901234567890e')
 
74
        self._check(-12345678901234567890L, 'i-12345678901234567890e')
 
75
 
 
76
    def test_malformed_int(self):
 
77
        self._run_check_error(ValueError, 'ie')
 
78
        self._run_check_error(ValueError, 'i-e')
 
79
        self._run_check_error(ValueError, 'i-010e')
 
80
        self._run_check_error(ValueError, 'i-0e')
 
81
        self._run_check_error(ValueError, 'i00e')
 
82
        self._run_check_error(ValueError, 'i01e')
 
83
        self._run_check_error(ValueError, 'i-03e')
 
84
        self._run_check_error(ValueError, 'i')
 
85
        self._run_check_error(ValueError, 'i123')
 
86
        self._run_check_error(ValueError, 'i341foo382e')
 
87
 
 
88
    def test_string(self):
 
89
        self._check('', '0:')
 
90
        self._check('abc', '3:abc')
 
91
        self._check('1234567890', '10:1234567890')
 
92
 
 
93
    def test_large_string(self):
 
94
        self.assertRaises(ValueError, self.bencode.bdecode, "2147483639:foo")
 
95
 
 
96
    def test_malformed_string(self):
 
97
        self._run_check_error(ValueError, '10:x')
 
98
        self._run_check_error(ValueError, '10:')
 
99
        self._run_check_error(ValueError, '10')
 
100
        self._run_check_error(ValueError, '01:x')
 
101
        self._run_check_error(ValueError, '00:')
 
102
        self._run_check_error(ValueError, '35208734823ljdahflajhdf')
 
103
        self._run_check_error(ValueError, '432432432432432:foo')
 
104
        self._run_check_error(ValueError, ' 1:x') # leading whitespace
 
105
        self._run_check_error(ValueError, '-1:x') # negative
 
106
        self._run_check_error(ValueError, '1 x') # space vs colon
 
107
        self._run_check_error(ValueError, '1x') # missing colon
 
108
        self._run_check_error(ValueError, ('1' * 1000) + ':')
 
109
 
 
110
    def test_list(self):
 
111
        self._check([], 'le')
 
112
        self._check(['', '', ''], 'l0:0:0:e')
 
113
        self._check([1, 2, 3], 'li1ei2ei3ee')
 
114
        self._check(['asd', 'xy'], 'l3:asd2:xye')
 
115
        self._check([['Alice', 'Bob'], [2, 3]], 'll5:Alice3:Bobeli2ei3eee')
 
116
 
 
117
    def test_list_deepnested(self):
 
118
        self._run_check_error(RuntimeError, ("l" * 10000) + ("e" * 10000))
 
119
 
 
120
    def test_malformed_list(self):
 
121
        self._run_check_error(ValueError, 'l')
 
122
        self._run_check_error(ValueError, 'l01:ae')
 
123
        self._run_check_error(ValueError, 'l0:')
 
124
        self._run_check_error(ValueError, 'li1e')
 
125
        self._run_check_error(ValueError, 'l-3:e')
 
126
 
 
127
    def test_dict(self):
 
128
        self._check({}, 'de')
 
129
        self._check({'':3}, 'd0:i3ee')
 
130
        self._check({'age': 25, 'eyes': 'blue'}, 'd3:agei25e4:eyes4:bluee')
 
131
        self._check({'spam.mp3': {'author': 'Alice', 'length': 100000}},
 
132
                            'd8:spam.mp3d6:author5:Alice6:lengthi100000eee')
 
133
 
 
134
    def test_dict_deepnested(self):
 
135
        self._run_check_error(RuntimeError, ("d0:" * 10000) + 'i1e' + ("e" * 10000))
 
136
 
 
137
    def test_malformed_dict(self):
 
138
        self._run_check_error(ValueError, 'd')
 
139
        self._run_check_error(ValueError, 'defoobar')
 
140
        self._run_check_error(ValueError, 'd3:fooe')
 
141
        self._run_check_error(ValueError, 'di1e0:e')
 
142
        self._run_check_error(ValueError, 'd1:b0:1:a0:e')
 
143
        self._run_check_error(ValueError, 'd1:a0:1:a0:e')
 
144
        self._run_check_error(ValueError, 'd0:0:')
 
145
        self._run_check_error(ValueError, 'd0:')
 
146
        self._run_check_error(ValueError, 'd432432432432432432:e')
 
147
 
 
148
    def test_empty_string(self):
 
149
        self.assertRaises(ValueError, self.bencode.bdecode, '')
 
150
 
 
151
    def test_junk(self):
 
152
        self._run_check_error(ValueError, 'i6easd')
 
153
        self._run_check_error(ValueError, '2:abfdjslhfld')
 
154
        self._run_check_error(ValueError, '0:0:')
 
155
        self._run_check_error(ValueError, 'leanfdldjfh')
 
156
 
 
157
    def test_unknown_object(self):
 
158
        self.assertRaises(ValueError, self.bencode.bdecode, 'relwjhrlewjh')
 
159
 
 
160
    def test_unsupported_type(self):
 
161
        self._run_check_error(TypeError, float(1.5))
 
162
        self._run_check_error(TypeError, None)
 
163
        self._run_check_error(TypeError, lambda x: x)
 
164
        self._run_check_error(TypeError, object)
 
165
        self._run_check_error(TypeError, u"ie")
 
166
 
 
167
    def test_decoder_type_error(self):
 
168
        self.assertRaises(TypeError, self.bencode.bdecode, 1)
 
169
 
 
170
 
 
171
class TestBencodeEncode(tests.TestCase):
 
172
 
 
173
    bencode = None
 
174
 
 
175
    def _check(self, expected, source):
 
176
        self.assertEquals(expected, self.bencode.bencode(source))
 
177
 
 
178
    def test_int(self):
 
179
        self._check('i4e', 4)
 
180
        self._check('i0e', 0)
 
181
        self._check('i-10e', -10)
 
182
 
 
183
    def test_long(self):
 
184
        self._check('i12345678901234567890e', 12345678901234567890L)
 
185
        self._check('i-12345678901234567890e', -12345678901234567890L)
 
186
 
 
187
    def test_string(self):
 
188
        self._check('0:', '')
 
189
        self._check('3:abc', 'abc')
 
190
        self._check('10:1234567890', '1234567890')
 
191
 
 
192
    def test_list(self):
 
193
        self._check('le', [])
 
194
        self._check('li1ei2ei3ee', [1, 2, 3])
 
195
        self._check('ll5:Alice3:Bobeli2ei3eee', [['Alice', 'Bob'], [2, 3]])
 
196
 
 
197
    def test_list_as_tuple(self):
 
198
        self._check('le', ())
 
199
        self._check('li1ei2ei3ee', (1, 2, 3))
 
200
        self._check('ll5:Alice3:Bobeli2ei3eee', (('Alice', 'Bob'), (2, 3)))
 
201
 
 
202
    def test_list_deep_nested(self):
 
203
        top = []
 
204
        l = top
 
205
        for i in range(10000):
 
206
            l.append([])
 
207
            l = l[0]
 
208
        self.assertRaises(RuntimeError, self.bencode.bencode, 
 
209
            top)
 
210
 
 
211
    def test_dict(self):
 
212
        self._check('de', {})
 
213
        self._check('d3:agei25e4:eyes4:bluee', {'age': 25, 'eyes': 'blue'})
 
214
        self._check('d8:spam.mp3d6:author5:Alice6:lengthi100000eee',
 
215
                            {'spam.mp3': {'author': 'Alice',
 
216
                                          'length': 100000}})
 
217
 
 
218
    def test_dict_deep_nested(self):
 
219
        d = top = {}
 
220
        for i in range(10000):
 
221
            d[''] = {}
 
222
            d = d['']
 
223
        self.assertRaises(RuntimeError, self.bencode.bencode, 
 
224
            top)
 
225
 
 
226
    def test_bencached(self):
 
227
        self._check('i3e', self.bencode.Bencached(self.bencode.bencode(3)))
 
228
 
 
229
    def test_invalid_dict(self):
 
230
        self.assertRaises(TypeError, self.bencode.bencode, {1:"foo"})
 
231
 
 
232
    def test_bool(self):
 
233
        self._check('i1e', True)
 
234
        self._check('i0e', False)
 
235