nyaa/tests/test_utils.py at master · nyaadevs/nyaa · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import unittest
from collections import OrderedDict

from hashlib import sha1
from nyaa import utils


class TestUtils(unittest.TestCase):

    def test_sha1_hash(self):
        bencoded_test_data = b'd5:hello5:world7:numbersli1ei2eee'
        self.assertEqual(
            utils.sha1_hash(bencoded_test_data),
            sha1(bencoded_test_data).digest())

    def test_sorted_pathdict(self):
        initial = {
            'api_handler.py': 11805,
            'routes.py': 34247,
            '__init__.py': 6499,
            'torrents.py': 11948,
            'static': {
                'img': {
                    'nyaa.png': 1200,
                    'sukebei.png': 1100,
                },
                'js': {
                    'main.js': 3000,
                },
            },
            'search.py': 5148,
            'models.py': 24293,
            'templates': {
                'upload.html': 3000,
                'home.html': 1200,
                'layout.html': 23000,
            },
            'utils.py': 14700,
        }
        expected = OrderedDict({
            'static': OrderedDict({
                'img': OrderedDict({
                    'nyaa.png': 1200,
                    'sukebei.png': 1100,
                }),
                'js': OrderedDict({
                    'main.js': 3000,
                }),
            }),
            'templates': OrderedDict({
                'home.html': 1200,
                'layout.html': 23000,
                'upload.html': 3000,
            }),
            '__init__.py': 6499,
            'api_handler.py': 11805,
            'models.py': 24293,
            'routes.py': 34247,
            'search.py': 5148,
            'torrents.py': 11948,
            'utils.py': 14700,
        })
        self.assertDictEqual(utils.sorted_pathdict(initial), expected)

    @unittest.skip('Not yet implemented')
    def test_cached_function(self):
        # TODO: Test with a function that generates something random?
        pass

    def test_flatten_dict(self):
        initial = OrderedDict({
            'static': OrderedDict({
                'img': OrderedDict({
                    'nyaa.png': 1200,
                    'sukebei.png': 1100,
                }),
                'js': OrderedDict({
                    'main.js': 3000,
                }),
                'favicon.ico': 1000,
            }),
            'templates': [
                {'home.html': 1200},
                {'layout.html': 23000},
                {'upload.html': 3000},
            ],
            '__init__.py': 6499,
            'api_handler.py': 11805,
            'models.py': 24293,
            'routes.py': 34247,
            'search.py': 5148,
            'torrents.py': 11948,
            'utils.py': 14700,
        })
        expected = {
            'static/img/nyaa.png': 1200,
            'static/img/sukebei.png': 1100,
            'static/js/main.js': 3000,
            'static/favicon.ico': 1000,
            'templates/home.html': 1200,
            'templates/layout.html': 23000,
            'templates/upload.html': 3000,
            '__init__.py': 6499,
            'api_handler.py': 11805,
            'models.py': 24293,
            'routes.py': 34247,
            'search.py': 5148,
            'utils.py': 14700,
            'torrents.py': 11948,
        }
        self.assertDictEqual(utils.flatten_dict(initial), expected)


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