Function TestSMTPAPI_Codes

modules/smtp/api_test.go:388–464

TestSMTPAPI_Codes verifies OTP code extraction.

(t *testing.T)

// TestSMTPAPI_Codes verifies OTP code extraction.
func TestSMTPAPI_Codes(t *testing.T) {
	mux, store, _ := setupAPITest(t)

	storeSMTPEvent(t, store, "code-uuid1", ParsedEmail{
		Subject: "Your OTP",
		Text:    "Your code is 123456. Do not share it.",
	}, 1000.0, "default")

	t.Run("default pattern", func(t *testing.T) {
		r := httptest.NewRequest("GET", "/api/smtp/message/code-uuid1/codes", nil)
		w := httptest.NewRecorder()
		mux.ServeHTTP(w, r)

		if w.Code != 200 {
			t.Fatalf("status = %d", w.Code)
		}
		var resp map[string]any
		json.NewDecoder(w.Body).Decode(&resp)
		data := resp["data"].([]any)
		found := false
		for _, c := range data {
			if c.(string) == "123456" {
				found = true
			}
		}
		if !found {
			t.Errorf("expected to find 123456 in %v", data)
		}
	})

	t.Run("custom pattern", func(t *testing.T) {
		r := httptest.NewRequest("GET", "/api/smtp/message/code-uuid1/codes?pattern=\\d{6}", nil)
		w := httptest.NewRecorder()
		mux.ServeHTTP(w, r)

		var resp map[string]any
		json.NewDecoder(w.Body).Decode(&resp)
		data := resp["data"].([]any)
		if len(data) == 0 {
			t.Error("expected at least one code with custom pattern")
		}
	})

	t.Run("invalid pattern", func(t *testing.T) {
		r := httptest.NewRequest("GET", "/api/smtp/message/code-uuid1/codes?pattern=[invalid", nil)
		w := httptest.NewRecorder()
		mux.ServeHTTP(w, r)

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

	t.Run("html body codes", func(t *testing.T) {
		storeSMTPEvent(t, store, "code-uuid2", ParsedEmail{
			Subject: "HTML OTP",
			HTML:    "<p>Your verification code: <strong>654321</strong></p>",
		}, 1001.0, "default")