sudoku-api/test_server.py at master · navshaikh/sudoku-api · 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
import json
import unittest
from collections import namedtuple
from app import app

Sudoku = namedtuple('Sudoku', ['puzzle', 'solution'])


class SudokuAPITesT(unittest.TestCase):

    def setUp(self):
        app.config['TESTING'] = True
        self.sudoku = Sudoku(
            '.1...8...3.472169...6....1....9.253..421.378..358.6....9....1...213874.9...5...2.',
            '719638254354721698286495317678942531942153786135876942893264175521387469467519823')
        self.client = app.test_client()

    def assert_response(self, response, data_length, puzzles, solutions,
                        status, messages):
        ''' Given a response, unpack the payload and test that client receives
            the right format and contents. Sudoku API will return a response
            of the format:

            {
                data: [
                    {
                        'puzzle': <input>,
                        'solution': <solution if one is found>,
                        'status_code': [ok|error],
                        'message': None if sudoku is solved else error message
                ]
            }
        '''
        response_data = json.loads(response.data)

        self.assertEqual(response.status_code, 200)
        self.assertTrue(response_data)
        self.assertTrue('data' in response_data)
        self.assertTrue(len(response_data['data']) == data_length)

        for ndx, data in enumerate(response_data['data']):
            self.assertEqual(puzzles[ndx], data['puzzle'])
            self.assertEqual(solutions[ndx], data['solution'])
            self.assertEqual(status[ndx].lower(), data['status'].lower())
            if messages[ndx] is None:
                self.assertEqual(messages[ndx], data['message'])
            else:
                self.assertIn(messages[ndx], data['message'])

    def test_missing_key_in_form(self):
        # Send an invalid key 'sudoku1' instead of 'sudoku'
        response = self.client.post('/', data={'sudoku1': self.sudoku.puzzle})
        self.assertEqual(response.status_code, 400)
        self.assertIn('sudoku', response.data)

    def test_form_post(self):
        response = self.client.post('/', data={'sudoku': self.sudoku.puzzle})
        self.assert_response(response, 1, [self.sudoku.puzzle],
                             [self.sudoku.solution], ['ok'], [None])

    def test_invalid_len_in_form(self):
        # Append an extra char to our 81-chars input
        invalid_puzzle = self.sudoku.puzzle + '.'
        response = self.client.post('/', data={'sudoku': invalid_puzzle})
        self.assert_response(response, 1, [invalid_puzzle], [None], ['error'],
                             ['Input length is 82'])

    def test_empty_puzzle_in_form(self):
        response = self.client.post('/', data={'sudoku': ''})
        self.assert_response(response, 1, [''], [None], ['error'],
                             ['Input length is 0'])

    def test_empty_post(self):
        # No form data provided
        response = self.client.post('/')
        self.assertEqual(response.status_code, 400)

    def test_invalid_content_type(self):
        response = self.client.post('/', content_type='blah')
        self.assertEqual(response.status_code, 400)

    def test_json_post(self):
        json_data = json.dumps({'sudoku': [self.sudoku.puzzle]})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assert_response(response, 1, [self.sudoku.puzzle],
                             [self.sudoku.solution], ['ok'], [None])

    def test_missing_key_in_json(self):
        json_data = json.dumps({'sudoku1': [self.sudoku.puzzle]})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assertEqual(response.status_code, 400)
        self.assertIn('sudoku', response.data)

    def test_invalid_len_in_json(self):
        invalid_puzzle = self.sudoku.puzzle + '.'
        json_data = json.dumps({'sudoku': [invalid_puzzle]})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assert_response(response, 1, [invalid_puzzle], [None],
                             ['error'], ['Input length is 82'])

    def test_missing_list_in_json(self):
        json_data = json.dumps({'sudoku': ''})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assertEqual(response.status_code, 400)

    def test_empty_puzzle_in_json(self):
        json_data = json.dumps({'sudoku': ['']})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assert_response(response, 1, [''], [None], ['error'],
                             ['Input length is 0'])

    def test_multiple_puzzles_in_json(self):
        json_data = json.dumps({'sudoku': [self.sudoku.puzzle,
                                           self.sudoku.puzzle]})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assert_response(response, 2,
                             [self.sudoku.puzzle, self.sudoku.puzzle],
                             [self.sudoku.solution, self.sudoku.solution],
                             ['ok', 'ok'], [None, None])

    def test_multiple_invalid_puzzles_in_json(self):
        invalid_puzzle = self.sudoku.puzzle + '.'
        json_data = json.dumps({'sudoku':
                                [self.sudoku.puzzle, invalid_puzzle]})
        response = self.client.post('/', data=json_data,
                                    content_type='application/json')
        self.assert_response(response, 2,
                             [self.sudoku.puzzle, invalid_puzzle],
                             [self.sudoku.solution, None],
                             ['ok', 'error'],
                             [None, 'Input length is 82'])


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