mirror of
https://gitlab.com/SIGBUS/nyaa.git
synced 2024-11-18 07:47:50 +00:00
b1d187613a
read(1) returns b'' on EOF, which wasn't checked for in a couple cases, which could cause empty/truncated input to enter one of the `while True` loops without hope of exit. programming is hard.
95 lines
3.7 KiB
Python
95 lines
3.7 KiB
Python
import unittest
|
|
|
|
from nyaa import bencode
|
|
|
|
|
|
class TestBencode(unittest.TestCase):
|
|
|
|
def test_pairwise(self):
|
|
# test list with even length
|
|
initial = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
|
|
expected = [(0, 1), (2, 3), (4, 5), (6, 7), (8, 9)]
|
|
|
|
for index, values in enumerate(bencode._pairwise(initial)):
|
|
self.assertEqual(values, expected[index])
|
|
|
|
# test list with odd length
|
|
initial = [0, 1, 2, 3, 4]
|
|
expected = [(0, 1), (2, 3), 4]
|
|
|
|
for index, values in enumerate(bencode._pairwise(initial)):
|
|
self.assertEqual(values, expected[index])
|
|
|
|
# test non-iterable
|
|
initial = b'012345'
|
|
expected = [(48, 49), (50, 51), (52, 53)] # decimal ASCII
|
|
for index, values in enumerate(bencode._pairwise(initial)):
|
|
self.assertEqual(values, expected[index])
|
|
|
|
def test_encode(self):
|
|
exception_test_cases = [ # (raw, raised_exception, expected_result_regexp)
|
|
# test unsupported type
|
|
(None, bencode.BencodeException,
|
|
r'Unsupported type'),
|
|
(1.6, bencode.BencodeException,
|
|
r'Unsupported type'),
|
|
]
|
|
|
|
test_cases = [ # (raw, expected_result)
|
|
(100, b'i100e'), # int
|
|
(-5, b'i-5e'), # int
|
|
('test', b'4:test'), # str
|
|
(b'test', b'4:test'), # byte
|
|
(['test', 100], b'l4:testi100ee'), # list
|
|
({'numbers': [1, 2], 'hello': 'world'}, b'd5:hello5:world7:numbersli1ei2eee') # dict
|
|
]
|
|
|
|
for raw, raised_exception, expected_result_regexp in exception_test_cases:
|
|
self.assertRaisesRegexp(raised_exception, expected_result_regexp, bencode.encode, raw)
|
|
|
|
for raw, expected_result in test_cases:
|
|
self.assertEqual(bencode.encode(raw), expected_result)
|
|
|
|
def test_decode(self):
|
|
exception_test_cases = [ # (raw, raised_exception, expected_result_regexp)
|
|
# test malformed bencode
|
|
(b'l4:hey', bencode.MalformedBencodeException,
|
|
r'Read only \d+ bytes, \d+ wanted'),
|
|
(b'ie', bencode.MalformedBencodeException,
|
|
r'Unable to parse int'),
|
|
(b'i64', bencode.MalformedBencodeException,
|
|
r'EOF, expecting more integer'),
|
|
(b'', bencode.MalformedBencodeException,
|
|
r'EOF, expecting kind'),
|
|
(b'i6-4', bencode.MalformedBencodeException,
|
|
r'Unexpected input while reading an integer'),
|
|
(b'4#string', bencode.MalformedBencodeException,
|
|
r'Unexpected input while reading string length'),
|
|
(b'4', bencode.MalformedBencodeException,
|
|
r'EOF, expecting more string len'),
|
|
(b'$:string', bencode.MalformedBencodeException,
|
|
r'Unexpected data type'),
|
|
(b'd5:world7:numbersli1ei2eee', bencode.MalformedBencodeException,
|
|
r'Uneven amount of key/value pairs'),
|
|
]
|
|
|
|
test_cases = [ # (raw, expected_result)
|
|
(b'i100e', 100), # int
|
|
(b'i-5e', -5), # int
|
|
('4:test', b'test'), # str
|
|
(b'4:test', b'test'), # byte
|
|
(b'15:thisisalongone!', b'thisisalongone!'), # big byte
|
|
(b'l4:testi100ee', [b'test', 100]), # list
|
|
(b'd5:hello5:world7:numbersli1ei2eee', {'hello': b'world', 'numbers': [1, 2]}) # dict
|
|
]
|
|
|
|
for raw, raised_exception, expected_result_regexp in exception_test_cases:
|
|
self.assertRaisesRegexp(raised_exception, expected_result_regexp, bencode.decode, raw)
|
|
|
|
for raw, expected_result in test_cases:
|
|
self.assertEqual(bencode.decode(raw), expected_result)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|