1 files changed, 40 insertions, 0 deletions
diff --git a/src/pybind/mgr/cli_api/tests/test_cli_api.py b/src/pybind/mgr/cli_api/tests/test_cli_api.py
new file mode 100644
index 000000000..ee42dc96a
--- /dev/null
+++ b/src/pybind/mgr/cli_api/tests/test_cli_api.py
@@ -0,0 +1,40 @@
+import unittest
+
+from ..module import CLI, BenchmarkException, HandleCommandResult
+
+
+class BenchmarkRunnerTest(unittest.TestCase):
+    def setUp(self):
+        self.cli = CLI('CLI', 0, 0)
+
+    def test_number_of_calls_on_start_fails(self):
+        with self.assertRaises(BenchmarkException) as ctx:
+            self.cli.benchmark(0, 10, 'list_servers', [])
+        self.assertEqual(str(ctx.exception),
+                         "Number of calls and number "
+                         "of parallel calls must be greater than 0")
+
+    def test_number_of_parallel_calls_on_start_fails(self):
+        with self.assertRaises(BenchmarkException) as ctx:
+            self.cli.benchmark(100, 0, 'list_servers', [])
+        self.assertEqual(str(ctx.exception),
+                         "Number of calls and number "
+                         "of parallel calls must be greater than 0")
+
+    def test_number_of_parallel_calls_on_start_works(self):
+        CLI.benchmark(10, 10, "get", "osd_map")
+
+    def test_function_name_fails(self):
+        for iterations in [0, 1]:
+            threads = 0 if iterations else 1
+            with self.assertRaises(BenchmarkException) as ctx:
+                self.cli.benchmark(iterations, threads, 'fake_method', [])
+            self.assertEqual(str(ctx.exception),
+                             "Number of calls and number "
+                             "of parallel calls must be greater than 0")
+        result: HandleCommandResult = self.cli.benchmark(1, 1, 'fake_method', [])
+        self.assertEqual(result.stderr, "Could not find the public "
+                         "function you are requesting")
+
+    def test_function_name_works(self):
+        CLI.benchmark(10, 10, "get", "osd_map")