summaryrefslogtreecommitdiffhomepage
path: root/test/unit/status.py
blob: 95096a96f84a5c8c0e3f4a027619cd4697916fd0 (plain) (blame)
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
from unit.control import Control


class Status:
    _status = None
    control = Control()

    def _check_zeros():
        assert Status.control.conf_get('/status') == {
            'connections': {
                'accepted': 0,
                'active': 0,
                'idle': 0,
                'closed': 0,
            },
            'requests': {'total': 0},
            'applications': {},
        }

    def init(status=None):
        Status._status = (
            status if status is not None else Status.control.conf_get('/status')
        )

    def diff():
        def find_diffs(d1, d2):
            if isinstance(d1, dict) and isinstance(d2, dict):
                return {
                    k: find_diffs(d1.get(k, 0), d2.get(k, 0))
                    for k in d1
                    if k in d2
                }

            return d1 - d2

        return find_diffs(Status.control.conf_get('/status'), Status._status)

    def get(path='/'):
        path_lst = path.split('/')[1:]
        diff = Status.diff()

        for part in path_lst:
            diff = diff[part]

        return diff