from __future__ import unicode_literals

import collections
import os.path
import unittest

from torrent_parser import TorrentFileParser, parse_torrent_file


class TestParse(unittest.TestCase):
    TEST_FILES_DIR = os.path.join(os.path.dirname(__file__), 'test_files')
    REAL_FILE = os.path.join(TEST_FILES_DIR, 'real.torrent')
    NEG_FILE = os.path.join(TEST_FILES_DIR, 'neg.torrent')
    STRING_FILE = os.path.join(TEST_FILES_DIR, 'outmost.string.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)

    def test_dont_need_dict_outmost(self):
        data = parse_torrent_file(self.STRING_FILE)
        self.assertEqual(data, 'announce')


if __name__ == '__main__':
    unittest.main()