Function TestHandler_DrainUnreadOnConnect

internal/ws/handler_test.go:282–371

(t *testing.T)

}

func TestHandler_DrainUnreadOnConnect(t *testing.T) {
	hub := NewHub()
	defer hub.Close()

	now := time.Now()
	store := &mockStore{
		user:  newTestUser(),
		agent: newTestAgent("user_1"),
		messages: []identity.Message{
			{
				ID:             "msg_1",
				AgentID:        "agent_test",
				Sender:         "alice@example.com",
				Recipient:      "bot@agents.e2a.dev",
				Subject:        "Hello",
				ConversationID: "conv_1",
				CreatedAt:      now,
			},
			{
				ID:        "msg_2",
				AgentID:   "agent_test",
				Sender:    "bob@example.com",
				Recipient: "bot@agents.e2a.dev",
				Subject:   "Hi there",
				CreatedAt: now,
			},
		},
	}
	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")
	}
	defer conn.Close(websocket.StatusNormalClosure, "")

	// Read two drained notifications
	for i := 0; i < 2; i++ {
		ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
		_, data, err := conn.Read(ctx)
		cancel()
		if err != nil {
			t.Fatalf("read notification %d: %v", i+1, err)
		}

		var notif struct {
			MessageID      string `json:"message_id"`
			ConversationID string `json:"conversation_id,omitempty"`
			From           string `json:"from"`
			Recipient      string `json:"recipient"`
			Subject        string `json:"subject"`
		}
		if err := json.Unmarshal(data, &notif); err != nil {
			t.Fatalf("unmarshal notification %d: %v", i+1, err)
		}

		if i == 0 {