Merge pull request #724 from dgageot/tool-format-tests

Add tests for tool call formatting

Author: rumpl

pkg/cli/text_test.go

@@ -0,0 +1,75 @@
+package cli
+
+import (
+	"testing"
+
+	"gotest.tools/v3/assert"
+)
+
+func TestFormatToolCallResponse_Empty(t *testing.T) {
+	formatted := formatToolCallResponse(``)
+
+	assert.Equal(t, ` → ()`, formatted)
+}
+
+func TestFormatToolCallResponse_Map(t *testing.T) {
+	formatted := formatToolCallResponse(`{"text": "hello"}`)
+
+	assert.Equal(t, ` → (text: "hello")`, formatted)
+}
+
+func TestFormatToolCallResponse_MapOfEmptyArray(t *testing.T) {
+	formatted := formatToolCallResponse(`{"array": []}`)
+	assert.Equal(t, ` → (array: [])`, formatted)
+}
+
+func TestFormatToolCallResponse_MapOfArray(t *testing.T) {
+	formatted := formatToolCallResponse(`{"array": [1,2,3]}`)
+	assert.Equal(t, ` → (
+  array: [
+  1,
+  2,
+  3
+]
+)`, formatted)
+}
+
+func TestFormatToolCallResponse_PlainText(t *testing.T) {
+	formatted := formatToolCallResponse(`Plain Text`)
+
+	assert.Equal(t, ` → "Plain Text"`, formatted)
+}
+
+func TestFormatToolCallArguments_Empty(t *testing.T) {
+	formatted := formatToolCallArguments(``)
+
+	assert.Equal(t, `()`, formatted)
+}
+
+func TestFormatToolCallArguments_Map(t *testing.T) {
+	formatted := formatToolCallArguments(`{"text": "hello"}`)
+
+	assert.Equal(t, `(text: "hello")`, formatted)
+}
+
+func TestFormatToolCallArguments_MapOfArray(t *testing.T) {
+	formatted := formatToolCallArguments(`{"array": [1,2,3]}`)
+	assert.Equal(t, `(
+  array: [
+  1,
+  2,
+  3
+]
+)`, formatted)
+}
+
+func TestFormatToolCallArguments_MapOfEmptyArray(t *testing.T) {
+	formatted := formatToolCallArguments(`{"array": []}`)
+	assert.Equal(t, `(array: [])`, formatted)
+}
+
+func TestFormatToolCallArguments_PlainText(t *testing.T) {
+	formatted := formatToolCallArguments(`Plain Text`)
+
+	assert.Equal(t, `(Plain Text)`, formatted)
+}