New test for client.py

Author: tisnik

tests/unit/test_client.py

@@ -6,6 +6,29 @@
 from models.config import LlamaStackConfiguration
 
 
+def test_client_get_client_method() -> None:
+    """Test how get_client method works for unitialized client."""
+
+    client = LlamaStackClientHolder()
+
+    with pytest.raises(
+        RuntimeError,
+        match="LlamaStackClient has not been initialised. Ensure 'load\\(..\\)' has been called.",
+    ):
+        client.get_client()
+
+
+def test_async_client_get_client_method() -> None:
+    """Test how get_client method works for unitialized client."""
+    client = AsyncLlamaStackClientHolder()
+
+    with pytest.raises(
+        RuntimeError,
+        match="AsyncLlamaStackClient has not been initialised. Ensure 'load\\(..\\)' has been called.",
+    ):
+        client.get_client()
+
+
 def test_get_llama_stack_library_client() -> None:
     """Test if Llama Stack can be initialized in library client mode."""
     cfg = LlamaStackConfiguration(
@@ -18,6 +41,9 @@ def test_get_llama_stack_library_client() -> None:
     client.load(cfg)
     assert client is not None
 
+    ls_client = client.get_client()
+    assert ls_client is not None
+
 
 def test_get_llama_stack_remote_client() -> None:
     """Test if Llama Stack can be initialized in remove client (server) mode."""
@@ -31,6 +57,9 @@ def test_get_llama_stack_remote_client() -> None:
     client.load(cfg)
     assert client is not None
 
+    ls_client = client.get_client()
+    assert ls_client is not None
+
 
 def test_get_llama_stack_wrong_configuration() -> None:
     """Test if configuration is checked before Llama Stack is initialized."""
@@ -61,6 +90,9 @@ async def test_get_async_llama_stack_library_client() -> None:
     await client.load(cfg)
     assert client is not None
 
+    ls_client = client.get_client()
+    assert ls_client is not None
+
 
 async def test_get_async_llama_stack_remote_client() -> None:
     """Test the initialization of asynchronous Llama Stack client in server mode."""
@@ -74,6 +106,9 @@ async def test_get_async_llama_stack_remote_client() -> None:
     await client.load(cfg)
     assert client is not None
 
+    ls_client = client.get_client()
+    assert ls_client is not None
+
 
 async def test_get_async_llama_stack_wrong_configuration() -> None:
     """Test if configuration is checked before Llama Stack is initialized."""