diff --git a/test/main.py b/test/main.py index c2610fa2a2af2acfb7b0cca484aa4b6eee935b07..0195ad446be39e08bb80fc9d25cf8134467ae6e1 100644 --- a/test/main.py +++ b/test/main.py @@ -5,6 +5,7 @@ from test_bench_sendfile import TestBenchSendfile from test_bench_benchmark import TestBenchBenchmark from test_bench_status import TestBenchStatus from test_bench_available import TestBenchAvailable +from test_bench_terminate import TestBenchTerminate def RunModelCase(): @@ -13,6 +14,7 @@ def RunModelCase(): suite.addTest(TestBenchBenchmark('test_bench_server_FUN_benchmark')) suite.addTest(TestBenchStatus('test_bench_server_FUN_status')) suite.addTest(TestBenchAvailable('test_bench_server_FUN_available')) + suite.addTest(TestBenchTerminate('test_bench_server_FUN_terminate')) return suite diff --git a/test/test_bench_terminate.py b/test/test_bench_terminate.py new file mode 100644 index 0000000000000000000000000000000000000000..fe91d35111fd1cb1f25039e69f57afabd3cce9a2 --- /dev/null +++ b/test/test_bench_terminate.py @@ -0,0 +1,22 @@ +import json +import requests +import unittest + + +class TestBenchTerminate(unittest.TestCase): + def setUp(self) -> None: + self.proxies={"http": None, "https": None} + url = "http://{}:{}/status".format("localhost", "9874") + re = requests.get(url, proxies=self.proxies) + if re.status_code != 200: + print("ERROR: Can't reach KeenTune-Bench.") + exit() + + def tearDown(self) -> None: + pass + + def test_bench_server_FUN_terminate(self): + url = "http://{}:{}/{}".format("localhost", "9874", "terminate") + result = requests.get(url, proxies=self.proxies) + self.assertEqual(result.status_code, 200) + self.assertIn('"suc": true', result.text)