imageloader/test_imageloader.py at master · tfrisk/imageloader · 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
"""Test cases for Image Loader

(C) 2017 Teemu Frisk
Distributed under the MIT License
"""

import unittest
from imageloader import ImageLoader

class ImageLoaderTest(unittest.TestCase):
    """Test cases"""
    def test_config_case1(self):
        """Config arguments case1"""
        loader = ImageLoader(['-u', 'http://www.google.com', '-d'])
        self.assertEqual(loader.args.url, 'http://www.google.com')
        self.assertEqual(loader.debug_mode, True)
        self.assertEqual(loader.args.dir, 'saved-images')

    def test_config_case2(self):
        """Config arguments case2"""
        loader = ImageLoader(['-u', 'http://www.google.com', '-D', 'dldir'])
        self.assertEqual(loader.args.url, 'http://www.google.com')
        self.assertEqual(loader.debug_mode, False)
        self.assertEqual(loader.args.dir, 'dldir')

    def test_downloadrawurl_ok(self):
        """Get OK response from proper url"""
        loader = ImageLoader(['-u', 'http://www.google.com'])
        _, status = loader.get_url_response("http://www.google.com/")
        self.assertEqual(status, 200)

    def test_geturlresponse_notfound(self):
        """Get error response from incorrect url"""
        loader = ImageLoader(['-u', 'http://www.google.com'])
        _, status = loader.get_url_response("http://www.google.com/this-does-not-exist.xml")
        self.assertEqual(status, 404)

    def test_find_key_or_empty(self):
        """Test dictionary key wrapper"""
        val = ImageLoader.find_key_or_empty({"a":1, "b":2}, "a")
        self.assertEqual(val, 1)
        val = ImageLoader.find_key_or_empty({"a":1, "b":2}, "c")
        self.assertEqual(val, '')
        val = ImageLoader.find_key_or_empty({}, "a")
        self.assertEqual(val, '')

    def test_form_proposed_url(self):
        """Test url forming"""
        loader = ImageLoader(['-u', 'http://www.google.com'])
        url1 = ImageLoader.form_proposed_url(loader, "//google.com/media/test.png", "append")
        self.assertEqual(url1, "http://google.com/media/test.png")
        url2 = ImageLoader.form_proposed_url(loader, "/media/test.png", "combine")
        self.assertEqual(url2, "http://www.google.com//media/test.png")

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