diff options
| -rw-r--r-- | test/unit.py | 52 |
1 files changed, 39 insertions, 13 deletions
diff --git a/test/unit.py b/test/unit.py index 4b9c8531..3ed7b495 100644 --- a/test/unit.py +++ b/test/unit.py @@ -82,20 +82,38 @@ class TestUnitControl(TestUnit): def get(self, path='/'):
with self._control_sock() as sock:
- sock.sendall(('GET ' + path
- + ' HTTP/1.1\r\nHost: localhost\r\n\r\n').encode())
- r = self._recvall(sock)
+ req = ('GET ' + path
+ + ' HTTP/1.1\r\nHost: localhost\r\n\r\n').encode()
- return self._body_json(r)
+ sock.sendall(req)
+
+ if '--verbose' in sys.argv:
+ print('>>>\n', req)
+
+ resp = self._recvall(sock)
+
+ if '--verbose' in sys.argv:
+ print('<<<\n', resp)
+
+ return self._body_json(resp)
def delete(self, path='/'):
with self._control_sock() as sock:
- sock.sendall(('DELETE ' + path
- + ' HTTP/1.1\r\nHost: localhost\r\n\r\n').encode())
- r = self._recvall(sock)
+ req = ('DELETE ' + path
+ + ' HTTP/1.1\r\nHost: localhost\r\n\r\n').encode()
+
+ sock.sendall(req)
+
+ if '--verbose' in sys.argv:
+ print('>>>\n', req)
- return self._body_json(r)
+ resp = self._recvall(sock)
+
+ if '--verbose' in sys.argv:
+ print('<<<\n', resp)
+
+ return self._body_json(resp)
def put(self, path='/', data=''):
@@ -103,12 +121,20 @@ class TestUnitControl(TestUnit): data = data.encode()
with self._control_sock() as sock:
- sock.sendall(('PUT ' + path + (' HTTP/1.1\nHost: localhost\n'
- 'Content-Length: ') + str(len(data)) + '\r\n\r\n').encode()
- + data)
- r = self._recvall(sock)
+ req = ('PUT ' + path + (' HTTP/1.1\nHost: localhost\n'
+ 'Content-Length: ') + str(len(data)) + '\r\n\r\n').encode() + data
+
+ sock.sendall(req)
+
+ if '--verbose' in sys.argv:
+ print('>>>\n', req)
+
+ resp = self._recvall(sock)
+
+ if '--verbose' in sys.argv:
+ print('<<<\n', resp)
- return self._body_json(r)
+ return self._body_json(resp)
def _control_sock(self):
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|