/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: John Arbash Meinel
  • Date: 2006-04-25 15:05:42 UTC
  • mfrom: (1185.85.85 bzr-encoding)
  • mto: This revision was merged to the branch mainline in revision 1752.
  • Revision ID: john@arbash-meinel.com-20060425150542-c7b518dca9928691
[merge] the old bzr-encoding changes, reparenting them on bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test__bencode.py
1
 
# Copyright (C) 2007, 2009, 2010, 2016 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
 
import sys
20
 
 
21
 
from .. import tests
22
 
 
23
 
 
24
 
def load_tests(loader, standard_tests, pattern):
25
 
    suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
26
 
                                                 'breezy.util._bencode_py', 'breezy._bencode_pyx')
27
 
    return suite
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
 
 
45
 
class TestBencodeDecode(tests.TestCase):
46
 
 
47
 
    module = None
48
 
 
49
 
    def _check(self, expected, source):
50
 
        self.assertEqual(expected, self.module.bdecode(source))
51
 
 
52
 
    def _run_check_error(self, exc, bad):
53
 
        """Check that bdecoding a string raises a particular exception."""
54
 
        self.assertRaises(exc, self.module.bdecode, bad)
55
 
 
56
 
    def test_int(self):
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')
62
 
 
63
 
    def test_long(self):
64
 
        self._check(12345678901234567890, b'i12345678901234567890e')
65
 
        self._check(-12345678901234567890, b'i-12345678901234567890e')
66
 
 
67
 
    def test_malformed_int(self):
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')
78
 
 
79
 
    def test_string(self):
80
 
        self._check(b'', b'0:')
81
 
        self._check(b'abc', b'3:abc')
82
 
        self._check(b'1234567890', b'10:1234567890')
83
 
 
84
 
    def test_large_string(self):
85
 
        self.assertRaises(ValueError, self.module.bdecode, b"2147483639:foo")
86
 
 
87
 
    def test_malformed_string(self):
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':')
100
 
 
101
 
    def test_list(self):
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')
107
 
 
108
 
    def test_list_deepnested(self):
109
 
        with RecursionLimit():
110
 
            self._run_check_error(RuntimeError, (b"l" * 100) + (b"e" * 100))
111
 
 
112
 
    def test_malformed_list(self):
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')
118
 
 
119
 
    def test_dict(self):
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')
125
 
 
126
 
    def test_dict_deepnested(self):
127
 
        with RecursionLimit():
128
 
            self._run_check_error(
129
 
                RuntimeError, (b"d0:" * 1000) + b'i1e' + (b"e" * 1000))
130
 
 
131
 
    def test_malformed_dict(self):
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')
141
 
 
142
 
    def test_empty_string(self):
143
 
        self.assertRaises(ValueError, self.module.bdecode, b'')
144
 
 
145
 
    def test_junk(self):
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')
150
 
 
151
 
    def test_unknown_object(self):
152
 
        self.assertRaises(ValueError, self.module.bdecode, b'relwjhrlewjh')
153
 
 
154
 
    def test_unsupported_type(self):
155
 
        self._run_check_error(TypeError, float(1.5))
156
 
        self._run_check_error(TypeError, None)
157
 
        self._run_check_error(TypeError, lambda x: x)
158
 
        self._run_check_error(TypeError, object)
159
 
        self._run_check_error(TypeError, u"ie")
160
 
 
161
 
    def test_decoder_type_error(self):
162
 
        self.assertRaises(TypeError, self.module.bdecode, 1)
163
 
 
164
 
 
165
 
class TestBencodeEncode(tests.TestCase):
166
 
 
167
 
    module = None
168
 
 
169
 
    def _check(self, expected, source):
170
 
        self.assertEqual(expected, self.module.bencode(source))
171
 
 
172
 
    def test_int(self):
173
 
        self._check(b'i4e', 4)
174
 
        self._check(b'i0e', 0)
175
 
        self._check(b'i-10e', -10)
176
 
 
177
 
    def test_long(self):
178
 
        self._check(b'i12345678901234567890e', 12345678901234567890)
179
 
        self._check(b'i-12345678901234567890e', -12345678901234567890)
180
 
 
181
 
    def test_string(self):
182
 
        self._check(b'0:', b'')
183
 
        self._check(b'3:abc', b'abc')
184
 
        self._check(b'10:1234567890', b'1234567890')
185
 
 
186
 
    def test_list(self):
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]])
190
 
 
191
 
    def test_list_as_tuple(self):
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)))
195
 
 
196
 
    def test_list_deep_nested(self):
197
 
        top = []
198
 
        l = top
199
 
        for i in range(1000):
200
 
            l.append([])
201
 
            l = l[0]
202
 
        with RecursionLimit():
203
 
            self.assertRaises(RuntimeError, self.module.bencode, top)
204
 
 
205
 
    def test_dict(self):
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}})
210
 
 
211
 
    def test_dict_deep_nested(self):
212
 
        d = 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)
218
 
 
219
 
    def test_bencached(self):
220
 
        self._check(b'i3e', self.module.Bencached(self.module.bencode(3)))
221
 
 
222
 
    def test_invalid_dict(self):
223
 
        self.assertRaises(TypeError, self.module.bencode, {1: b"foo"})
224
 
 
225
 
    def test_bool(self):
226
 
        self._check(b'i1e', True)
227
 
        self._check(b'i0e', False)