Function TestSMTPAPI_Wait

modules/smtp/api_test.go:509–574

TestSMTPAPI_Wait verifies the long-poll wait endpoint.

(t *testing.T)

// TestSMTPAPI_Wait verifies the long-poll wait endpoint.
func TestSMTPAPI_Wait(t *testing.T) {
	mux, store, mod := setupAPITest(t)

	t.Run("timeout", func(t *testing.T) {
		r := httptest.NewRequest("GET", "/api/smtp/messages/wait?timeout=50ms&to=nobody@example.com", nil)
		w := httptest.NewRecorder()
		mux.ServeHTTP(w, r)

		if w.Code != http.StatusRequestTimeout {
			t.Errorf("expected 408, got %d", w.Code)
		}
	})

	t.Run("existing matching event", func(t *testing.T) {
		storeSMTPEvent(t, store, "wait-uuid1", ParsedEmail{
			Subject: "Existing",
			To:      []EmailAddress{{Email: "wait-user@example.com"}},
		}, tsAgo(time.Second), "default")

		r := httptest.NewRequest("GET", "/api/smtp/messages/wait?to=wait-user@example.com&timeout=1s", nil)
		w := httptest.NewRecorder()
		mux.ServeHTTP(w, r)

		if w.Code != 200 {
			t.Errorf("expected 200, got %d", w.Code)
		}
	})

	t.Run("new event arrives", func(t *testing.T) {
		cursor := tsNow()

		resultCh := make(chan *httptest.ResponseRecorder, 1)
		go func() {
			url := fmt.Sprintf("/api/smtp/messages/wait?to=newuser@example.com&timeout=5s&since=%.6f", cursor)
			r := httptest.NewRequest("GET", url, nil)
			w := httptest.NewRecorder()
			mux.ServeHTTP(w, r)
			resultCh <- w
		}()

		// Give the goroutine time to subscribe before the event arrives.
		time.Sleep(30 * time.Millisecond)

		newEv := event.Event{
			UUID: "wait-uuid2",
			Type: "smtp",
			Payload: mustMarshalJSON(ParsedEmail{
				Subject: "New Arrival",
				To:      []EmailAddress{{Email: "newuser@example.com"}},
			}),
			Timestamp: tsNow(),
			Project:   "default",
		}
		store.Store(context.Background(), newEv)
		mod.OnEventStored(newEv)

		select {
		case w := <-resultCh: