23 lines
729 B
Go
23 lines
729 B
Go
package handler
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestValidateMessageContent(t *testing.T) {
|
|
validImage := "data:image/png;base64,aGVsbG8="
|
|
if content, err := validateMessageContent("image", validImage); err != nil || content != validImage {
|
|
t.Fatalf("合法图片校验失败: content=%q err=%v", content, err)
|
|
}
|
|
if _, err := validateMessageContent("image", "data:image/webp;base64,aGVsbG8="); err == nil {
|
|
t.Fatal("不支持的图片格式未被拦截")
|
|
}
|
|
if _, err := validateMessageContent("text", " "); err == nil {
|
|
t.Fatal("空白文本未被拦截")
|
|
}
|
|
if _, err := validateMessageContent("text", strings.Repeat("字", maxTextMessageLength+1)); err == nil {
|
|
t.Fatal("超长文本未被拦截")
|
|
}
|
|
}
|