fix: port mailbox validation from im server

2026-09-15 13:31:45 +08:00
parent 1d994c9302
commit cf08f7ceeb
5 changed files with 95 additions and 2 deletions
@@ -0,0 +1,51 @@
+package httpapi
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"git.118im.com/im_team/im_workbench/internal/store"
+)
+
+func TestValidateMailboxSubmissionRejectsTooLongContent(t *testing.T) {
+	req := httptest.NewRequest(http.MethodPost, "/v1/workbench/executive-mailbox/submissions", nil)
+	rec := httptest.NewRecorder()
+
+	if validateMailboxSubmission(rec, req, "建议", strings.Repeat("a", maxMailboxContentLength+1)) {
+		t.Fatal("validateMailboxSubmission returned true for too-long content")
+	}
+	if rec.Code != http.StatusBadRequest {
+		t.Fatalf("status=%d body=%s", rec.Code, rec.Body.String())
+	}
+	var body apiError
+	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
+		t.Fatalf("unmarshal response: %v", err)
+	}
+	if body.Reason != reasonValidationFailed {
+		t.Fatalf("reason=%q body=%s", body.Reason, rec.Body.String())
+	}
+	if body.Details["field"] != "content" {
+		t.Fatalf("details=%v", body.Details)
+	}
+}
+
+func TestWriteStoreErrorMapsPolicyToValidationFailed(t *testing.T) {
+	req := httptest.NewRequest(http.MethodPost, "/v1/workbench/executive-mailbox/submissions", nil)
+	rec := httptest.NewRecorder()
+
+	writeStoreError(rec, req, store.ErrPolicy)
+
+	if rec.Code != http.StatusBadRequest {
+		t.Fatalf("status=%d body=%s", rec.Code, rec.Body.String())
+	}
+	var body apiError
+	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
+		t.Fatalf("unmarshal response: %v", err)
+	}
+	if body.Reason != reasonValidationFailed {
+		t.Fatalf("reason=%q body=%s", body.Reason, rec.Body.String())
+	}
+}