Show-Tracker/test.py at master · arch-stack/Show-Tracker · 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
116
117
#!/usr/bin/python2.6

import unittest
import exceptions
import datetime
import tempfile
import os
import shutil
from datetime import datetime

from src.dataclasses import show, season, episode
from src.storage import storage
from src.settings import settings

class dataclasses_datastructures(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_show(self):
        obj = show(u'a', u'b', u'c', u'd', 1, [u'a'], u'e', datetime.min, [u'b'], u'f', u'g', 2.1, 3, u'h')
        self.assertEqual(obj.name, u'a', 'name is assigned incorrectly')
        self.assertEqual(obj.description, u'b', 'description is assigned incorrectly')
        self.assertEqual(obj.image, u'c', 'image is assigned incorrectly')
        self.assertEqual(obj.id, u'd', 'id is assigned incorrectly')
        self.assertEqual(obj.data, 1, 'data is assigned incorrectly')

        self.assertEqual(obj.actors, [u'a'], 'actors is assigned incorrectly')
        self.assertEqual(obj.contentrating, u'e', 'contentrating is assigned incorrectly')
        self.assertEqual(obj.firstaired, datetime.min, 'firstaired is assigned incorrectly')
        self.assertEqual(obj.genre, [u'b'], 'genre is assigned incorrectly')
        self.assertEqual(obj.imdb, u'f', 'imdb is assigned incorrectly')
        self.assertEqual(obj.network, u'g', 'network is assigned incorrectly')
        self.assertEqual(obj.rating, 2.1, 'rating is assigned incorrectly')
        self.assertEqual(obj.runtime, 3, 'runtime is assigned incorrectly')
        self.assertEqual(obj.status, u'h', 'status is assigned incorrectly')

        self.assertRaises(exceptions.TypeError, show, (1, 1, 1, 1, None, '', '', 1, 1, '', 1, 1, '', '', 1))

    def test_season(self):
        obj = season(u'a', u'b', 1, u'c', u'd', 2)
        self.assertEqual(obj.description, u'a', 'description is assigned incorrectly')
        self.assertEqual(obj.image, u'b', 'image is assigned incorrectly')
        self.assertEqual(obj.number, 1, 'number is assigned incorrectly')
        self.assertEqual(obj.id, u'c', 'id is assigned incorrectly')
        self.assertEqual(obj.showid, u'd', 'showid is assigned incorrectly')
        self.assertEqual(obj.data, 2, 'data is assigned incorrectly')

        self.assertRaises(exceptions.TypeError, season, (1, 1, '', 1, 1))

    def test_episode(self):
        obj = episode(u'a', u'b', 1, datetime(2000, 1, 1), u'c', u'd', u'e', True, 2)
        self.assertEqual(obj.name, u'a', 'name is assigned incorrectly')
        self.assertEqual(obj.description, u'b', 'description is assigned incorrectly')
        self.assertEqual(obj.number, 1, 'number is assigned incorrectly')
        self.assertEqual(obj.date, datetime(2000, 1, 1), 'date is assigned incorrectly')
        self.assertEqual(obj.id, u'c', 'id is assigned incorrectly')
        self.assertEqual(obj.showid, u'd', 'showid is assigned incorrectly')
        self.assertEqual(obj.seasonid, u'e', 'seasonid is assigned incorrectly')
        self.assertTrue(obj.watched, 'watched is assigned incorrectly')
        self.assertEqual(obj.data, 2, 'data is assigned incorrectly')

        self.assertRaises(exceptions.TypeError, episode, (1, 1, '', '', 1, 1, 1, 1, ''))

class storage_test(unittest.TestCase):
    def setUp(self):
        self.temppath = tempfile.mkdtemp()

    def tearDown(self):
        if os.path.exists(self.temppath):
            shutil.rmtree(self.temppath, True)

    def test_exists(self):
        obj = storage(self.temppath)
        obj.savedata('testex', '123')
        self.assertTrue(obj.exists('testex'), 'Object should exist')
        self.assertFalse(obj.exists('testnoex'), 'Object shouldn\'t exist')

    def test_saveload(self):
        obj = storage(self.temppath)
        obj.savedata('test', '123')
        self.assertEqual(obj.getdata('test'), '123', 'Data not loaded correctly')

class settings_test(unittest.TestCase):
    def setUp(self):
        self.obj = settings()

    def tearDown(self):
        pass

    def test_default(self):
        self.assertTrue(self.obj.get('test', 'testdef', bool, True))

    def test_conversion(self):
        self.obj.set('test', 'testconv', 1)
        self.assertTrue(self.obj.get('test', 'testconv', bool))

    def test_get(self):
        self.obj.set('test', 'test1', True)
        self.assertFalse(self.assertRaises(exceptions.TypeError, self.obj.get('test', 'test1', bool)))
        self.assertTrue(self.obj.get('test', 'test1', bool), 'Setting not loaded correctly')

    def test_remove(self):
        self.obj.set('test', 'testrem', 1)
        self.obj.remove('test', 'testrem')
        self.assertNotEqual(self.obj.get('test', 'testrem', int, 2), 1)


if __name__ == "__main__":
    suite = unittest.TestSuite([
        unittest.TestLoader().loadTestsFromTestCase(dataclasses_datastructures),
        unittest.TestLoader().loadTestsFromTestCase(storage_test),
        unittest.TestLoader().loadTestsFromTestCase(settings_test)
    ])
    unittest.TextTestRunner(verbosity = 2).run(suite)