diff --git a/src/main/java/org/studiorailgun/conversation/AgentLoop.java b/src/main/java/org/studiorailgun/conversation/AgentLoop.java index 3883620..1ce54e4 100644 --- a/src/main/java/org/studiorailgun/conversation/AgentLoop.java +++ b/src/main/java/org/studiorailgun/conversation/AgentLoop.java @@ -4,7 +4,6 @@ import java.util.Scanner; import org.studiorailgun.AI; import org.studiorailgun.Globals; -import org.studiorailgun.conversation.evaluators.EvaluationTree; import org.studiorailgun.conversation.parser.CommandParser; public class AgentLoop { diff --git a/src/test/java/org/studiorailgun/GreetingTests.java b/src/test/java/org/studiorailgun/GreetingTests.java index 0024eb2..9a38186 100644 --- a/src/test/java/org/studiorailgun/GreetingTests.java +++ b/src/test/java/org/studiorailgun/GreetingTests.java @@ -3,6 +3,7 @@ package org.studiorailgun; import static org.junit.jupiter.api.Assertions.*; import org.junit.jupiter.api.Test; +import org.studiorailgun.conversation.evaluators.goal.GoalData.ConversationGoal; /** * Tests greeting @@ -18,4 +19,13 @@ public class GreetingTests { assertEquals(Globals.conversation.getGreetingData().getHaveGreeted().size(), 1); } + @Test + public void testGoal(){ + Globals.init("./data/test/webs/web.json"); + + AI.simFrame("Hello"); + + assertEquals(Globals.conversation.getGoalData().getGoal(), ConversationGoal.GREET); + } + }