Carlos Florêncio
79 lines
1.8 KiB
Go
79 lines
1.8 KiB
Go
package websocket
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/kandev/kandev/internal/common/logger"
|
|
)
|
|
|
|
func newTestProxyLogger() *logger.Logger {
|
|
log, _ := logger.NewLogger(logger.LoggingConfig{
|
|
Level: "error",
|
|
Format: "json",
|
|
})
|
|
return log
|
|
}
|
|
|
|
func TestNewVscodeProxyHandler(t *testing.T) {
|
|
log := newTestProxyLogger()
|
|
h := NewVscodeProxyHandler(nil, log)
|
|
|
|
if h == nil {
|
|
t.Fatal("expected non-nil handler")
|
|
}
|
|
if h.proxies == nil {
|
|
t.Error("expected proxies map to be initialized")
|
|
}
|
|
if h.lifecycleMgr != nil {
|
|
t.Error("expected nil lifecycleMgr when nil passed")
|
|
}
|
|
}
|
|
|
|
func TestInvalidateProxy(t *testing.T) {
|
|
log := newTestProxyLogger()
|
|
h := NewVscodeProxyHandler(nil, log)
|
|
|
|
// Manually insert a cache entry
|
|
h.mu.Lock()
|
|
h.proxies["session-1"] = &proxyEntry{proxy: nil, target: "127.0.0.1:8080"}
|
|
h.proxies["session-2"] = &proxyEntry{proxy: nil, target: "127.0.0.1:8081"}
|
|
h.mu.Unlock()
|
|
|
|
// Invalidate one
|
|
h.InvalidateProxy("session-1")
|
|
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
if _, ok := h.proxies["session-1"]; ok {
|
|
t.Error("expected session-1 proxy to be removed")
|
|
}
|
|
if _, ok := h.proxies["session-2"]; !ok {
|
|
t.Error("expected session-2 proxy to still exist")
|
|
}
|
|
}
|
|
|
|
func TestInvalidateProxy_Idempotent(t *testing.T) {
|
|
log := newTestProxyLogger()
|
|
h := NewVscodeProxyHandler(nil, log)
|
|
|
|
// Invalidate a non-existent entry — should not panic
|
|
h.InvalidateProxy("nonexistent")
|
|
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
if len(h.proxies) != 0 {
|
|
t.Errorf("expected empty proxies map, got %d entries", len(h.proxies))
|
|
}
|
|
}
|
|
|
|
func TestProxyCacheEmpty_OnCreation(t *testing.T) {
|
|
log := newTestProxyLogger()
|
|
h := NewVscodeProxyHandler(nil, log)
|
|
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
if len(h.proxies) != 0 {
|
|
t.Errorf("expected empty proxies on creation, got %d", len(h.proxies))
|
|
}
|
|
}
|