CERP-backend/tests/test_views.py at master · CodeForFoco/CERP-backend · 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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
import unittest
import json
import cerp


class CERPTestCase(unittest.TestCase):

    def setUp(self):
        cerp.app.testing = True
        self.app = cerp.app.test_client()

    def convert_to_json(self, obj):
        return json.loads(obj.data.decode("utf-8"))

    def test_index(self):
        """ Is there an index page in the app """
        self.assertIn(b'<html>', self.app.get('/').data)

    def test_topics(self):
        """ Is there an index page in the app """
        page = self.convert_to_json(
            self.app.get('/api/topics')
        )
        self.assertTrue(page['result'])
        self.assertTrue(isinstance(page['topics'], list))

    def test_api(self):
        """ Insure the self discovery is working """
        endpoints = self.convert_to_json(self.app.get('/api'))
        self.assertIn('/', endpoints['paths'])
        self.assertIn('/api', endpoints['paths'])
        self.assertIn(
            '/api/<topic>/<precinctNum>/pie',
            endpoints['paths']
        )
        self.assertTrue(endpoints['result'])

    def test_presidential_precinct_pie(self):
        """ TEST ENDPOINT /api/<topic>/<precNum>/pie for:
        Valid Number, all, and invalid Number """

        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/2235235101/pie')
        )
        # Result was found
        self.assertTrue(page['result'])
        # data is a list
        self.assertTrue(isinstance(page['data'], list))
        # data has at least 20 items
        self.assertGreater(len(page['data']), 20)
        # data items match designated shape
        for item in page['data']:
            self.assertTrue(isinstance(item, list))
            self.assertTrue(isinstance(item[0], str))
            self.assertTrue(isinstance(item[1], int))

        # Test Bad Case
        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/223523510199999/pie')
        )
        # Result was found
        self.assertFalse(page['result'])
        # data is a list
        self.assertEqual(page['data'], None)
        self.assertTrue(isinstance(page['reason'], str))

        # Test All Case
        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/all/pie')
        )
        # Result was found
        self.assertTrue(page['result'])
        # data is a list
        self.assertTrue(isinstance(page['data'], list))
        # data has at least 100 items
        self.assertEqual(len(page['data']), 22)
        # data items match designated shape
        for item in page['data']:
            self.assertTrue(isinstance(item, list))
            self.assertTrue(isinstance(item[0], str))
            self.assertTrue(isinstance(item[1], int))

    def test_presidential_precinct_meta(self):
        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/2235235101/meta')
        )
        # Result was found
        self.assertTrue(page['result'])
        # data is a dict
        self.assertTrue(isinstance(page['data'], dict))
        # Keys are known, and only what we expect
        keys = set(['percentTurnout', 'registeredVoters', 'totalVotes'])
        self.assertEqual(page['data'].keys(), keys)

        # values are numbers
        for item in page['data']:
            self.assertTrue(isinstance(page['data'][item], (int, float)))

        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/all/meta')
        )
        # Result was found
        self.assertTrue(page['result'])
        # data is a dict
        self.assertTrue(isinstance(page['data'], dict))
        # Keys are known, and only what we expect
        keys = set(['percentTurnout', 'registeredVoters', 'totalVotes'])

        for item in page['data']:
            self.assertEqual(page['data'][item].keys(), keys)
            # values are numbers
            for sub_item in page['data'][item]:
                self.assertTrue(
                    isinstance(
                        page['data'][item][sub_item], (int, float)))

    def test_api_topic_precinctNum_valid(self):
        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/2235235101/valid')
        )
        # Result was found
        self.assertTrue(page['result'])
        self.assertTrue(page['data'])

        page = self.convert_to_json(
            self.app.get('/api/Presidential Election-2016/all/valid')
        )
        # Result was found
        self.assertTrue(page['result'])
        self.assertTrue(isinstance(page['data'], list))

    def test_api_topic_precinctNum_diff(self):
        page = self.convert_to_json(self.app.get(
            '/api/Presidential Election-2016/2235235101/diff?comp1=Clinton/Kaine&comp2=Trump/Pence'))
        # Result was found
        self.assertTrue(page['result'])
        self.assertTrue(page['data'])

        page = self.convert_to_json(self.app.get(
            '/api/Presidential Election-2016/all/diff?comp1=Clinton/Kaine&comp2=Trump/Pence'))
        # Result was found
        self.assertTrue(page['result'])
        self.assertTrue(isinstance(page['data'], dict))

    def test_404_500(self):
        self.assertFalse(
            self.convert_to_json(
                self.app.get('/api/not_an_endpoint'))['result'])

        # Normally in debug mode, the 500 errors are caught and diplayed nicely
        # so we can quickly fix them.  However to test that the api is actually
        # sending a json object when something goes wrong, we need to mimic the
        # production settings
        cerp.app.testing = False
        cerp.app.debug = False
        # This stops scary messages from populating the test environ over
        # something we designed to fail
        cerp.app.logger.disabled = True
        prod_app = cerp.app.test_client()
        self.assertFalse(self.convert_to_json(prod_app.get(
            '/api/Presidential Election-2016/dasfasdfa/meta'))['result'])