WeKnora/internal/agent/tools/wiki_tools_test.go

package tools

import (
	"testing"
)

func TestTruncateForSummary(t *testing.T) {
	tests := []struct {
		name    string
		content string
		maxLen  int
		want    string
	}{
		{
			name:    "short content",
			content: "Hello world",
			maxLen:  50,
			want:    "Hello world",
		},
		{
			name:    "strips heading prefix",
			content: "# My Title\n\nSome content",
			maxLen:  50,
			want:    "My Title",
		},
		{
			name:    "strips h2 prefix",
			content: "## Section\n\nContent here",
			maxLen:  50,
			want:    "Section",
		},
		{
			name:    "truncates long content",
			content: "This is a very long paragraph that should be truncated at some point because it exceeds the maximum length",
			maxLen:  20,
			want:    "This is a very long ..."},
		{
			name:    "takes first paragraph",
			content: "First paragraph.\n\nSecond paragraph.",
			maxLen:  100,
			want:    "First paragraph.",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := truncateForSummary(tt.content, tt.maxLen)
			if got != tt.want {
				t.Errorf("truncateForSummary() = %q, want %q", got, tt.want)
			}
		})
	}
}

func TestWikiToolConstants(t *testing.T) {
	// Verify all wiki tool constants are defined and unique
	names := []string{
		ToolWikiReadPage,
		ToolWikiSearch,
	}

	seen := make(map[string]bool)
	for _, name := range names {
		if name == "" {
			t.Error("Wiki tool constant is empty")
		}
		if seen[name] {
			t.Errorf("Duplicate wiki tool constant: %s", name)
		}
		seen[name] = true

		// Verify name length is within OpenAI limit
		if len(name) > maxFunctionNameLength {
			t.Errorf("Wiki tool name too long: %s (%d chars, max %d)", name, len(name), maxFunctionNameLength)
		}
	}
}

func TestWikiToolsInAvailableDefinitions(t *testing.T) {
	defs := AvailableToolDefinitions()
	wikiTools := map[string]bool{
		ToolWikiReadPage: false,
		ToolWikiSearch:   false,
	}

	for _, def := range defs {
		if _, ok := wikiTools[def.Name]; ok {
			wikiTools[def.Name] = true
		}
	}

	for name, found := range wikiTools {
		if !found {
			t.Errorf("Wiki tool %s missing from AvailableToolDefinitions()", name)
		}
	}
}