From 7196e7198efa6805a566842522c34096bef122a2 Mon Sep 17 00:00:00 2001 From: Roberto Hueso Gómez Date: Tue, 21 Jan 2020 13:08:52 +0100 Subject: Add /refresh test This test covers a correct request. --- tests/units/test_0008_refresh.py | 43 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 tests/units/test_0008_refresh.py diff --git a/tests/units/test_0008_refresh.py b/tests/units/test_0008_refresh.py new file mode 100644 index 0000000..b8f32d5 --- /dev/null +++ b/tests/units/test_0008_refresh.py @@ -0,0 +1,43 @@ +# +# Copyright (C) 2020 Soleta Networks +# +# 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 TestRefreshMethods(unittest.TestCase): + + def test_correct_get(self): + req = 'GET /refresh HTTP/1.0\r\nContent-Length:0\r\n' + \ + 'Content-Type:application/json\r\n\r\n' + 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_setup', response_json) + self.assertIn('disk', response_json) + + # Check partition_setup parameters. + for partition in response_json['partition_setup']: + self.assertIn('filesystem', partition) + self.assertIn('partition', partition) + self.assertIn('format', partition) + self.assertIn('code', partition) + self.assertIn('size', partition) + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3-18-g5258