70 lines
2.5 KiB
Go
70 lines
2.5 KiB
Go
package httppresentation
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
appusecase "lambdamail/protocols/internal/application/usecase"
|
|
)
|
|
|
|
func TestRouter_MtaSts_Autoconfig_SecurityTxt_Ingest(t *testing.T) {
|
|
uc := appusecase.NewIngestReportsUseCase(nil)
|
|
router := NewRouter(uc, func() error { return nil })
|
|
|
|
// 1. Health
|
|
req := httptest.NewRequest(http.MethodGet, "/health", nil)
|
|
w := httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK {
|
|
t.Errorf("health status = %d", w.Code)
|
|
}
|
|
|
|
// 2. MTA-STS
|
|
req = httptest.NewRequest(http.MethodGet, "/.well-known/mta-sts.txt", nil)
|
|
req.Host = "mta-sts.example.test"
|
|
w = httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK || !strings.Contains(w.Body.String(), "version: STSv1") {
|
|
t.Errorf("mta-sts response = %q", w.Body.String())
|
|
}
|
|
|
|
// 3. Thunderbird Autoconfig
|
|
req = httptest.NewRequest(http.MethodGet, "/mail/config-v1.1.xml", nil)
|
|
req.Host = "autoconfig.example.test"
|
|
w = httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK || !strings.Contains(w.Body.String(), "<clientConfig") {
|
|
t.Errorf("autoconfig response = %q", w.Body.String())
|
|
}
|
|
|
|
// 4. Outlook Autodiscover
|
|
req = httptest.NewRequest(http.MethodPost, "/autodiscover/autodiscover.xml", nil)
|
|
req.Host = "autodiscover.example.test"
|
|
w = httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK || !strings.Contains(w.Body.String(), "<Autodiscover") {
|
|
t.Errorf("autodiscover response = %q", w.Body.String())
|
|
}
|
|
|
|
// 5. Security.txt
|
|
req = httptest.NewRequest(http.MethodGet, "/.well-known/security.txt", nil)
|
|
req.Host = "example.test"
|
|
w = httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK || !strings.Contains(w.Body.String(), "Contact: mailto:security@example.test") {
|
|
t.Errorf("security.txt response = %q", w.Body.String())
|
|
}
|
|
|
|
// 6. Report Ingestion (DMARC)
|
|
dmarcPayload := `<?xml version="1.0"?><feedback><report_metadata><org_name>org</org_name><report_id>id</report_id><date_range><begin>1</begin><end>2</end></date_range></report_metadata><policy_published><domain>dom</domain></policy_published></feedback>`
|
|
req = httptest.NewRequest(http.MethodPost, "/api/v1/reports/dmarc", bytes.NewBufferString(dmarcPayload))
|
|
w = httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK || !strings.Contains(w.Body.String(), `"status":"ok"`) {
|
|
t.Errorf("dmarc ingest status = %d, body = %q", w.Code, w.Body.String())
|
|
}
|
|
}
|