Tests: unit module refactoring.

2019-03-28 18:43:13 +03:00
parent 06b9a11494
commit 19eba1730a
33 changed files with 931 additions and 918 deletions
--- a/test/unit/control.py
+++ b/test/unit/control.py
@@ -0,0 +1,48 @@
+import json
+from unit.http import TestHTTP
+
+
+class TestControl(TestHTTP):
+
+    # TODO socket reuse
+    # TODO http client
+
+    def conf(self, conf, path='/config'):
+        if isinstance(conf, dict) or isinstance(conf, list):
+            conf = json.dumps(conf)
+
+        if path[:1] != '/':
+            path = '/config/' + path
+
+        return json.loads(
+            self.put(
+                url=path,
+                body=conf,
+                sock_type='unix',
+                addr=self.testdir + '/control.unit.sock',
+            )['body']
+        )
+
+    def conf_get(self, path='/config'):
+        if path[:1] != '/':
+            path = '/config/' + path
+
+        return json.loads(
+            self.get(
+                url=path,
+                sock_type='unix',
+                addr=self.testdir + '/control.unit.sock',
+            )['body']
+        )
+
+    def conf_delete(self, path='/config'):
+        if path[:1] != '/':
+            path = '/config/' + path
+
+        return json.loads(
+            self.delete(
+                url=path,
+                sock_type='unix',
+                addr=self.testdir + '/control.unit.sock',
+            )['body']
+        )