From 47f3ff0beb56e17bdf38e3d059cd454dfb9f1011 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=82=B5=E9=9D=99?= Date: Mon, 8 May 2023 17:55:18 +0800 Subject: [PATCH] test: add interface test cases for keentune-bench MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit add interface test cases for keentune-bench Signed-off-by: 邵静 --- test/main.py | 2 ++ test/test_bench_terminate.py | 22 ++++++++++++++++++++++ 2 files changed, 24 insertions(+) create mode 100644 test/test_bench_terminate.py diff --git a/test/main.py b/test/main.py index c2610fa..0195ad4 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 0000000..fe91d35 --- /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) -- Gitee