/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 breezy/tests/test__bencode.py

  • Committer: Breezy landing bot
  • Author(s): Colin Watson
  • Date: 2020-11-16 21:47:08 UTC
  • mfrom: (7521.1.1 remove-lp-workaround)
  • Revision ID: breezy.the.bot@gmail.com-20201116214708-jos209mgxi41oy15
Remove breezy.git workaround for bazaar.launchpad.net.

Merged from https://code.launchpad.net/~cjwatson/brz/remove-lp-workaround/+merge/393710

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test__bencode.py
1
 
# Copyright (C) 2007, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2007, 2009, 2010, 2016 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
18
18
 
19
19
import sys
20
20
 
21
 
from bzrlib import tests
22
 
 
23
 
def load_tests(standard_tests, module, loader):
 
21
from .. import tests
 
22
 
 
23
 
 
24
def load_tests(loader, standard_tests, pattern):
24
25
    suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
25
 
        'bzrlib.util._bencode_py', 'bzrlib._bencode_pyx')
 
26
                                                 'breezy.util._bencode_py', 'breezy._bencode_pyx')
26
27
    return suite
27
28
 
28
29
 
 
30
class RecursionLimit(object):
 
31
    """Context manager that lowers recursion limit for testing."""
 
32
 
 
33
    def __init__(self, limit=100):
 
34
        self._new_limit = limit
 
35
        self._old_limit = sys.getrecursionlimit()
 
36
 
 
37
    def __enter__(self):
 
38
        sys.setrecursionlimit(self._new_limit)
 
39
        return self
 
40
 
 
41
    def __exit__(self, *exc_info):
 
42
        sys.setrecursionlimit(self._old_limit)
 
43
 
 
44
 
29
45
class TestBencodeDecode(tests.TestCase):
30
46
 
31
47
    module = None
32
48
 
33
49
    def _check(self, expected, source):
34
 
        self.assertEquals(expected, self.module.bdecode(source))
 
50
        self.assertEqual(expected, self.module.bdecode(source))
35
51
 
36
52
    def _run_check_error(self, exc, bad):
37
53
        """Check that bdecoding a string raises a particular exception."""
38
54
        self.assertRaises(exc, self.module.bdecode, bad)
39
55
 
40
56
    def test_int(self):
41
 
        self._check(0, 'i0e')
42
 
        self._check(4, 'i4e')
43
 
        self._check(123456789, 'i123456789e')
44
 
        self._check(-10, 'i-10e')
45
 
        self._check(int('1' * 1000), 'i' + ('1' * 1000) + 'e')
 
57
        self._check(0, b'i0e')
 
58
        self._check(4, b'i4e')
 
59
        self._check(123456789, b'i123456789e')
 
60
        self._check(-10, b'i-10e')
 
61
        self._check(int('1' * 1000), b'i' + (b'1' * 1000) + b'e')
46
62
 
47
63
    def test_long(self):
48
 
        self._check(12345678901234567890L, 'i12345678901234567890e')
49
 
        self._check(-12345678901234567890L, 'i-12345678901234567890e')
 
64
        self._check(12345678901234567890, b'i12345678901234567890e')
 
65
        self._check(-12345678901234567890, b'i-12345678901234567890e')
50
66
 
51
67
    def test_malformed_int(self):
52
 
        self._run_check_error(ValueError, 'ie')
53
 
        self._run_check_error(ValueError, 'i-e')
54
 
        self._run_check_error(ValueError, 'i-010e')
55
 
        self._run_check_error(ValueError, 'i-0e')
56
 
        self._run_check_error(ValueError, 'i00e')
57
 
        self._run_check_error(ValueError, 'i01e')
58
 
        self._run_check_error(ValueError, 'i-03e')
59
 
        self._run_check_error(ValueError, 'i')
60
 
        self._run_check_error(ValueError, 'i123')
61
 
        self._run_check_error(ValueError, 'i341foo382e')
 
68
        self._run_check_error(ValueError, b'ie')
 
69
        self._run_check_error(ValueError, b'i-e')
 
70
        self._run_check_error(ValueError, b'i-010e')
 
71
        self._run_check_error(ValueError, b'i-0e')
 
72
        self._run_check_error(ValueError, b'i00e')
 
73
        self._run_check_error(ValueError, b'i01e')
 
74
        self._run_check_error(ValueError, b'i-03e')
 
75
        self._run_check_error(ValueError, b'i')
 
76
        self._run_check_error(ValueError, b'i123')
 
77
        self._run_check_error(ValueError, b'i341foo382e')
62
78
 
63
79
    def test_string(self):
64
 
        self._check('', '0:')
65
 
        self._check('abc', '3:abc')
66
 
        self._check('1234567890', '10:1234567890')
 
80
        self._check(b'', b'0:')
 
81
        self._check(b'abc', b'3:abc')
 
82
        self._check(b'1234567890', b'10:1234567890')
67
83
 
68
84
    def test_large_string(self):
69
 
        self.assertRaises(ValueError, self.module.bdecode, "2147483639:foo")
 
85
        self.assertRaises(ValueError, self.module.bdecode, b"2147483639:foo")
70
86
 
71
87
    def test_malformed_string(self):
72
 
        self._run_check_error(ValueError, '10:x')
73
 
        self._run_check_error(ValueError, '10:')
74
 
        self._run_check_error(ValueError, '10')
75
 
        self._run_check_error(ValueError, '01:x')
76
 
        self._run_check_error(ValueError, '00:')
77
 
        self._run_check_error(ValueError, '35208734823ljdahflajhdf')
78
 
        self._run_check_error(ValueError, '432432432432432:foo')
79
 
        self._run_check_error(ValueError, ' 1:x') # leading whitespace
80
 
        self._run_check_error(ValueError, '-1:x') # negative
81
 
        self._run_check_error(ValueError, '1 x') # space vs colon
82
 
        self._run_check_error(ValueError, '1x') # missing colon
83
 
        self._run_check_error(ValueError, ('1' * 1000) + ':')
 
88
        self._run_check_error(ValueError, b'10:x')
 
89
        self._run_check_error(ValueError, b'10:')
 
90
        self._run_check_error(ValueError, b'10')
 
91
        self._run_check_error(ValueError, b'01:x')
 
92
        self._run_check_error(ValueError, b'00:')
 
93
        self._run_check_error(ValueError, b'35208734823ljdahflajhdf')
 
94
        self._run_check_error(ValueError, b'432432432432432:foo')
 
95
        self._run_check_error(ValueError, b' 1:x')  # leading whitespace
 
96
        self._run_check_error(ValueError, b'-1:x')  # negative
 
97
        self._run_check_error(ValueError, b'1 x')  # space vs colon
 
98
        self._run_check_error(ValueError, b'1x')  # missing colon
 
99
        self._run_check_error(ValueError, (b'1' * 1000) + b':')
84
100
 
85
101
    def test_list(self):
86
 
        self._check([], 'le')
87
 
        self._check(['', '', ''], 'l0:0:0:e')
88
 
        self._check([1, 2, 3], 'li1ei2ei3ee')
89
 
        self._check(['asd', 'xy'], 'l3:asd2:xye')
90
 
        self._check([['Alice', 'Bob'], [2, 3]], 'll5:Alice3:Bobeli2ei3eee')
 
102
        self._check([], b'le')
 
103
        self._check([b'', b'', b''], b'l0:0:0:e')
 
104
        self._check([1, 2, 3], b'li1ei2ei3ee')
 
105
        self._check([b'asd', b'xy'], b'l3:asd2:xye')
 
106
        self._check([[b'Alice', b'Bob'], [2, 3]], b'll5:Alice3:Bobeli2ei3eee')
91
107
 
92
108
    def test_list_deepnested(self):
93
 
        self._run_check_error(RuntimeError, ("l" * 10000) + ("e" * 10000))
 
109
        with RecursionLimit():
 
110
            self._run_check_error(RuntimeError, (b"l" * 100) + (b"e" * 100))
94
111
 
95
112
    def test_malformed_list(self):
96
 
        self._run_check_error(ValueError, 'l')
97
 
        self._run_check_error(ValueError, 'l01:ae')
98
 
        self._run_check_error(ValueError, 'l0:')
99
 
        self._run_check_error(ValueError, 'li1e')
100
 
        self._run_check_error(ValueError, 'l-3:e')
 
113
        self._run_check_error(ValueError, b'l')
 
114
        self._run_check_error(ValueError, b'l01:ae')
 
115
        self._run_check_error(ValueError, b'l0:')
 
116
        self._run_check_error(ValueError, b'li1e')
 
117
        self._run_check_error(ValueError, b'l-3:e')
101
118
 
102
119
    def test_dict(self):
103
 
        self._check({}, 'de')
104
 
        self._check({'':3}, 'd0:i3ee')
105
 
        self._check({'age': 25, 'eyes': 'blue'}, 'd3:agei25e4:eyes4:bluee')
106
 
        self._check({'spam.mp3': {'author': 'Alice', 'length': 100000}},
107
 
                            'd8:spam.mp3d6:author5:Alice6:lengthi100000eee')
 
120
        self._check({}, b'de')
 
121
        self._check({b'': 3}, b'd0:i3ee')
 
122
        self._check({b'age': 25, b'eyes': b'blue'}, b'd3:agei25e4:eyes4:bluee')
 
123
        self._check({b'spam.mp3': {b'author': b'Alice', b'length': 100000}},
 
124
                    b'd8:spam.mp3d6:author5:Alice6:lengthi100000eee')
108
125
 
109
126
    def test_dict_deepnested(self):
110
 
        # The recursion here provokes CPython into emitting a warning on
111
 
        # stderr, "maximum recursion depth exceeded in __subclasscheck__", due
112
 
        # to running out of stack space while evaluating "except (...):" in
113
 
        # _bencode_py.  This is harmless, so we temporarily override stderr to
114
 
        # avoid distracting noise in the test output.
115
 
        self.overrideAttr(sys, 'stderr', self._log_file)
116
 
        self._run_check_error(
117
 
            RuntimeError, ("d0:" * 10000) + 'i1e' + ("e" * 10000))
 
127
        with RecursionLimit():
 
128
            self._run_check_error(
 
129
                RuntimeError, (b"d0:" * 1000) + b'i1e' + (b"e" * 1000))
118
130
 
119
131
    def test_malformed_dict(self):
120
 
        self._run_check_error(ValueError, 'd')
121
 
        self._run_check_error(ValueError, 'defoobar')
122
 
        self._run_check_error(ValueError, 'd3:fooe')
123
 
        self._run_check_error(ValueError, 'di1e0:e')
124
 
        self._run_check_error(ValueError, 'd1:b0:1:a0:e')
125
 
        self._run_check_error(ValueError, 'd1:a0:1:a0:e')
126
 
        self._run_check_error(ValueError, 'd0:0:')
127
 
        self._run_check_error(ValueError, 'd0:')
128
 
        self._run_check_error(ValueError, 'd432432432432432432:e')
 
132
        self._run_check_error(ValueError, b'd')
 
133
        self._run_check_error(ValueError, b'defoobar')
 
134
        self._run_check_error(ValueError, b'd3:fooe')
 
135
        self._run_check_error(ValueError, b'di1e0:e')
 
136
        self._run_check_error(ValueError, b'd1:b0:1:a0:e')
 
137
        self._run_check_error(ValueError, b'd1:a0:1:a0:e')
 
138
        self._run_check_error(ValueError, b'd0:0:')
 
139
        self._run_check_error(ValueError, b'd0:')
 
140
        self._run_check_error(ValueError, b'd432432432432432432:e')
129
141
 
130
142
    def test_empty_string(self):
131
 
        self.assertRaises(ValueError, self.module.bdecode, '')
 
143
        self.assertRaises(ValueError, self.module.bdecode, b'')
132
144
 
133
145
    def test_junk(self):
134
 
        self._run_check_error(ValueError, 'i6easd')
135
 
        self._run_check_error(ValueError, '2:abfdjslhfld')
136
 
        self._run_check_error(ValueError, '0:0:')
137
 
        self._run_check_error(ValueError, 'leanfdldjfh')
 
146
        self._run_check_error(ValueError, b'i6easd')
 
147
        self._run_check_error(ValueError, b'2:abfdjslhfld')
 
148
        self._run_check_error(ValueError, b'0:0:')
 
149
        self._run_check_error(ValueError, b'leanfdldjfh')
138
150
 
139
151
    def test_unknown_object(self):
140
 
        self.assertRaises(ValueError, self.module.bdecode, 'relwjhrlewjh')
 
152
        self.assertRaises(ValueError, self.module.bdecode, b'relwjhrlewjh')
141
153
 
142
154
    def test_unsupported_type(self):
143
155
        self._run_check_error(TypeError, float(1.5))
155
167
    module = None
156
168
 
157
169
    def _check(self, expected, source):
158
 
        self.assertEquals(expected, self.module.bencode(source))
 
170
        self.assertEqual(expected, self.module.bencode(source))
159
171
 
160
172
    def test_int(self):
161
 
        self._check('i4e', 4)
162
 
        self._check('i0e', 0)
163
 
        self._check('i-10e', -10)
 
173
        self._check(b'i4e', 4)
 
174
        self._check(b'i0e', 0)
 
175
        self._check(b'i-10e', -10)
164
176
 
165
177
    def test_long(self):
166
 
        self._check('i12345678901234567890e', 12345678901234567890L)
167
 
        self._check('i-12345678901234567890e', -12345678901234567890L)
 
178
        self._check(b'i12345678901234567890e', 12345678901234567890)
 
179
        self._check(b'i-12345678901234567890e', -12345678901234567890)
168
180
 
169
181
    def test_string(self):
170
 
        self._check('0:', '')
171
 
        self._check('3:abc', 'abc')
172
 
        self._check('10:1234567890', '1234567890')
 
182
        self._check(b'0:', b'')
 
183
        self._check(b'3:abc', b'abc')
 
184
        self._check(b'10:1234567890', b'1234567890')
173
185
 
174
186
    def test_list(self):
175
 
        self._check('le', [])
176
 
        self._check('li1ei2ei3ee', [1, 2, 3])
177
 
        self._check('ll5:Alice3:Bobeli2ei3eee', [['Alice', 'Bob'], [2, 3]])
 
187
        self._check(b'le', [])
 
188
        self._check(b'li1ei2ei3ee', [1, 2, 3])
 
189
        self._check(b'll5:Alice3:Bobeli2ei3eee', [[b'Alice', b'Bob'], [2, 3]])
178
190
 
179
191
    def test_list_as_tuple(self):
180
 
        self._check('le', ())
181
 
        self._check('li1ei2ei3ee', (1, 2, 3))
182
 
        self._check('ll5:Alice3:Bobeli2ei3eee', (('Alice', 'Bob'), (2, 3)))
 
192
        self._check(b'le', ())
 
193
        self._check(b'li1ei2ei3ee', (1, 2, 3))
 
194
        self._check(b'll5:Alice3:Bobeli2ei3eee', ((b'Alice', b'Bob'), (2, 3)))
183
195
 
184
196
    def test_list_deep_nested(self):
185
197
        top = []
186
198
        l = top
187
 
        for i in range(10000):
 
199
        for i in range(1000):
188
200
            l.append([])
189
201
            l = l[0]
190
 
        self.assertRaises(RuntimeError, self.module.bencode, 
191
 
            top)
 
202
        with RecursionLimit():
 
203
            self.assertRaises(RuntimeError, self.module.bencode, top)
192
204
 
193
205
    def test_dict(self):
194
 
        self._check('de', {})
195
 
        self._check('d3:agei25e4:eyes4:bluee', {'age': 25, 'eyes': 'blue'})
196
 
        self._check('d8:spam.mp3d6:author5:Alice6:lengthi100000eee',
197
 
                            {'spam.mp3': {'author': 'Alice',
198
 
                                          'length': 100000}})
 
206
        self._check(b'de', {})
 
207
        self._check(b'd3:agei25e4:eyes4:bluee', {b'age': 25, b'eyes': b'blue'})
 
208
        self._check(b'd8:spam.mp3d6:author5:Alice6:lengthi100000eee',
 
209
                    {b'spam.mp3': {b'author': b'Alice', b'length': 100000}})
199
210
 
200
211
    def test_dict_deep_nested(self):
201
212
        d = top = {}
202
 
        for i in range(10000):
203
 
            d[''] = {}
204
 
            d = d['']
205
 
        self.assertRaises(RuntimeError, self.module.bencode, 
206
 
            top)
 
213
        for i in range(1000):
 
214
            d[b''] = {}
 
215
            d = d[b'']
 
216
        with RecursionLimit():
 
217
            self.assertRaises(RuntimeError, self.module.bencode, top)
207
218
 
208
219
    def test_bencached(self):
209
 
        self._check('i3e', self.module.Bencached(self.module.bencode(3)))
 
220
        self._check(b'i3e', self.module.Bencached(self.module.bencode(3)))
210
221
 
211
222
    def test_invalid_dict(self):
212
 
        self.assertRaises(TypeError, self.module.bencode, {1:"foo"})
 
223
        self.assertRaises(TypeError, self.module.bencode, {1: b"foo"})
213
224
 
214
225
    def test_bool(self):
215
 
        self._check('i1e', True)
216
 
        self._check('i0e', False)
217
 
 
 
226
        self._check(b'i1e', True)
 
227
        self._check(b'i0e', False)