From aa6ca1ed5ff4ebc9dffc5470cb3e9a86036507e1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Han?= <seb@redhat.com>
Date: Wed, 5 Mar 2025 16:58:57 +0100
Subject: [PATCH] test: add inspect unit test
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Add test for the inspect endpoint.

Signed-off-by: Sébastien Han <seb@redhat.com>
---
 tests/integration/inspect/__init__.py     |  5 +++++
 tests/integration/inspect/test_inspect.py | 24 +++++++++++++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 tests/integration/inspect/__init__.py
 create mode 100644 tests/integration/inspect/test_inspect.py

diff --git a/tests/integration/inspect/__init__.py b/tests/integration/inspect/__init__.py
new file mode 100644
index 000000000..756f351d8
--- /dev/null
+++ b/tests/integration/inspect/__init__.py
@@ -0,0 +1,5 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
diff --git a/tests/integration/inspect/test_inspect.py b/tests/integration/inspect/test_inspect.py
new file mode 100644
index 000000000..da704178d
--- /dev/null
+++ b/tests/integration/inspect/test_inspect.py
@@ -0,0 +1,24 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
+
+import pytest
+from llama_stack_client import LlamaStackClient
+
+from llama_stack import LlamaStackAsLibraryClient
+
+
+class TestInspect:
+    @pytest.mark.asyncio
+    def test_health(self, llama_stack_client: LlamaStackAsLibraryClient | LlamaStackClient):
+        health = llama_stack_client.inspect.health()
+        assert health is not None
+        assert health.status == "OK"
+
+    @pytest.mark.asyncio
+    def test_version(self, llama_stack_client: LlamaStackAsLibraryClient | LlamaStackClient):
+        version = llama_stack_client.inspect.version()
+        assert version is not None
+        assert version.version is not None