From 880cc90c14f5d8ffa555d12f691785e79a3d6c54 Mon Sep 17 00:00:00 2001 From: Javier Sánchez Parra Date: Fri, 5 Jul 2019 12:40:06 +0200 Subject: #915 Improve test output using unittest discover * Call the tests with python module unittest. In particular, the discover function. * Rename tests files to meet the default requirements of discover function. --- tests/units/test_0004_post_shell_run.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/units/test_0004_post_shell_run.py (limited to 'tests/units/test_0004_post_shell_run.py') diff --git a/tests/units/test_0004_post_shell_run.py b/tests/units/test_0004_post_shell_run.py new file mode 100644 index 0000000..7be1fa9 --- /dev/null +++ b/tests/units/test_0004_post_shell_run.py @@ -0,0 +1,20 @@ +import requests +import unittest + +class TestPostShellRunMethods(unittest.TestCase): + + def setUp(self): + self.url = 'http://localhost:8888/shell/run' + self.headers = {'Authorization' : '07b3bfe728954619b58f0107ad73acc1'} + self.json = { 'clients' : [ '192.168.2.1', '192.168.2.2' ], 'run' : 'ls' } + + def test_post(self): + returned = requests.post(self.url, headers=self.headers, json=self.json) + self.assertEqual(returned.status_code, 200) + + def test_get(self): + returned = requests.post(self.url, headers=self.headers) + self.assertEqual(returned.status_code, 404) + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3-18-g5258