1 files changed, 37 insertions, 0 deletions
diff --git a/tests/aiet/test_cli_common.py b/tests/aiet/test_cli_common.py
new file mode 100644
index 0000000..d018e44
--- /dev/null
+++ b/tests/aiet/test_cli_common.py
@@ -0,0 +1,37 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Test for cli common module."""
+from typing import Any
+
+import pytest
+
+from aiet.cli.common import print_command_details
+from aiet.cli.common import raise_exception_at_signal
+
+
+def test_print_command_details(capsys: Any) -> None:
+    """Test print_command_details function."""
+    command = {
+        "command_strings": ["echo test"],
+        "user_params": [
+            {"name": "param_name", "description": "param_description"},
+            {
+                "name": "param_name2",
+                "description": "param_description2",
+                "alias": "alias2",
+            },
+        ],
+    }
+    print_command_details(command)
+    captured = capsys.readouterr()
+    assert "echo test" in captured.out
+    assert "param_name" in captured.out
+    assert "alias2" in captured.out
+
+
+def test_raise_exception_at_signal() -> None:
+    """Test raise_exception_at_signal graceful shutdown."""
+    with pytest.raises(Exception) as err:
+        raise_exception_at_signal(1, "")
+
+    assert str(err.value) == "Middleware shutdown requested"