diff options
| -rw-r--r-- | tests/units/test_0007_software.py | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/tests/units/test_0007_software.py b/tests/units/test_0007_software.py new file mode 100644 index 0000000..0984c56 --- /dev/null +++ b/tests/units/test_0007_software.py @@ -0,0 +1,38 @@ +# +# Copyright (C) 2020 Soleta Networks <info@soleta.eu> +# +# This program is free software: you can redistribute it and/or modify it under +# the terms of the GNU Affero General Public License as published by the +# Free Software Foundation, version 3. +# + +from server import Server +from client import Client +import unittest +import json + +class TestSoftwareMethods(unittest.TestCase): + + def test_correct_post(self): + req_json = '{"disk": 1, "partition": 1}' + req = 'POST /software HTTP/1.0\r\nContent-Length:' + \ + str(len(req_json)) + \ + '\r\nContent-Type:application/json\r\n\r\n' + req_json + c = Client() + s = Server() + s.connect() + s.send(req) + client_response = s.recv() + s.stop() + c.stop() + + self.assertRegex(client_response, '^HTTP/1.0 200 OK\r\n*') + client_response = client_response.split('\r\n\r\n') + response_json = json.loads(client_response[1]) + + self.assertIn('partition', response_json) + self.assertIn('software', response_json) + self.assertIn('disk', response_json) + +if __name__ == '__main__': + unittest.main() |