Renaming project to json-api for clarity sake

json-api/client-tests/web/__init__.py

@@ -0,0 +1 @@
+from . import http

json-api/client-tests/web/http.py

@@ -0,0 +1,126 @@
+import sys
+import requests
+import time
+import json
+
+class RequestError(Exception):
+    pass
+
+class Response:
+    '''
+    Response is wrapper for reading + extracting information we care about 
+    Primarily created by Requests that get `make`'d.
+    '''
+    def __init__(self, method: str, url: str, body: str, code: int, expected: int, out=sys.stdout, color=True,
+                 truncate_long_body=True):
+
+        self.method = method
+        self.url = url
+        self.body = body #typically a string before parsing anything
+        self.code = code #u16
+        self.expected = expected #u16
+
+        self.out = out # file handle to write to normally sys.stdout
+        self.color = color # bool telling if log should color anything (on by default)
+        self.truncate_long_body = truncate_long_body
+
+    def _color(self, cc, string):
+        nc = '\033[0m'
+        return f'{cc}{string}{nc}'
+
+    def _color_failing(self, string):
+        red = '\033[1;31m'
+        return self._color(red, string)
+
+
+    def _color_passing(self, string):
+        green = '\033[1;32m'
+        return self._color(green, string)
+
+    def __write_msg(self, s):
+        # mega dumb wrapper to reduce visual noise i think
+        if (len(s) == 1 or len(s) == 0) is False: print(s, file=self.out)
+
+    def _log_body(self):
+        if self.truncate_long_body:
+            if len(self.body) > 80:
+                msg = self.body
+                while len(msg) > 80:
+                    msg = msg[:len(msg)//2] + msg[len(msg)//2+1:]
+                msg = '.....'.join([msg[:40], msg[44]])
+
+                self.__write_msg(msg)
+            else:
+                self.__write_msg(f'\t{self.body}')
+        else:
+            self.__write_msg(f'\t{self.body}')
+
+    def log(self):
+        if self.code != self.expected:
+            msg = f'Failed {self.method.upper()} {self.url}\t{self.code} expected {self.expected}'
+            if self.color:
+                msg = self._color_failing(msg)
+
+            self.__write_msg(msg)
+            self._log_body()
+        else:
+            msg = f'Passing: {self.method} {self.url}'
+            if self.color:
+                msg = self._color_passing(msg)
+
+            self.__write_msg(msg)
+
+    def json(self):
+        '''
+        Force an interpretation of json from the body
+        NOTE: this method is rather obtuse and a massive afterthough so its usage
+        should be limited as much as possible
+        '''
+        try:
+            return json.loads(self.body)
+        except:
+            return {}
+
+    def __str__(self):
+        '''
+        Returns: str(Response) -> `code => response.bdoy`
+        '''
+        return f'{self.code} => {self.body}'
+
+class Request: 
+    def __init__(self, method: str, url: str, params: dict):
+        assert(method in ['get', 'post', 'delete'])
+
+        self.method = method
+        self.url    = url
+        self.params = params
+
+    def _make_request(self, method: str, hope: int):
+        # Lower driver for actuall making the request we are looking for
+        method = method.lower()
+
+        params = json.dumps(self.params)
+        if method == 'get':
+            resp = requests.get(self.url, data=params)
+            return Response('get', self.url, resp.text, resp.status_code, hope)
+        elif method == 'post':
+            resp = requests.post(self.url, data=params)
+            return Response('post', self.url, resp.text, resp.status_code, hope)
+        elif method == 'delete':
+            resp = requests.delete(self.url, data=params)
+            return Response('delete', self.url, resp.text, resp.status_code, hope)
+
+        else:
+            raise RequestError('Invalid method passed')
+
+        return resp
+
+
+    def make(self, hope: int) -> Response:
+        '''
+        @param  hope: int -> status code we hope to get back
+        @return Response  -> Wrapper around server http response
+        '''
+        return self._make_request(self.method, hope)
+        
+