path: root/test/unit/status.py

from unit.control import Control


class Status:
    _status = None
    control = Control()

    def _check_zeros():
        status = Status.control.conf_get('/status')

        assert status['connections'] == {
                'accepted': 0,
                'active': 0,
                'idle': 0,
                'closed': 0,
        }
        assert status['requests'] == {'total': 0}
        assert status['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
                }

            if isinstance(d1, str) or isinstance(d1, list):
                return d1 == 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