177d1c7de9
- hash fields method and parameter allow user customize hash field list - hash raw parameter allow the output of hash field to be raw bytes - BEncoder now support encode raw bytes
24 lines
816 B
Python
24 lines
816 B
Python
from __future__ import unicode_literals
|
|
|
|
import os.path
|
|
import unittest
|
|
|
|
from torrent_parser import decode, encode
|
|
|
|
|
|
class TestHashRaw(unittest.TestCase):
|
|
TEST_FILES_DIR = os.path.join(os.path.dirname(__file__), 'test_files')
|
|
FILE = os.path.join(TEST_FILES_DIR, 'utf8.encoding.error.torrent')
|
|
|
|
def test_hash_raw_decode(self):
|
|
data = b'd4:hash4:\xAA\xBB\xCC\xDDe'
|
|
res = decode(data, hash_fields={'hash': (4, False)}, hash_raw=False)
|
|
self.assertEqual(res['hash'], 'aabbccdd')
|
|
res = decode(data, hash_fields={'hash': (4, False)}, hash_raw=True)
|
|
self.assertEqual(res['hash'], b'\xAA\xBB\xCC\xDD')
|
|
|
|
def test_raw_bytes_encode(self):
|
|
res = {'hash': b'\xAA\xBB\xCC\xDD'}
|
|
data = encode(res)
|
|
self.assertEqual(data, b'd4:hash4:\xAA\xBB\xCC\xDDe')
|