micro--go-micro/ai/tools_test.go

package ai

import (
	"context"
	"testing"

	"go-micro.dev/v6/registry"
)

func TestToolJSONType(t *testing.T) {
	cases := map[string]string{
		"string":  "string",
		"int":     "integer",
		"int64":   "integer",
		"float64": "number",
		"bool":    "boolean",
		"User":    "object",
		"":        "object",
	}
	for in, want := range cases {
		if got := toolJSONType(in); got != want {
			t.Errorf("toolJSONType(%q) = %q, want %q", in, got, want)
		}
	}
}

func TestDiscoverTools_Empty(t *testing.T) {
	reg := registry.NewMemoryRegistry()
	tools, err := DiscoverTools(reg)
	if err != nil {
		t.Fatalf("DiscoverTools: %v", err)
	}
	if len(tools) != 0 {
		t.Errorf("expected 0 tools, got %d", len(tools))
	}
}

func TestDiscoverTools_DiscoversEndpoints(t *testing.T) {
	reg := registry.NewMemoryRegistry()
	svc := &registry.Service{
		Name:    "users",
		Version: "1.0.0",
		Nodes: []*registry.Node{
			{Id: "users-1", Address: "127.0.0.1:9000"},
		},
		Endpoints: []*registry.Endpoint{
			{
				Name: "Users.Get",
				Metadata: map[string]string{
					"description": "Fetch a user by ID",
				},
				Request: &registry.Value{
					Name: "GetRequest",
					Type: "GetRequest",
					Values: []*registry.Value{
						{Name: "id", Type: "string"},
						{Name: "expand", Type: "bool"},
					},
				},
			},
		},
	}
	if err := reg.Register(svc); err != nil {
		t.Fatalf("Register: %v", err)
	}

	tools, err := DiscoverTools(reg)
	if err != nil {
		t.Fatalf("DiscoverTools: %v", err)
	}
	if len(tools) != 1 {
		t.Fatalf("expected 1 tool, got %d", len(tools))
	}

	tool := tools[0]
	if tool.Name != "users_Users_Get" {
		t.Errorf("safe name = %q", tool.Name)
	}
	if tool.OriginalName != "users.Users.Get" {
		t.Errorf("original = %q", tool.OriginalName)
	}
	if tool.Description != "Fetch a user by ID" {
		t.Errorf("description = %q", tool.Description)
	}
}

func TestTools_HandlerResolvesSafeName(t *testing.T) {
	tools := NewTools(registry.NewMemoryRegistry())
	tools.names.put("users_Users_Get", "users.Users.Get")

	resolved, ok := tools.names.get("users_Users_Get")
	if !ok || resolved != "users.Users.Get" {
		t.Errorf("name map lookup = (%q, %v)", resolved, ok)
	}
}

func TestTools_HandlerInvalidName(t *testing.T) {
	tools := NewTools(registry.NewMemoryRegistry())
	h := tools.Handler()

	res := h(context.Background(), ToolCall{Name: "foo", Input: map[string]any{}})
	if res.Value == nil {
		t.Fatal("expected error result")
	}
	if res.Content == "" {
		t.Error("expected non-empty content")
	}
}

func TestWithTools(t *testing.T) {
	tools := NewTools(registry.NewMemoryRegistry())
	opts := NewOptions(WithTools(tools))
	if opts.ToolHandler == nil {
		t.Error("WithTools did not set a ToolHandler")
	}
}