ToolExecutionResultMessage tests. (#488)

2024-01-15 00:21:39 -08:00 · 2024-01-15 00:21:39 -08:00 · fd4a50a6f1
parent 1e97f84f41
commit fd4a50a6f1
1 changed files with 67 additions and 0 deletions
--- a/langchain4j-core/src/test/java/dev/langchain4j/data/message/ToolExecutionResultMessageTest.java
+++ b/langchain4j-core/src/test/java/dev/langchain4j/data/message/ToolExecutionResultMessageTest.java
@ -0,0 +1,67 @@
+package dev.langchain4j.data.message;
+
+import dev.langchain4j.agent.tool.ToolExecutionRequest;
+import org.assertj.core.api.WithAssertions;
+import org.junit.jupiter.api.Test;
+
+class ToolExecutionResultMessageTest implements WithAssertions {
+    @Test
+    public void test_methods() {
+        ToolExecutionResultMessage tm = new ToolExecutionResultMessage("id", "toolName", "text");
+        assertThat(tm.id()).isEqualTo("id");
+        assertThat(tm.toolName()).isEqualTo("toolName");
+        assertThat(tm.text()).isEqualTo("text");
+        assertThat(tm.type()).isEqualTo(ChatMessageType.TOOL_EXECUTION_RESULT);
+
+        assertThat(tm)
+                .hasToString(
+                "ToolExecutionResultMessage " +
+                        "{ id = \"id\" toolName = \"toolName\" text = \"text\" }");
+    }
+
+    @Test
+    public void test_equals_hashCode() {
+        ToolExecutionResultMessage t1 = new ToolExecutionResultMessage("id", "toolName", "text");
+        ToolExecutionResultMessage t2 = new ToolExecutionResultMessage("id", "toolName", "text");
+
+        ToolExecutionResultMessage t3 = new ToolExecutionResultMessage("foo", "toolName", "text");
+        ToolExecutionResultMessage t4 = new ToolExecutionResultMessage("foo", "toolName", "text");
+
+        assertThat(t1)
+                .isEqualTo(t1)
+                .isNotEqualTo(null)
+                .isNotEqualTo(new Object())
+                .isEqualTo(t2)
+                .hasSameHashCodeAs(t2)
+                .isNotEqualTo(ToolExecutionResultMessage.from(
+                        "changed", "toolName", "text"))
+                .isNotEqualTo(ToolExecutionResultMessage.from(
+                        "id", "changed", "text"))
+                .isNotEqualTo(ToolExecutionResultMessage.from(
+                        "id", "toolName", "changed"))
+                .isNotEqualTo(t3)
+                .doesNotHaveSameHashCodeAs(t3);
+
+        assertThat(t3)
+                .isEqualTo(t3)
+                .isEqualTo(t4)
+                .hasSameHashCodeAs(t4);
+    }
+
+    @Test
+    public void test_builders() {
+        ToolExecutionRequest request = ToolExecutionRequest.builder()
+                .id("id")
+                .name("toolName")
+                .arguments("arguments")
+                .build();
+
+        assertThat(new ToolExecutionResultMessage("id", "toolName", "text"))
+                .isEqualTo(ToolExecutionResultMessage.from("id", "toolName", "text"))
+                .isEqualTo(ToolExecutionResultMessage.from(request, "text"))
+                .isEqualTo(ToolExecutionResultMessage.toolExecutionResultMessage(
+                        "id", "toolName", "text"))
+                .isEqualTo(ToolExecutionResultMessage.toolExecutionResultMessage(request, "text"));
+
+    }
+}