Function TestHandler_DisconnectUnregisters

internal/ws/handler_test.go:373–403

(t *testing.T)

}

func TestHandler_DisconnectUnregisters(t *testing.T) {
	hub := NewHub()
	defer hub.Close()
	store := &mockStore{
		user:  newTestUser(),
		agent: newTestAgent("user_1"),
	}
	handler := NewHandler(hub, store)
	srv := startServer(t, handler)

	conn, _ := dialWS(t, srv, "bot@agents.e2a.dev", "valid_key")
	if conn == nil {
		t.Fatal("expected successful WS connection")
	}

	waitConnected(t, hub, "agent_test")

	// Close the connection
	conn.Close(websocket.StatusNormalClosure, "bye")

	// Wait for unregister to propagate
	deadline := time.After(2 * time.Second)
	for hub.IsConnected("agent_test") {
		select {
		case <-deadline:
			t.Fatal("timed out waiting for unregister")
		default:
			time.Sleep(10 * time.Millisecond)
		}
	}
}

func TestHandler_SendAfterConnect(t *testing.T) {
	hub := NewHub()