90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package mujica
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
)
|
|
|
|
func toolSample() []AgentTool {
|
|
return []AgentTool{
|
|
{Name: "web_search", Description: "Search the web", Enabled: true, Params: []ToolParam{
|
|
{Name: "query", Type: "string", Required: true, Description: "What to look for"},
|
|
{Name: "limit", Type: "integer"},
|
|
}},
|
|
{Name: "read_file", Description: "Read a local file"},
|
|
}
|
|
}
|
|
|
|
func TestToolRegistryPanelToggleAndPreview(t *testing.T) {
|
|
var s ListState[string]
|
|
tools := toolSample()
|
|
disabled := false
|
|
var toggled []string
|
|
tt := ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
ui.Column(c).Height(400).Children(func() {
|
|
r := ToolRegistryPanel(c, &s, tools, ToolRegistryPanelOptions{Disabled: disabled})
|
|
r.Element.Grow(1)
|
|
if name, on, ok := r.Toggled(); ok {
|
|
toggled = append(toggled, name)
|
|
for i := range tools {
|
|
if tools[i].Name == name {
|
|
tools[i].Enabled = on
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}, 800, 420)
|
|
if !tt.HasText("Choose a tool to see its parameters") {
|
|
t.Fatalf("empty preview: %q", tt.Texts())
|
|
}
|
|
if err := tt.Click("Search the web"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !tt.HasText("query") || !tt.HasText("required") || !tt.HasText("integer") {
|
|
t.Fatalf("schema: %q", tt.Texts())
|
|
}
|
|
if err := tt.Click("read_file"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !tt.HasText("No parameters") {
|
|
t.Fatalf("no params: %q", tt.Texts())
|
|
}
|
|
if err := tt.Click("Enable read_file"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(toggled) != 1 || toggled[0] != "read_file" || !tools[1].Enabled {
|
|
t.Fatalf("toggle %q enabled %v", toggled, tools[1].Enabled)
|
|
}
|
|
disabled = true
|
|
tt.Frame()
|
|
_ = tt.Click("Enable read_file")
|
|
if len(toggled) != 1 {
|
|
t.Fatal("disabled panel toggled")
|
|
}
|
|
}
|
|
|
|
func TestToolRegistryPanelInvalid(t *testing.T) {
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("no panic")
|
|
}
|
|
}()
|
|
ui.Render(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
var s ListState[string]
|
|
ToolRegistryPanel(c, &s, []AgentTool{{Name: "a"}, {Name: "a"}}, ToolRegistryPanelOptions{})
|
|
}, 300, 200, 1)
|
|
}
|
|
|
|
func TestToolRegistryPanelSnapshot(t *testing.T) {
|
|
tools := toolSample()
|
|
snapshotAfter(t, "tool_registry_panel", 760, 300, func(c *ui.Context) {
|
|
s := ui.Local(c.Root(), "s", func() *ListState[string] { return &ListState[string]{} })
|
|
ui.Column(c).Padding(16).Height(300).Children(func() {
|
|
ToolRegistryPanel(c, *s, tools, ToolRegistryPanelOptions{}).Element.Grow(1)
|
|
})
|
|
}, func(tt *ui.Tester) { _ = tt.Click("Search the web") })
|
|
}
|