ai-agent/internal/ai/runtime/tools/helper_test.go

package tools

import (
	"testing"

	"code.tczkiot.com/wlw/ai-agent/internal/pkg/toolx"
)

func TestNewRuntimeStaticTool(t *testing.T) {
	items := []string{
		toolx.GraphTriageServiceRequest.Code,
		toolx.GraphAnalyzeConversation.Code,
		toolx.GraphHandoffConversation.Code,
	}
	for _, item := range items {
		tool := NewRuntimeStaticTool(item)
		if tool == nil {
			t.Fatalf("expected runtime static tool for %s", item)
		}
		if tool.Spec().Code != item {
			t.Fatalf("unexpected tool code for %s: %s", item, tool.Spec().Code)
		}
	}
}

func TestNewRuntimeStaticToolReturnsNilForUnknownTool(t *testing.T) {
	if tool := NewRuntimeStaticTool("builtin/unknown_tool"); tool != nil {
		t.Fatalf("expected nil tool for unknown tool code")
	}
}

func TestParseConfirmationDecisionSupportsBusinessActionWords(t *testing.T) {
	for _, input := range []string{"确认", "提交", "确定办理", "执行"} {
		if got := ParseConfirmationDecision(input); got != DecisionConfirm {
			t.Fatalf("ParseConfirmationDecision(%q) = %q", input, got)
		}
	}
	for _, input := range []string{"不确认", "好的，取消", "不提交", "不要办理", "暂不执行"} {
		if got := ParseConfirmationDecision(input); got != DecisionCancel {
			t.Fatalf("ParseConfirmationDecision(%q) = %q", input, got)
		}
	}
}