Tests: added tests for basic statistics.

1 files changed, 45 insertions, 0 deletions

diff --git a/test/unit/status.py b/test/unit/status.py
new file mode 100644
index 00000000..17416f17
--- /dev/null
+++ b/test/unit/status.py
@@ -0,0 +1,45 @@
+from unit.control import TestControl
+
+
+class Status:
+    _status = None
+    control = TestControl()
+
+    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
+                }
+            else:
+                return d1 - d2
+
+        return find_diffs(Status.control.conf_get('/status'), Status._status)
+
+    def get(path='/'):
+        path = path.split('/')[1:]
+        diff = Status.diff()
+
+        for p in path:
+            diff = diff[p]
+
+        return diff