Integration test

Author: johnsonr

pom.xml

@@ -36,6 +36,13 @@
             <version>${embabel-agent.version}</version>
         </dependency>
 
+        <dependency>
+            <groupId>com.embabel.agent</groupId>
+            <artifactId>embabel-agent-test</artifactId>
+            <version>${embabel-agent.version}</version>
+            <scope>test</scope>
+        </dependency>
+
 
         <!-- Unit and Integration Testing -->
         <dependency>

src/test/java/com/embabel/template/agent/WriteAndReviewAgentIntegrationTest.java

@@ -0,0 +1,45 @@
+package com.embabel.template.agent;
+
+import com.embabel.agent.api.common.autonomy.AgentInvocation;
+import com.embabel.agent.domain.io.UserInput;
+import com.embabel.agent.testing.integration.EmbabelMockitoIntegrationTest;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+/**
+ * Use framework superclass to test the complete workflow of writing and reviewing a story.
+ * This will run under Spring Boot against an AgentPlatform instance
+ * that has loaded all our agents.
+ */
+class WriteAndReviewAgentIntegrationTest extends EmbabelMockitoIntegrationTest {
+
+    @Test
+    void shouldExecuteCompleteWorkflow() {
+        var input = new UserInput("Write about artificial intelligence");
+
+        var story = new Story("AI will transform our world...");
+        var reviewedStory = new ReviewedStory(story, "Excellent exploration of AI themes.", Personas.REVIEWER);
+
+        whenCreateObject(prompt -> prompt.contains("Craft a short story"), Story.class)
+                .thenReturn(story);
+
+        // The second call uses generateText
+        whenGenerateText(prompt -> prompt.contains("You will be given a short story to review"))
+                .thenReturn(reviewedStory.review());
+
+        var invocation = AgentInvocation.create(agentPlatform, ReviewedStory.class);
+        var reviewedStoryResult = invocation.invoke(input);
+
+        assertNotNull(reviewedStoryResult);
+        assertTrue(reviewedStoryResult.getContent().contains(story.text()),
+                "Expected story content to be present: " + reviewedStoryResult.getContent());
+        assertEquals(reviewedStory, reviewedStoryResult,
+                "Expected review to match: " + reviewedStoryResult);
+
+        verifyCreateObjectMatching(prompt -> prompt.contains("Craft a short story"), Story.class,
+                llm -> llm.getLlm().getTemperature() == 0.7 && llm.getToolGroups().isEmpty());
+        verifyGenerateTextMatching(prompt -> prompt.contains("You will be given a short story to review"));
+        verifyNoMoreInteractions();
+    }
+}