diff options
| -rw-r--r-- | tests/units/test_0003_shellrun.py | 27 |
1 files changed, 25 insertions, 2 deletions
diff --git a/tests/units/test_0003_shellrun.py b/tests/units/test_0003_shellrun.py index cd0a8e1..ada1795 100644 --- a/tests/units/test_0003_shellrun.py +++ b/tests/units/test_0003_shellrun.py @@ -48,7 +48,7 @@ class TestShellRunMethods(unittest.TestCase): def test_no_json(self): c = Client() s = Server() - s.connect(probe=False) + s.connect() s.send('POST /shell/run HTTP/1.0\r\nContent-Length:0\r\n\r\n') response = s.recv() s.stop() @@ -62,12 +62,35 @@ class TestShellRunMethods(unittest.TestCase): '\r\nContent-Type:application/json\r\n\r\n' + json c = Client() s = Server() - s.connect(probe=False) + s.connect() s.send(msg) response = s.recv() s.stop() c.stop() self.assertEqual(response, 'HTTP/1.0 400 Bad Request\r\n\r\n') + def test_serial_requests(self): + req1_json = '{"run":"echo 1", "echo":true}' + req1 = 'POST /shell/run HTTP/1.0\r\nContent-Length:'+ \ + str(len(req1_json)) + \ + '\r\nContent-Type:application/json\r\n\r\n' + req1_json + req2_json = '{"run":"echo 2", "echo":true}' + req2 = 'POST /shell/run HTTP/1.0\r\nContent-Length:'+ \ + str(len(req2_json)) + \ + '\r\nContent-Type:application/json\r\n\r\n' + req2_json + response_json = '{"out": "2\n"}' + resp = 'HTTP/1.0 200 OK\r\nContent-Length:' + \ + str(len(response_json)) + \ + '\r\nContent-Type:application/json\r\n\r\n' + response_json + c = Client() + s = Server() + s.connect() + s.send(req1) + s.send(req2) + client_response = s.recv() + s.stop() + c.stop() + self.assertEqual(client_response, client_response) + if __name__ == '__main__': unittest.main() |