← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_bencode.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-03 01:37:27 UTC
  • mfrom: (2694.5.22 bencode-pyrex)
  • mto: This revision was merged to the branch mainline in revision 4410.
  • Revision ID: john@arbash-meinel.com-20090603013727-wtxochxrd0zqf1vg
Merge the bencode implementation.

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._bencode_py as py_module
 
25
    scenarios = [('python', {'bencode': py_module})]
 
26
    if CompiledBencodeFeature.available():
 
27
        import bzrlib._bencode_c 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_c
 
44
        except ImportError:
 
45
            return False
 
46
        return True
 
47
 
 
48
    def feature_name(self):
 
49
        return 'bzrlib._bencode_c'
 
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, 'i' + ('1' * 1000) + ':')
 
105
 
 
106
    def test_list(self):
 
107
        self._check([], 'le')
 
108
        self._check(['', '', ''], 'l0:0:0:e')
 
109
        self._check([1, 2, 3], 'li1ei2ei3ee')
 
110
        self._check(['asd', 'xy'], 'l3:asd2:xye')
 
111
        self._check([['Alice', 'Bob'], [2, 3]], 'll5:Alice3:Bobeli2ei3eee')
 
112
 
 
113
    def test_list_deepnested(self):
 
114
        self._run_check_error(RuntimeError, ("l" * 10000) + ("e" * 10000))
 
115
 
 
116
    def test_malformed_list(self):
 
117
        self._run_check_error(ValueError, 'l')
 
118
        self._run_check_error(ValueError, 'l01:ae')
 
119
        self._run_check_error(ValueError, 'l0:')
 
120
        self._run_check_error(ValueError, 'li1e')
 
121
        self._run_check_error(ValueError, 'l-3:e')
 
122
 
 
123
    def test_dict(self):
 
124
        self._check({}, 'de')
 
125
        self._check({'':3}, 'd0:i3ee')
 
126
        self._check({'age': 25, 'eyes': 'blue'}, 'd3:agei25e4:eyes4:bluee')
 
127
        self._check({'spam.mp3': {'author': 'Alice', 'length': 100000}},
 
128
                            'd8:spam.mp3d6:author5:Alice6:lengthi100000eee')
 
129
 
 
130
    def test_dict_deepnested(self):
 
131
        self._run_check_error(RuntimeError, ("d0:" * 10000) + 'i1e' + ("e" * 10000))
 
132
 
 
133
    def test_malformed_dict(self):
 
134
        self._run_check_error(ValueError, 'd')
 
135
        self._run_check_error(ValueError, 'defoobar')
 
136
        self._run_check_error(ValueError, 'd3:fooe')
 
137
        self._run_check_error(ValueError, 'di1e0:e')
 
138
        self._run_check_error(ValueError, 'd1:b0:1:a0:e')
 
139
        self._run_check_error(ValueError, 'd1:a0:1:a0:e')
 
140
        self._run_check_error(ValueError, 'd0:0:')
 
141
        self._run_check_error(ValueError, 'd0:')
 
142
        self._run_check_error(ValueError, 'd432432432432432432:e')
 
143
 
 
144
    def test_empty_string(self):
 
145
        self.assertRaises(ValueError, self.bencode.bdecode, '')
 
146
 
 
147
    def test_junk(self):
 
148
        self._run_check_error(ValueError, 'i6easd')
 
149
        self._run_check_error(ValueError, '2:abfdjslhfld')
 
150
        self._run_check_error(ValueError, '0:0:')
 
151
        self._run_check_error(ValueError, 'leanfdldjfh')
 
152
 
 
153
    def test_unknown_object(self):
 
154
        self.assertRaises(ValueError, self.bencode.bdecode, 'relwjhrlewjh')
 
155
 
 
156
    def test_unsupported_type(self):
 
157
        self._run_check_error(TypeError, float(1.5))
 
158
        self._run_check_error(TypeError, None)
 
159
        self._run_check_error(TypeError, lambda x: x)
 
160
        self._run_check_error(TypeError, object)
 
161
        self._run_check_error(TypeError, u"ie")
 
162
 
 
163
    def test_decoder_type_error(self):
 
164
        self.assertRaises(TypeError, self.bencode.bdecode, 1)
 
165
 
 
166
 
 
167
class TestBencodeEncode(tests.TestCase):
 
168
 
 
169
    bencode = None
 
170
 
 
171
    def _check(self, expected, source):
 
172
        self.assertEquals(expected, self.bencode.bencode(source))
 
173
 
 
174
    def test_int(self):
 
175
        self._check('i4e', 4)
 
176
        self._check('i0e', 0)
 
177
        self._check('i-10e', -10)
 
178
 
 
179
    def test_long(self):
 
180
        self._check('i12345678901234567890e', 12345678901234567890L)
 
181
        self._check('i-12345678901234567890e', -12345678901234567890L)
 
182
 
 
183
    def test_string(self):
 
184
        self._check('0:', '')
 
185
        self._check('3:abc', 'abc')
 
186
        self._check('10:1234567890', '1234567890')
 
187
 
 
188
    def test_list(self):
 
189
        self._check('le', [])
 
190
        self._check('li1ei2ei3ee', [1, 2, 3])
 
191
        self._check('ll5:Alice3:Bobeli2ei3eee', [['Alice', 'Bob'], [2, 3]])
 
192
 
 
193
    def test_list_as_tuple(self):
 
194
        self._check('le', ())
 
195
        self._check('li1ei2ei3ee', (1, 2, 3))
 
196
        self._check('ll5:Alice3:Bobeli2ei3eee', (('Alice', 'Bob'), (2, 3)))
 
197
 
 
198
    def test_list_deep_nested(self):
 
199
        top = []
 
200
        l = top
 
201
        for i in range(10000):
 
202
            l.append([])
 
203
            l = l[0]
 
204
        self.assertRaises(RuntimeError, self.bencode.bencode, 
 
205
            top)
 
206
 
 
207
    def test_dict(self):
 
208
        self._check('de', {})
 
209
        self._check('d3:agei25e4:eyes4:bluee', {'age': 25, 'eyes': 'blue'})
 
210
        self._check('d8:spam.mp3d6:author5:Alice6:lengthi100000eee',
 
211
                            {'spam.mp3': {'author': 'Alice',
 
212
                                          'length': 100000}})
 
213
 
 
214
    def test_dict_deep_nested(self):
 
215
        d = top = {}
 
216
        for i in range(10000):
 
217
            d[''] = {}
 
218
            d = d['']
 
219
        self.assertRaises(RuntimeError, self.bencode.bencode, 
 
220
            top)
 
221
 
 
222
    def test_bencached(self):
 
223
        self._check('i3e', self.bencode.Bencached(self.bencode.bencode(3)))
 
224
 
 
225
    def test_invalid_dict(self):
 
226
        self.assertRaises(TypeError, self.bencode.bencode, {1:"foo"})
 
227
 
 
228
    def test_bool(self):
 
229
        self._check('i1e', True)
 
230
        self._check('i0e', False)
 
231