Carlos Florêncio
77 lines
2.7 KiB
Go
77 lines
2.7 KiB
Go
package instance
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/kandev/kandev/internal/agentctl/server/config"
|
|
"github.com/kandev/kandev/internal/agentctl/server/process"
|
|
"github.com/kandev/kandev/internal/common/logger"
|
|
"github.com/kandev/kandev/pkg/agent"
|
|
)
|
|
|
|
// TestCreateInstance_RecordsCreateReadyMillis pins the diagnostic
|
|
// agentctl_create_ready_ms metric's write side: CreateInstance must stamp a
|
|
// positive elapsed-milliseconds value onto the InstanceConfig it builds, so
|
|
// api.handleSystemMetrics has something other than the "not yet recorded"
|
|
// zero sentinel to report for a real instance.
|
|
func TestCreateInstance_RecordsCreateReadyMillis(t *testing.T) {
|
|
log := newTestLogger(t)
|
|
mgr := NewManager(&config.Config{
|
|
Ports: config.PortConfig{Base: 0, Max: 0},
|
|
Defaults: config.InstanceDefaults{Protocol: agent.ProtocolACP},
|
|
}, log)
|
|
t.Cleanup(func() { _ = mgr.Shutdown(context.Background()) })
|
|
|
|
var captured *config.InstanceConfig
|
|
mgr.SetServerFactory(func(cfg *config.InstanceConfig, procMgr *process.Manager, log *logger.Logger) http.Handler {
|
|
captured = cfg
|
|
return http.NotFoundHandler()
|
|
})
|
|
|
|
resp, err := mgr.CreateInstance(context.Background(), &CreateRequest{WorkspacePath: t.TempDir()})
|
|
if err != nil {
|
|
t.Fatalf("CreateInstance: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = mgr.StopInstance(context.Background(), resp.ID) })
|
|
|
|
if captured == nil {
|
|
t.Fatal("server factory was never invoked; no InstanceConfig captured")
|
|
}
|
|
if captured.CreateReadyMillis == nil {
|
|
t.Fatal("CreateReadyMillis is nil on a config built by NewInstanceConfig")
|
|
}
|
|
if millis := captured.CreateReadyMillis.Load(); millis <= 0 {
|
|
t.Errorf("CreateReadyMillis = %d, want > 0 after a completed creation", millis)
|
|
}
|
|
}
|
|
|
|
func TestCreateInstance_PropagatesMCPToolNamePresentationCapability(t *testing.T) {
|
|
log := newTestLogger(t)
|
|
mgr := NewManager(&config.Config{
|
|
Ports: config.PortConfig{Base: 0, Max: 0},
|
|
Defaults: config.InstanceDefaults{Protocol: agent.ProtocolACP},
|
|
}, log)
|
|
t.Cleanup(func() { _ = mgr.Shutdown(context.Background()) })
|
|
|
|
var captured *config.InstanceConfig
|
|
mgr.SetServerFactory(func(cfg *config.InstanceConfig, procMgr *process.Manager, log *logger.Logger) http.Handler {
|
|
captured = cfg
|
|
return http.NotFoundHandler()
|
|
})
|
|
|
|
resp, err := mgr.CreateInstance(context.Background(), &CreateRequest{
|
|
WorkspacePath: t.TempDir(),
|
|
NamespacesMCPToolsByServer: true,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("CreateInstance: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = mgr.StopInstance(context.Background(), resp.ID) })
|
|
|
|
if captured == nil || !captured.NamespacesMCPToolsByServer {
|
|
t.Fatalf("captured instance config = %+v, want namespaced MCP tools", captured)
|
|
}
|
|
}
|