7cd4606278
Add Node.TextLength() to walk the ProseMirror tree and sum actual user text. Introduce ProseMirrorDocumentMaxTextLength validator that enforces a 50k character limit on extracted text, paired with a 500k byte safety cap on the raw JSON string. Extend e2e tests for content length. Signed-off-by: Émile Ré <emile@getprobo.com>
99 lines
3.0 KiB
Go
99 lines
3.0 KiB
Go
// Copyright (c) 2026 Probo Inc <hello@getprobo.com>.
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
// PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
package validator
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestProseMirrorDocumentContent(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
validDoc := `{"type":"doc","content":[{"type":"paragraph","content":[{"type":"text","text":"hi"}]}]}`
|
|
|
|
tests := []struct {
|
|
name string
|
|
value any
|
|
wantError bool
|
|
}{
|
|
{"empty string", "", false},
|
|
{"whitespace only", " \n\t ", false},
|
|
{"valid doc", validDoc, false},
|
|
{"plain text", "not json", true},
|
|
{"non-doc root", `{"type":"paragraph","content":[]}`, true},
|
|
{"nil value", nil, false},
|
|
{"nil *string", (*string)(nil), false},
|
|
{"non-string", 1, true},
|
|
}
|
|
|
|
fn := ProseMirrorDocumentContent()
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
err := fn(tt.value)
|
|
if (err != nil) != tt.wantError {
|
|
t.Errorf("ProseMirrorDocumentContent() error = %v, wantError %v", err, tt.wantError)
|
|
}
|
|
if err != nil && err.Code != ErrorCodeInvalidFormat {
|
|
t.Errorf("expected code %s, got %s", ErrorCodeInvalidFormat, err.Code)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func proseMirrorDoc(text string) string {
|
|
return `{"type":"doc","content":[{"type":"paragraph","content":[{"type":"text","text":"` + text + `"}]}]}`
|
|
}
|
|
|
|
func TestProseMirrorDocumentMaxTextLength(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
const maxLen = 10
|
|
|
|
tests := []struct {
|
|
name string
|
|
value any
|
|
wantError bool
|
|
wantCode ErrorCode
|
|
}{
|
|
{"nil value", nil, false, ""},
|
|
{"nil *string", (*string)(nil), false, ""},
|
|
{"empty string", "", false, ""},
|
|
{"whitespace only", " \n\t ", false, ""},
|
|
{"under limit", proseMirrorDoc("hello"), false, ""},
|
|
{"at limit", proseMirrorDoc(strings.Repeat("a", maxLen)), false, ""},
|
|
{"over limit", proseMirrorDoc(strings.Repeat("a", maxLen+1)), true, ErrorCodeTooLong},
|
|
{"invalid json skipped", "not json", false, ""},
|
|
{"non-string", 42, true, ErrorCodeInvalidFormat},
|
|
}
|
|
|
|
fn := ProseMirrorDocumentMaxTextLength(maxLen)
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
err := fn(tt.value)
|
|
if (err != nil) != tt.wantError {
|
|
t.Errorf("ProseMirrorDocumentMaxTextLength() error = %v, wantError %v", err, tt.wantError)
|
|
}
|
|
if err != nil && tt.wantCode != "" && err.Code != tt.wantCode {
|
|
t.Errorf("expected code %s, got %s", tt.wantCode, err.Code)
|
|
}
|
|
})
|
|
}
|
|
}
|