error-tracker/tests/FlaskTest/test_end_point.py at dev · sonus21/error-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
# -*- coding: utf-8 -*-
#
#    Test all end points are working as expected
#
#    :copyright: 2019 Sonu Kumar
#    :license: BSD-3-Clause
#

import unittest

import pyquery
from .utils import TestCaseMixin
from .configs import pagination_config


class ViewTest(TestCaseMixin):
    db_prefix = "ViewTestCase"
    config_module = pagination_config

    def test_list_view(self):
        db_name = "test_list_view"
        app, db, _ = self.setUpApp(db_name)
        with app.test_client() as c:
            c.get('/value-error')
            c.post('/post-view')
            response = c.get('/dev/error', follow_redirects=True)
            html = response.data
            urls = [node.attrib['href'] for node in pyquery.PyQuery(html)('a')]
            # 2 links for delete operation and 2 links to navigate
            self.assertEqual(len(urls), 2 + 2)

            urls = [node.attrib['href'] for node in pyquery.PyQuery(html)('.view-link')]
            self.assertEqual(len(urls), 2)

    def test_detail_view(self):
        db_name = "test_detail_view"
        app, db, _ = self.setUpApp(db_name)
        with app.test_client() as c:
            c.get('/value-error')
            response = c.get('/dev/error', follow_redirects=True)
            html = response.data
            url = [node.attrib['href'] for node in pyquery.PyQuery(html)('.view-link')][0]
            response = c.get(url)
            row = pyquery.PyQuery(response.data)('.row')[0]
            # row = etree.tostring(row, pretty_print=False)
            p = pyquery.PyQuery(row)('p')
            divs = pyquery.PyQuery(row)('.row div')
            self.assertEqual(len(p), 5)
            self.assertEqual(len(divs), 2)

    def test_delete_view(self):
        db_name = "test_delete_view"
        app, db, error_tracker = self.setUpApp(db_name)
        with app.test_client() as c:
            c.get('/value-error')
            response = c.get('/dev/error', follow_redirects=True)
            html = response.data
            url = [node.attrib['href'] for node in pyquery.PyQuery(html)('.delete')][0]
            c.get(url, follow_redirects=True)
            self.assertEqual(len(error_tracker.get_exceptions()), 0)

    def test_pagination(self):
        db_name = "test_pagination"
        app, db, error_tracker = self.setUpApp(db_name)
        distinct_exceptions = set()

        with app.test_client() as c:
            c.get('/value-error')
            exception = error_tracker.get_exceptions()[0]
            hashx = exception.hash
            inserted = 0
            i = 0
            while inserted < 20:
                i += 1
                idx = str(i) + hashx[2:]
                inserted += 1
                error_tracker.create_or_update_exception(idx, exception.host, exception.path,
                                                         exception.method, exception.request_data,
                                                         exception.exception_name,
                                                         exception.traceback)

            response = c.get('/dev/error', follow_redirects=True)
            urls = [node.attrib['href'] for node in pyquery.PyQuery(response.data)('a')]
            self.assertEqual(len(urls), pagination_config.APP_DEFAULT_LIST_SIZE * 2 + 1)
            self.assertTrue('/dev/error/?page=2' in urls)

            response = c.get('/dev/error?page=2', follow_redirects=True)
            urls = [node.attrib['href'] for node in pyquery.PyQuery(response.data)('a')]
            self.assertEqual(len(urls), pagination_config.APP_DEFAULT_LIST_SIZE * 2 + 2)
            self.assertTrue('/dev/error/?page=1' in urls)
            self.assertTrue('/dev/error/?page=3' in urls)

            response = c.get('/dev/error?page=5', follow_redirects=True)
            urls = [node.attrib['href'] for node in pyquery.PyQuery(response.data)('a')]
            self.assertTrue('/dev/error/?page=4' in urls)

            response = c.get('/dev/error?page=6', follow_redirects=True)
            urls = [node.attrib['href'] for node in pyquery.PyQuery(response.data)('a')]
            self.assertEqual(len(urls), 1)


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