增加回复可见功能

service/content_test.go

@@ -0,0 +1,28 @@
+package service
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestRedactReplyOnlyHTML(t *testing.T) {
+	in := `<p>公开</p><reply-only><p>秘密答案</p></reply-only>`
+	out := RedactReplyOnlyHTML(in)
+	if strings.Contains(out, "秘密答案") {
+		t.Fatalf("不应保留回复可见正文，得到: %q", out)
+	}
+	if !strings.Contains(out, `data-locked="true"`) || !strings.Contains(out, "reply-only") {
+		t.Fatalf("应保留锁定壳，得到: %q", out)
+	}
+	if !strings.Contains(out, "公开") {
+		t.Fatalf("不应误删公开段落，得到: %q", out)
+	}
+}
+
+func TestRedactGatedPostHTML(t *testing.T) {
+	in := `<members-only><p>登录密</p></members-only><reply-only><p>回复密</p></reply-only>`
+	out := RedactGatedPostHTML(in)
+	if strings.Contains(out, "登录密") || strings.Contains(out, "回复密") {
+		t.Fatalf("门控正文应被遮盖，得到: %q", out)
+	}
+}