52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
|
package loop
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestLoadMCPTools_Good_ConvertsRecords(t *testing.T) {
|
||
|
|
handler := func(ctx context.Context, args map[string]any) (string, error) {
|
||
|
|
return "result", nil
|
||
|
|
}
|
||
|
|
|
||
|
|
tool := Tool{
|
||
|
|
Name: "file_read",
|
||
|
|
Description: "Read a file",
|
||
|
|
Parameters: map[string]any{"type": "object"},
|
||
|
|
Handler: handler,
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Equal(t, "file_read", tool.Name)
|
||
|
|
result, err := tool.Handler(context.Background(), map[string]any{"path": "/tmp/test"})
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, "result", result)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWrapRESTHandler_Good(t *testing.T) {
|
||
|
|
restHandler := func(ctx context.Context, body []byte) (any, error) {
|
||
|
|
var input map[string]any
|
||
|
|
json.Unmarshal(body, &input)
|
||
|
|
return map[string]string{"content": "hello from " + input["path"].(string)}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
wrapped := WrapRESTHandler(restHandler)
|
||
|
|
result, err := wrapped(context.Background(), map[string]any{"path": "/tmp/test"})
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Contains(t, result, "hello from /tmp/test")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWrapRESTHandler_Bad_HandlerError(t *testing.T) {
|
||
|
|
restHandler := func(ctx context.Context, body []byte) (any, error) {
|
||
|
|
return nil, assert.AnError
|
||
|
|
}
|
||
|
|
|
||
|
|
wrapped := WrapRESTHandler(restHandler)
|
||
|
|
_, err := wrapped(context.Background(), map[string]any{})
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|