1 files changed, 61 insertions, 0 deletions
diff --git a/test/unit/control.py b/test/unit/control.py
new file mode 100644
index 00000000..0b344ed5
--- /dev/null
+++ b/test/unit/control.py
@@ -0,0 +1,61 @@
+import json
+from unit.http import TestHTTP
+
+
+def args_handler(conf_func):
+    def args_wrapper(self, *args):
+        argcount = conf_func.__code__.co_argcount
+        url_default = '/config'
+        conf = None
+
+        if argcount == 2:
+            url = args[0] if len(args) == 1 else url_default
+
+        elif argcount == 3:
+            conf = args[0]
+
+            if isinstance(conf, dict) or isinstance(conf, list):
+                conf = json.dumps(conf)
+
+            url = args[1] if len(args) == 2 else url_default
+
+        url = url if url.startswith('/') else url_default + '/' + url
+        arguments = (self, url) if conf is None else (self, conf, url)
+
+        return json.loads(conf_func(*arguments))
+
+    return args_wrapper
+
+
+class TestControl(TestHTTP):
+
+    # TODO socket reuse
+    # TODO http client
+
+    @args_handler
+    def conf(self, conf, url):
+        return self.put(**self._get_args(url, conf))['body']
+
+    @args_handler
+    def conf_get(self, url):
+        return self.get(**self._get_args(url))['body']
+
+    @args_handler
+    def conf_delete(self, url):
+        return self.delete(**self._get_args(url))['body']
+
+    @args_handler
+    def conf_post(self, conf, url):
+        return self.post(**self._get_args(url, conf))['body']
+
+    def _get_args(self, url, conf=None):
+        args = {
+            'url': url,
+            'sock_type': 'unix',
+            'addr': self.testdir + '/control.unit.sock',
+        }
+
+        if conf is not None:
+            args['body'] = conf
+
+        return args