56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import unittest
|
|
import collections
|
|
|
|
from torrent_parser import TorrentFileParser, parse_torrent_file
|
|
|
|
|
|
class Test(unittest.TestCase):
|
|
REAL_FILE = 'tests/testfiles/real.torrent'
|
|
NEG_FILE = 'tests/testfiles/neg.torrent'
|
|
|
|
def test_parse_torrent_file_use_shortcut(self):
|
|
parse_torrent_file(self.REAL_FILE)
|
|
|
|
def test_parse_torrent_file_use_class(self):
|
|
with open(self.REAL_FILE, 'rb') as fp:
|
|
TorrentFileParser(fp).parse()
|
|
|
|
def test_encoding_auto(self):
|
|
with open(self.REAL_FILE, 'rb') as fp:
|
|
TorrentFileParser(fp, encoding='auto').parse()
|
|
|
|
def test_parse_torrent_file_to_ordered_dict(self):
|
|
data = parse_torrent_file(self.REAL_FILE, True)
|
|
self.assertIsInstance(data, collections.OrderedDict)
|
|
|
|
with open(self.REAL_FILE, 'rb') as fp:
|
|
data = TorrentFileParser(fp, True).parse()
|
|
self.assertIsInstance(data, collections.OrderedDict)
|
|
|
|
def test_parse_correctness(self):
|
|
data = parse_torrent_file(self.REAL_FILE)
|
|
self.assertIn(['udp://tracker.publicbt.com:80/announce'],
|
|
data['announce-list'])
|
|
self.assertEqual(data['creation date'], 1409254242)
|
|
|
|
def test_parse_two_times(self):
|
|
with open(self.REAL_FILE, 'rb') as fp:
|
|
parser = TorrentFileParser(fp)
|
|
data = parser.parse()
|
|
self.assertIn(['udp://tracker.publicbt.com:80/announce'],
|
|
data['announce-list'])
|
|
self.assertEqual(data['creation date'], 1409254242)
|
|
data = parser.parse()
|
|
self.assertIn(['udp://tracker.publicbt.com:80/announce'],
|
|
data['announce-list'])
|
|
self.assertEqual(data['creation date'], 1409254242)
|
|
|
|
def test_int_is_negative(self):
|
|
data = parse_torrent_file(self.NEG_FILE)
|
|
self.assertEqual(data['neg'], -1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |