Kyle Isom
08b3e2a472
All import paths updated to git.wntrmute.dev/mc/. Bumps mcdsl to v1.2.0, mc-proxy to v1.1.0. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package agent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.wntrmute.dev/mc/mcp/internal/registry"
|
|
)
|
|
|
|
func TestListenerForMode(t *testing.T) {
|
|
tests := []struct {
|
|
mode string
|
|
port int
|
|
want string
|
|
}{
|
|
{"l4", 8443, ":8443"},
|
|
{"l7", 443, ":443"},
|
|
{"l4", 9443, ":9443"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
got := listenerForMode(tt.mode, tt.port)
|
|
if got != tt.want {
|
|
t.Errorf("listenerForMode(%q, %d) = %q, want %q", tt.mode, tt.port, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNilProxyRouterIsNoop(t *testing.T) {
|
|
var p *ProxyRouter
|
|
|
|
// All methods should return nil on a nil ProxyRouter.
|
|
if err := p.RegisterRoutes(nil, "svc", nil, nil); err != nil {
|
|
t.Errorf("RegisterRoutes on nil: %v", err)
|
|
}
|
|
if err := p.RemoveRoutes(nil, "svc", nil); err != nil {
|
|
t.Errorf("RemoveRoutes on nil: %v", err)
|
|
}
|
|
if err := p.Close(); err != nil {
|
|
t.Errorf("Close on nil: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestRegisterRoutesSkipsZeroHostPort(t *testing.T) {
|
|
// A nil ProxyRouter should be a no-op, so this tests the skip logic
|
|
// indirectly. With a nil proxy, RegisterRoutes returns nil even
|
|
// with routes that have zero host ports.
|
|
var p *ProxyRouter
|
|
|
|
routes := []registry.Route{
|
|
{Name: "rest", Port: 8443, Mode: "l4"},
|
|
}
|
|
hostPorts := map[string]int{"rest": 0}
|
|
|
|
if err := p.RegisterRoutes(nil, "svc", routes, hostPorts); err != nil {
|
|
t.Errorf("RegisterRoutes: %v", err)
|
|
}
|
|
}
|