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("超长文本未被拦截")
	}
}