Merge pull request #9 from UiPath/test/add_init_tests

tests: add cli init config tests

Author: cristipufu

samples/quickstart-agent/uv.lock

@@ -0,0 +1,4 @@
+"""
+Test suite for UiPath LlamaIndex SDK.
+Contains test cases for all CLI commands and utilities.
+"""

tests/__init__.py

@@ -0,0 +1,227 @@
+import json
+import os
+
+from click.testing import CliRunner
+from uipath._cli.cli_init import init  # type: ignore
+
+
+class TestInit:
+    def test_init_basic_config_generation(
+        self, runner: CliRunner, temp_dir: str
+    ) -> None:
+        """Test configuration file generation with StartEvent and StopEvent."""
+        with runner.isolated_filesystem(temp_dir=temp_dir):
+            # Create agent script
+            script_content = """
+from llama_index.core.workflow import (
+    Event,
+    StartEvent,
+    StopEvent,
+    Workflow,
+    step,
+)
+from llama_index.llms.openai import OpenAI
+
+class JokeEvent(Event):
+    joke: str
+
+class JokeFlow(Workflow):
+    llm = OpenAI()
+
+    @step
+    async def generate_joke(self, ev: StartEvent) -> JokeEvent:
+        topic = ev.topic
+
+        prompt = f"Write your best joke about {topic}."
+        response = await self.llm.acomplete(prompt)
+        return JokeEvent(joke=str(response))
+
+    @step
+    async def critique_joke(self, ev: JokeEvent) -> StopEvent:
+        joke = ev.joke
+
+        prompt = f"Give a thorough analysis and critique of the following joke: {joke}"
+        response = await self.llm.acomplete(prompt)
+        return StopEvent(result=str(response))
+
+agent = JokeFlow(timeout=60, verbose=False)
+"""
+            with open("main.py", "w") as f:
+                f.write(script_content)
+
+            llama_config = """
+{
+  "dependencies": ["."],
+  "workflows": {
+    "agent": "main.py:agent"
+  },
+  "env": ".env"
+}
+"""
+            with open("llama_index.json", "w") as f:
+                f.write(llama_config)
+
+            result = runner.invoke(init)
+            assert result.exit_code == 0
+            assert os.path.exists("uipath.json")
+
+            with open("uipath.json", "r") as f:
+                config = json.load(f)
+
+                # Verify config structure
+                assert "entryPoints" in config
+                assert "bindings" in config
+
+                # Verify entryPoints properties
+                entry = config["entryPoints"][0]
+                assert entry["filePath"] == "agent"
+                assert entry["type"] == "agent"
+
+                # Verify input schema
+                assert "input" in entry
+                input_schema = entry["input"]
+                assert input_schema["type"] == "object"
+                assert "properties" in input_schema
+                assert "required" in input_schema
+                assert isinstance(input_schema["properties"], dict)
+                assert isinstance(input_schema["required"], list)
+
+                # Verify output schema
+                assert "output" in entry
+                output_schema = entry["output"]
+                assert output_schema["type"] == "object"
+                assert "properties" in output_schema
+                assert "required" in output_schema
+
+                # Verify bindings
+                assert config["bindings"]["version"] == "2.0"
+                assert "resources" in config["bindings"]
+                assert isinstance(config["bindings"]["resources"], list)
+
+    def test_init_custom_config_generation(
+        self, runner: CliRunner, temp_dir: str
+    ) -> None:
+        """Test configuration file generation with custom StartEvent and StopEvent."""
+        with runner.isolated_filesystem(temp_dir=temp_dir):
+            # Create agent script
+            script_content = """
+from typing import Optional
+
+from llama_index.core.workflow import (
+    Event,
+    StartEvent,
+    StopEvent,
+    Workflow,
+    step,
+)
+from llama_index.llms.openai import OpenAI
+
+class TopicEvent(StartEvent):
+    topic: str
+    param: Optional[str] = None
+
+class JokeEvent(Event):
+    joke: str
+
+class CritiqueEvent(StopEvent):
+    joke: str
+    critique: str
+    param: Optional[str] = None
+
+class JokeFlow(Workflow):
+    llm = OpenAI()
+
+    @step
+    async def generate_joke(self, ev: TopicEvent) -> JokeEvent:
+        topic = ev.topic
+
+        prompt = f"Write your best joke about {topic}."
+        response = await self.llm.acomplete(prompt)
+        return JokeEvent(joke=str(response))
+
+    @step
+    async def critique_joke(self, ev: JokeEvent) -> CritiqueEvent:
+        joke = ev.joke
+
+        prompt = f"Give a thorough analysis and critique of the following joke: {joke}"
+        response = await self.llm.acomplete(prompt)
+        return CritiqueEvent(joke=joke, critique=str(response))
+
+agent = JokeFlow(timeout=60, verbose=False)
+"""
+            with open("main.py", "w") as f:
+                f.write(script_content)
+
+            llama_config = """
+{
+  "dependencies": ["."],
+  "workflows": {
+    "agent": "main.py:agent"
+  },
+  "env": ".env"
+}
+"""
+            with open("llama_index.json", "w") as f:
+                f.write(llama_config)
+
+            result = runner.invoke(init)
+            assert result.exit_code == 0
+            assert os.path.exists("uipath.json")
+
+            with open("uipath.json", "r") as f:
+                config = json.load(f)
+
+                # Verify config structure
+                assert "entryPoints" in config
+                assert "bindings" in config
+
+                # Verify entryPoints properties
+                entry = config["entryPoints"][0]
+                assert entry["filePath"] == "agent"
+                assert entry["type"] == "agent"
+
+                # Verify input schema
+                assert "input" in entry
+                input_schema = entry["input"]
+                assert input_schema["type"] == "object"
+                assert "properties" in input_schema
+                assert "required" in input_schema
+
+                # Verify input properties
+                props = input_schema["properties"]
+                assert "topic" in props
+                assert props["topic"]["type"] == "string"
+
+                assert "param" in props
+                assert props["param"]["type"] == "string"
+                assert props["param"]["nullable"]
+
+                # Verify required fields in input
+                assert input_schema["required"] == ["topic"]
+
+                # Verify output schema
+                assert "output" in entry
+                output_schema = entry["output"]
+                assert output_schema["type"] == "object"
+                assert "properties" in output_schema
+                assert "required" in output_schema
+
+                # Verify output properties
+                out_props = output_schema["properties"]
+                assert "joke" in out_props
+                assert out_props["joke"]["type"] == "string"
+
+                assert "critique" in out_props
+                assert out_props["critique"]["type"] == "string"
+
+                assert "param" in out_props
+                assert out_props["param"]["type"] == "string"
+                assert out_props["param"]["nullable"]
+
+                # Verify required fields in output
+                assert output_schema["required"] == ["joke", "critique"]
+
+                # Verify bindings
+                assert config["bindings"]["version"] == "2.0"
+                assert "resources" in config["bindings"]
+                assert isinstance(config["bindings"]["resources"], list)

tests/cli/test_init.py

@@ -0,0 +1,18 @@
+import tempfile
+from typing import Generator
+
+import pytest
+from click.testing import CliRunner
+
+
+@pytest.fixture
+def runner() -> CliRunner:
+    """Provide a Click CLI test runner."""
+    return CliRunner()
+
+
+@pytest.fixture
+def temp_dir() -> Generator[str, None, None]:
+    """Provide a temporary directory for test files."""
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        yield tmp_dir