summaryrefslogtreecommitdiffstats
path: root/tests/units
diff options
context:
space:
mode:
authorRoberto Hueso Gómez <rhueso@soleta.eu>2020-01-16 11:33:26 +0100
committerAlvaro Neira Ayuso <alvaroneay@gmail.com>2020-01-19 19:50:44 +0100
commitbb65bd7bf03da7ad5a9ae25d6caab57d06211b99 (patch)
tree03f12a9e7b8e12a4ff3ea9efa20969ac0b693935 /tests/units
parent0c5cbee301db4be515fd9224f93c026857976488 (diff)
Add unit testing basic structure
Diffstat (limited to 'tests/units')
-rw-r--r--tests/units/__init__.py1
-rw-r--r--tests/units/client.py17
-rw-r--r--tests/units/server.py42
3 files changed, 60 insertions, 0 deletions
diff --git a/tests/units/__init__.py b/tests/units/__init__.py
new file mode 100644
index 0000000..8b13789
--- /dev/null
+++ b/tests/units/__init__.py
@@ -0,0 +1 @@
+
diff --git a/tests/units/client.py b/tests/units/client.py
new file mode 100644
index 0000000..4e780a7
--- /dev/null
+++ b/tests/units/client.py
@@ -0,0 +1,17 @@
+import subprocess
+import os
+
+class Client():
+
+ def __init__(self):
+ self.null = open(os.devnull, 'wb')
+ self.proc = subprocess.Popen(['python3', 'main.py'],
+ cwd='../',
+ stdout=self.null,
+ stderr=self.null)
+
+ def stop(self):
+ self.proc.terminate()
+ self.proc.kill()
+ self.proc.wait()
+ self.null.close()
diff --git a/tests/units/server.py b/tests/units/server.py
new file mode 100644
index 0000000..764ae6a
--- /dev/null
+++ b/tests/units/server.py
@@ -0,0 +1,42 @@
+import socket
+import sys
+
+HOST = '127.0.0.1'
+PORT = '1234'
+
+class Server():
+
+ _probe_json = '{"id": 0, "name": "test_local", "center": 0, "room": 0}'
+ _probe_msg = 'POST /probe HTTP/1.0\r\nContent-Length:'+ \
+ str(len(_probe_json)) + \
+ '\r\nContent-Type:application/json\r\n\r\n' + _probe_json
+
+ def __init__(self, host='127.0.0.1', port=1234):
+ self.host = host
+ self.port = port
+ self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+ self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+
+ def connect(self, probe=True):
+ try:
+ self.sock.bind((self.host, self.port))
+ except socket.error as msg:
+ print('Bind failed. Error Code : ' + str(msg[0]) + ' Message '
+ + msg[1])
+ sys.exit()
+
+ self.sock.listen(10)
+ self.conn, self.addr = self.sock.accept()
+ if probe:
+ self.send(self._probe_msg)
+ return self.recv()
+
+ def send(self, msg):
+ self.conn.send(msg.encode())
+
+ def recv(self):
+ return self.conn.recv(1024).decode('utf-8')
+
+ def stop(self):
+ self.conn.close()
+ self.sock.close()