/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: Jelmer Vernooij
  • Date: 2009-05-27 08:33:37 UTC
  • mto: (4398.5.1 bencode_serializer)
  • mto: This revision was merged to the branch mainline in revision 4410.
  • Revision ID: jelmer@samba.org-20090527083337-fij2klthhtsdl2hy
Fix copyright headers, add _bencode_py.py to the list of files that do not have to have canonical copyright.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_bencode.py
1
 
# Copyright (C) 2007, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
19
19
from bzrlib import tests
20
20
 
21
21
def load_tests(standard_tests, module, loader):
22
 
    suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
23
 
        'bzrlib.util._bencode_py', 'bzrlib._bencode_pyx')
 
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)
24
36
    return suite
25
37
 
26
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
 
27
54
class TestBencodeDecode(tests.TestCase):
28
55
 
29
 
    module = None
 
56
    bencode = None
30
57
 
31
58
    def _check(self, expected, source):
32
 
        self.assertEquals(expected, self.module.bdecode(source))
33
 
 
34
 
    def _run_check_error(self, exc, bad):
35
 
        """Check that bdecoding a string raises a particular exception."""
36
 
        self.assertRaises(exc, self.module.bdecode, bad)
 
59
        self.assertEquals(expected, self.bencode.bdecode(source))
 
60
 
 
61
    def _run_check(self, pairs):
 
62
        """Run _check for each (expected, source) in pairs list"""
 
63
        for expected, source in pairs:
 
64
            self._check(expected, source)
 
65
 
 
66
    def _run_check_error(self, exc, bads):
 
67
        """Check that bdecoding each string raises a particular exception."""
 
68
        for x in bads:
 
69
            self.assertRaises(exc, self.bencode.bdecode, x)
37
70
 
38
71
    def test_int(self):
39
 
        self._check(0, 'i0e')
40
 
        self._check(4, 'i4e')
41
 
        self._check(123456789, 'i123456789e')
42
 
        self._check(-10, 'i-10e')
43
 
        self._check(int('1' * 1000), 'i' + ('1' * 1000) + 'e')
 
72
        self._run_check([(0, 'i0e'),
 
73
                         (4, 'i4e'),
 
74
                         (123456789, 'i123456789e'),
 
75
                         (-10, 'i-10e')])
44
76
 
45
77
    def test_long(self):
46
 
        self._check(12345678901234567890L, 'i12345678901234567890e')
47
 
        self._check(-12345678901234567890L, 'i-12345678901234567890e')
 
78
        self._run_check([(12345678901234567890L, 'i12345678901234567890e'),
 
79
                         (-12345678901234567890L, 'i-12345678901234567890e')])
48
80
 
49
81
    def test_malformed_int(self):
50
 
        self._run_check_error(ValueError, 'ie')
51
 
        self._run_check_error(ValueError, 'i-e')
52
 
        self._run_check_error(ValueError, 'i-010e')
53
 
        self._run_check_error(ValueError, 'i-0e')
54
 
        self._run_check_error(ValueError, 'i00e')
55
 
        self._run_check_error(ValueError, 'i01e')
56
 
        self._run_check_error(ValueError, 'i-03e')
57
 
        self._run_check_error(ValueError, 'i')
58
 
        self._run_check_error(ValueError, 'i123')
59
 
        self._run_check_error(ValueError, 'i341foo382e')
 
82
        self._run_check_error(ValueError, ['ie', 'i-e', 'i-010e',
 
83
                               'i-0e', 'i00e', 'i01e', 'i-03e',
 
84
                               'i', 'i123', 
 
85
                               'i341foo382e'])
60
86
 
61
87
    def test_string(self):
62
 
        self._check('', '0:')
63
 
        self._check('abc', '3:abc')
64
 
        self._check('1234567890', '10:1234567890')
 
88
        self._run_check([('', '0:'),
 
89
                         ('abc', '3:abc'),
 
90
                         ('1234567890', '10:1234567890')])
65
91
 
66
92
    def test_large_string(self):
67
 
        self.assertRaises(ValueError, self.module.bdecode, "2147483639:foo")
 
93
        self.assertRaises(ValueError, self.bencode.bdecode, "2147483639:foo")
68
94
 
69
95
    def test_malformed_string(self):
70
 
        self._run_check_error(ValueError, '10:x')
71
 
        self._run_check_error(ValueError, '10:')
72
 
        self._run_check_error(ValueError, '10')
73
 
        self._run_check_error(ValueError, '01:x')
74
 
        self._run_check_error(ValueError, '00:')
75
 
        self._run_check_error(ValueError, '35208734823ljdahflajhdf')
76
 
        self._run_check_error(ValueError, '432432432432432:foo')
77
 
        self._run_check_error(ValueError, ' 1:x') # leading whitespace
78
 
        self._run_check_error(ValueError, '-1:x') # negative
79
 
        self._run_check_error(ValueError, '1 x') # space vs colon
80
 
        self._run_check_error(ValueError, '1x') # missing colon
81
 
        self._run_check_error(ValueError, ('1' * 1000) + ':')
 
96
        self._run_check_error(ValueError, ['10:x', '10:', '10',
 
97
                               '01:x', '00:',
 
98
                               '35208734823ljdahflajhdf',
 
99
                               '432432432432432:foo'])
82
100
 
83
101
    def test_list(self):
84
 
        self._check([], 'le')
85
 
        self._check(['', '', ''], 'l0:0:0:e')
86
 
        self._check([1, 2, 3], 'li1ei2ei3ee')
87
 
        self._check(['asd', 'xy'], 'l3:asd2:xye')
88
 
        self._check([['Alice', 'Bob'], [2, 3]], 'll5:Alice3:Bobeli2ei3eee')
89
 
 
90
 
    def test_list_deepnested(self):
91
 
        self._run_check_error(RuntimeError, ("l" * 10000) + ("e" * 10000))
 
102
        self._run_check([
 
103
                         ([], 'le'),
 
104
                         (['', '', ''], 'l0:0:0:e'),
 
105
                         ([1, 2, 3], 'li1ei2ei3ee'),
 
106
                         (['asd', 'xy'], 'l3:asd2:xye'),
 
107
                         ([['Alice', 'Bob'], [2, 3]],
 
108
                              'll5:Alice3:Bobeli2ei3eee'),
 
109
                        ])
92
110
 
93
111
    def test_malformed_list(self):
94
 
        self._run_check_error(ValueError, 'l')
95
 
        self._run_check_error(ValueError, 'l01:ae')
96
 
        self._run_check_error(ValueError, 'l0:')
97
 
        self._run_check_error(ValueError, 'li1e')
98
 
        self._run_check_error(ValueError, 'l-3:e')
 
112
        self._run_check_error(ValueError, [
 
113
            'l', 'l01:ae', 'l0:', 'li1e', 'l-3:e'])
99
114
 
100
115
    def test_dict(self):
101
 
        self._check({}, 'de')
102
 
        self._check({'':3}, 'd0:i3ee')
103
 
        self._check({'age': 25, 'eyes': 'blue'}, 'd3:agei25e4:eyes4:bluee')
104
 
        self._check({'spam.mp3': {'author': 'Alice', 'length': 100000}},
105
 
                            'd8:spam.mp3d6:author5:Alice6:lengthi100000eee')
106
 
 
107
 
    def test_dict_deepnested(self):
108
 
        self._run_check_error(RuntimeError, ("d0:" * 10000) + 'i1e' + ("e" * 10000))
 
116
        self._run_check([({}, 'de'),
 
117
                         ({'':3}, 'd0:i3ee'),
 
118
                         ({'age': 25, 'eyes': 'blue'},
 
119
                            'd3:agei25e4:eyes4:bluee'),
 
120
                         ({'spam.mp3': {'author': 'Alice', 'length': 100000}},
 
121
                            'd8:spam.mp3d6:author5:Alice6:lengthi100000eee')])
109
122
 
110
123
    def test_malformed_dict(self):
111
 
        self._run_check_error(ValueError, 'd')
112
 
        self._run_check_error(ValueError, 'defoobar')
113
 
        self._run_check_error(ValueError, 'd3:fooe')
114
 
        self._run_check_error(ValueError, 'di1e0:e')
115
 
        self._run_check_error(ValueError, 'd1:b0:1:a0:e')
116
 
        self._run_check_error(ValueError, 'd1:a0:1:a0:e')
117
 
        self._run_check_error(ValueError, 'd0:0:')
118
 
        self._run_check_error(ValueError, 'd0:')
119
 
        self._run_check_error(ValueError, 'd432432432432432432:e')
 
124
        self._run_check_error(ValueError, ['d', 'defoobar',
 
125
                               'd3:fooe', 'di1e0:e',
 
126
                               'd1:b0:1:a0:e',
 
127
                               'd1:a0:1:a0:e',
 
128
                               'd0:0:', 'd0:',
 
129
                               'd432432432432432432:e', ])
120
130
 
121
131
    def test_empty_string(self):
122
 
        self.assertRaises(ValueError, self.module.bdecode, '')
 
132
        self.assertRaises(ValueError, self.bencode.bdecode, '')
123
133
 
124
134
    def test_junk(self):
125
 
        self._run_check_error(ValueError, 'i6easd')
126
 
        self._run_check_error(ValueError, '2:abfdjslhfld')
127
 
        self._run_check_error(ValueError, '0:0:')
128
 
        self._run_check_error(ValueError, 'leanfdldjfh')
 
135
        self._run_check_error(ValueError, ['i6easd', '2:abfdjslhfld',
 
136
                               '0:0:', 'leanfdldjfh'])
129
137
 
130
138
    def test_unknown_object(self):
131
 
        self.assertRaises(ValueError, self.module.bdecode, 'relwjhrlewjh')
 
139
        self.assertRaises(ValueError, self.bencode.bdecode, 'relwjhrlewjh')
132
140
 
133
141
    def test_unsupported_type(self):
134
 
        self._run_check_error(TypeError, float(1.5))
135
 
        self._run_check_error(TypeError, None)
136
 
        self._run_check_error(TypeError, lambda x: x)
137
 
        self._run_check_error(TypeError, object)
138
 
        self._run_check_error(TypeError, u"ie")
 
142
        self._run_check_error(TypeError, [
 
143
            float(1.5), None, lambda x: x, object, u"ie"])
139
144
 
140
145
    def test_decoder_type_error(self):
141
 
        self.assertRaises(TypeError, self.module.bdecode, 1)
 
146
        self.assertRaises(TypeError, self.bencode.bdecode, 1)
142
147
 
143
148
 
144
149
class TestBencodeEncode(tests.TestCase):
145
150
 
146
 
    module = None
 
151
    bencode = None
147
152
 
148
153
    def _check(self, expected, source):
149
 
        self.assertEquals(expected, self.module.bencode(source))
 
154
        self.assertEquals(expected, self.bencode.bencode(source))
 
155
 
 
156
    def _run_check(self, pairs):
 
157
        for expected, source in pairs:
 
158
            self._check(expected, source)
150
159
 
151
160
    def test_int(self):
152
 
        self._check('i4e', 4)
153
 
        self._check('i0e', 0)
154
 
        self._check('i-10e', -10)
 
161
        self._run_check([('i4e', 4),
 
162
                         ('i0e', 0),
 
163
                         ('i-10e', -10)])
155
164
 
156
165
    def test_long(self):
157
 
        self._check('i12345678901234567890e', 12345678901234567890L)
158
 
        self._check('i-12345678901234567890e', -12345678901234567890L)
 
166
        self._run_check([('i12345678901234567890e', 12345678901234567890L),
 
167
                         ('i-12345678901234567890e', -12345678901234567890L)])
159
168
 
160
169
    def test_string(self):
161
 
        self._check('0:', '')
162
 
        self._check('3:abc', 'abc')
163
 
        self._check('10:1234567890', '1234567890')
 
170
        self._run_check([('0:', ''),
 
171
                         ('3:abc', 'abc'),
 
172
                         ('10:1234567890', '1234567890')])
164
173
 
165
174
    def test_list(self):
166
 
        self._check('le', [])
167
 
        self._check('li1ei2ei3ee', [1, 2, 3])
168
 
        self._check('ll5:Alice3:Bobeli2ei3eee', [['Alice', 'Bob'], [2, 3]])
 
175
        self._run_check([('le', []),
 
176
                         ('li1ei2ei3ee', [1, 2, 3]),
 
177
                         ('ll5:Alice3:Bobeli2ei3eee',
 
178
                            [['Alice', 'Bob'], [2, 3]])
 
179
                        ])
169
180
 
170
181
    def test_list_as_tuple(self):
171
 
        self._check('le', ())
172
 
        self._check('li1ei2ei3ee', (1, 2, 3))
173
 
        self._check('ll5:Alice3:Bobeli2ei3eee', (('Alice', 'Bob'), (2, 3)))
174
 
 
175
 
    def test_list_deep_nested(self):
176
 
        top = []
177
 
        l = top
178
 
        for i in range(10000):
179
 
            l.append([])
180
 
            l = l[0]
181
 
        self.assertRaises(RuntimeError, self.module.bencode, 
182
 
            top)
 
182
        self._run_check([('le', ()),
 
183
                         ('li1ei2ei3ee', (1, 2, 3)),
 
184
                         ('ll5:Alice3:Bobeli2ei3eee',
 
185
                            (('Alice', 'Bob'), (2, 3)))
 
186
                        ])
183
187
 
184
188
    def test_dict(self):
185
 
        self._check('de', {})
186
 
        self._check('d3:agei25e4:eyes4:bluee', {'age': 25, 'eyes': 'blue'})
187
 
        self._check('d8:spam.mp3d6:author5:Alice6:lengthi100000eee',
 
189
        self._run_check([('de', {}),
 
190
                         ('d3:agei25e4:eyes4:bluee',
 
191
                            {'age': 25, 'eyes': 'blue'}),
 
192
                         ('d8:spam.mp3d6:author5:Alice6:lengthi100000eee',
188
193
                            {'spam.mp3': {'author': 'Alice',
189
194
                                          'length': 100000}})
190
 
 
191
 
    def test_dict_deep_nested(self):
192
 
        d = top = {}
193
 
        for i in range(10000):
194
 
            d[''] = {}
195
 
            d = d['']
196
 
        self.assertRaises(RuntimeError, self.module.bencode, 
197
 
            top)
 
195
                         ])
198
196
 
199
197
    def test_bencached(self):
200
 
        self._check('i3e', self.module.Bencached(self.module.bencode(3)))
 
198
        self._check('i3e', self.bencode.Bencached(self.bencode.bencode(3)))
201
199
 
202
200
    def test_invalid_dict(self):
203
 
        self.assertRaises(TypeError, self.module.bencode, {1:"foo"})
 
201
        self.assertRaises(TypeError, self.bencode.bencode, {1:"foo"})
204
202
 
205
203
    def test_bool(self):
206
 
        self._check('i1e', True)
207
 
        self._check('i0e', False)
 
204
        self._run_check([('i1e', True),
 
205
                         ('i0e', False)])
208
206