diff options
| -rw-r--r-- | tests/units/test_0012_post_software.py | 5 | ||||
| -rw-r--r-- | tests/units/test_0028_get_software.py | 32 |
2 files changed, 32 insertions, 5 deletions
diff --git a/tests/units/test_0012_post_software.py b/tests/units/test_0012_post_software.py index f298e35..477a787 100644 --- a/tests/units/test_0012_post_software.py +++ b/tests/units/test_0012_post_software.py @@ -27,11 +27,6 @@ class TestPostSoftwareMethods(unittest.TestCase): json=malformed_payload) self.assertEqual(returned.status_code, 400) - def test_get(self): - returned = requests.get(self.url, headers=self.headers, json={ 'client' : ['192.168.56.11'], - 'disk' : 1, - 'partition' : 1 }) - self.assertEqual(returned.status_code, 200) if __name__ == '__main__': unittest.main() diff --git a/tests/units/test_0028_get_software.py b/tests/units/test_0028_get_software.py new file mode 100644 index 0000000..add26d1 --- /dev/null +++ b/tests/units/test_0028_get_software.py @@ -0,0 +1,32 @@ +import requests +import unittest + +class TestGetSoftwareMethods(unittest.TestCase): + + def setUp(self): + self.url = 'http://localhost:8888/software' + self.headers = {'Authorization' : '07b3bfe728954619b58f0107ad73acc1'} + self.json = { 'client' : [ '192.168.2.1' ], + 'disk' : 1, + 'partition': 1 } + + def test_get(self): + returned = requests.get(self.url, headers=self.headers, json=self.json) + self.assertEqual(returned.status_code, 200) + + def test_no_payload(self): + returned = requests.get(self.url, headers=self.headers, json=None) + self.assertEqual(returned.status_code, 400) + + def test_malformed_payload(self): + for parameter in self.json: + malformed_payload = self.json.copy() + malformed_payload.pop(parameter) + returned = requests.pogetst(self.url, + headers=self.headers, + json=malformed_payload) + self.assertEqual(returned.status_code, 400) + + +if __name__ == '__main__': + unittest.main() |