Function TestHandleMessageSuccess

lib/event/router_test.go:95–151

(t *testing.T)

}

func TestHandleMessageSuccess(t *testing.T) {
	r := NewRouter()

	var handledV1, handledV2 bool
	eventID1 := "event1"
	eventID2 := "event2"

	testHandler1 := func(_ context.Context, eventID string, e TestEventV1) error {
		handledV1 = true
		if e.ID != "v1-id" {
			t.Errorf("v1 handler received wrong ID: %s", e.ID)
		}

		if eventID != "event1" {
			t.Errorf("v1 handler received wrong eventID: %s", eventID)
		}

		return nil
	}
	testHandler2 := func(_ context.Context, eventID string, e TestEventV2) error {
		handledV2 = true
		if e.ID != "v2-id" || e.Priority != 5 {
			t.Errorf("v2 handler received wrong data: %+v", e)
		}

		if eventID != "event2" {
			t.Errorf("v2 handler received wrong eventID: %s", eventID)
		}

		return nil
	}

	// Register handlers
	Register(r, testHandler1)
	Register(r, testHandler2)

	// Create and route V1 event
	v1Payload, err := New(TestEventV1{ID: "v1-id"})
	if err != nil {
		t.Fatalf("failed to create v1 event: %v", err)
	}
	if err := r.HandleMessage(t.Context(), eventID1, v1Payload); err != nil {
		t.Errorf("failed to handle v1: %v", err)
	}
	if !handledV1 {
		t.Error("v1 handler was not called")
	}

	// Create and route V2 event
	v2Payload, _ := New(TestEventV2{ID: "v2-id", Priority: 5})
	if err := r.HandleMessage(t.Context(), eventID2, v2Payload); err != nil {
		t.Errorf("failed to handle v2: %v", err)
	}
	if !handledV2 {
		t.Error("v2 handler was not called")
	}
}