def test_parse_escaped_byte_string(self): decoder = Decoder(b'ab3:\x01\x01\x01f') decoder.idx = 2 b = decoder._Decoder__parse() print(b) self.assertIsInstance(b, bytes, msg='Bencode string returned wrong type.') self.assertEqual(b, b'\x01\x01\x01', msg='Failed to decode bencode escaped byte string.')
def test_nested_complex(self): expected_result = OrderedDict() expected_result[b'KeyA'] = b'Value' expected_result[b'KeyB'] = [b'item1', b'item2'] expected_result[b'KeyC'] = 42 decoder = Decoder(b'd4:KeyA5:Value4:KeyBl5:item15:item2e4:KeyCi42ee') b = decoder._Decoder__parse() self.assertIsInstance(b, OrderedDict, msg='Incorrect type.') self.assertIsInstance(b[b'KeyA'], bytes, msg='Incorrect type.') self.assertIsInstance(b[b'KeyB'], list, msg='Incorrect type.') self.assertIsInstance(b[b'KeyC'], int, msg='Incorrect type.') self.assertDictEqual(b, expected_result, msg='Failed to decode bencode complex type. ')
def test__parse_invalid_character_exception(self): decoder = Decoder(b'abc') decoder.idx = 0 with self.assertRaises(DecodingError): decoder._Decoder__parse()
def test__parse_eof_exception(self): decoder = Decoder(b'abc') decoder.idx = 3 with self.assertRaises(DecodingError): decoder._Decoder__parse()
def test_parse_a_list(self): decoder = Decoder(b'abl1:A1:Becdefghijklmn') decoder.idx = 2 b = decoder._Decoder__parse() self.assertIsInstance(b, list, msg='Bencode list returned wrong type.') self.assertEqual(b, [b'A', b'B'], msg='Failed to decode bencode list.')
def test_parse_a_dictionary(self): decoder = Decoder(b'abd1:A1:Beghijklmn') decoder.idx = 2 b = decoder._Decoder__parse() self.assertIsInstance(b, dict, msg='Bencode dictionary returned wrong type.') self.assertEqual(b, {b'A': b'B'}, msg='Failed to decode bencode dictionary.')
def test_parse_a_integer(self): decoder = Decoder(b'abi42eefhijklmnop') decoder.idx = 2 b = decoder._Decoder__parse() self.assertIsInstance(b, int, msg='Bencode integer returned wrong type.') self.assertEqual(b, 42, msg='Failed to decode bencode integer.')
def test_parse_a_string(self): decoder = Decoder(b'ab3:cdefhijklmnop') decoder.idx = 2 b = decoder._Decoder__parse() self.assertIsInstance(b, bytes, msg='Bencode string returned wrong type.') self.assertEqual(b, b'cde', msg='Failed to decode bencode string.')