48k.eu-mirror
/
ogclient
mirror of https://git.48k.eu/ogclient
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add /probe tests for multiple probes and extra parameter in json

more_events
Roberto Hueso Gómez 2020-01-17 13:45:06 +01:00 committed by Alvaro Neira Ayuso
parent 025e4da5a9
commit 2c5e477dfb
1 changed files with 32 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/units/test_0001_probe.py
View File

@ -12,15 +12,18 @@ import unittest
class TestProbeMethods(unittest.TestCase): class TestProbeMethods(unittest.TestCase):
def setUp(self):
self.ok_response = 'HTTP/1.0 200 OK\r\nContent-Length:17\r\n' \
'Content-Type:application/json\r\n\r\n' + \
'{"status": "OPG"}'
def test_post(self): 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() c = Client()
s = Server() s = Server()
server_response = s.connect() server_response = s.connect()
s.stop() s.stop()
c.stop() c.stop()
self.assertEqual(server_response, msg) self.assertEqual(server_response, self.ok_response)
def test_no_json(self): def test_no_json(self):
c = Client() c = Client()
@ -46,5 +49,31 @@ class TestProbeMethods(unittest.TestCase):
c.stop() c.stop()
self.assertEqual(response, 'HTTP/1.0 400 Bad Request\r\n\r\n') self.assertEqual(response, 'HTTP/1.0 400 Bad Request\r\n\r\n')
def test_multiple_probes(self):
c = Client()
s = Server()
s.connect(probe=False)
s.send(s._probe_msg)
s.send(s._probe_msg)
server_response = s.recv()
s.stop()
c.stop()
self.assertEqual(server_response, self.ok_response)
def test_extra_parameter_json(self):
json = '{"id": 0, "name": "test_local", "center": 0, "room": 0, ' + \
'"extra_param": true}'
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.assertRegex(response, '^HTTP/1.0 200 OK\r\n*')
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()