summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--tests/units/test_0001_probe.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/tests/units/test_0001_probe.py b/tests/units/test_0001_probe.py
new file mode 100644
index 0000000..dc0cc17
--- /dev/null
+++ b/tests/units/test_0001_probe.py
@@ -0,0 +1,42 @@
+from server import Server
+from client import Client
+import unittest
+
+class TestProbeMethods(unittest.TestCase):
+
+ def test_post(self):
+ msg = 'HTTP/1.0 200 OK\r\nContent-Length:17\r\n' \
+ 'Content-Type:application/json\r\n\r\n{"status": "OPG"}'
+ c = Client()
+ s = Server()
+ server_response = s.connect()
+ s.stop()
+ c.stop()
+ self.assertEqual(server_response, msg)
+
+ def test_no_json(self):
+ c = Client()
+ s = Server()
+ s.connect(probe=False)
+ s.send('POST /probe HTTP/1.0\r\nContent-Length:0\r\n\r\n')
+ response = s.recv()
+ s.stop()
+ c.stop()
+ self.assertEqual(response, 'HTTP/1.0 400 Bad Request\r\n\r\n')
+
+ def test_malformed_json(self):
+ json = '{"id": 0, "name": "test_local", "center": 0}'
+ len_json = str(len(json))
+ msg = 'POST /probe HTTP/1.0\r\nContent-Length:' + len_json + \
+ '\r\nContent-Type:application/json\r\n\r\n' + json
+ c = Client()
+ s = Server()
+ s.connect(probe=False)
+ s.send(msg)
+ response = s.recv()
+ s.stop()
+ c.stop()
+ self.assertEqual(response, 'HTTP/1.0 400 Bad Request\r\n\r\n')
+
+if __name__ == '__main__':
+ unittest.main()