mirror of
https://github.com/truewhile/MeBox.git
synced 2026-10-04 04:26:38 +08:00
truewhile
108 lines
3.7 KiB
Go
108 lines
3.7 KiB
Go
package reader
|
||
|
||
import (
|
||
"encoding/base64"
|
||
"fmt"
|
||
"net/http"
|
||
"net/http/httptest"
|
||
"strings"
|
||
"testing"
|
||
|
||
"github.com/truewhile/MeBox/internal/model"
|
||
)
|
||
|
||
// 漫画源（聚合类漫画源就是这么写的）正文规则直接给 <img src="…"> 的 HTML。
|
||
// 整行当地址会被代理成一堆取不回的「图」，页面上全是破图；必须按标签抽 src。
|
||
func TestImageRefsInLine(t *testing.T) {
|
||
cases := []struct {
|
||
name string
|
||
line string
|
||
want []string
|
||
}{
|
||
{"纯地址原样返回", "https://cdn.example.com/1.webp", []string{"https://cdn.example.com/1.webp"}},
|
||
{"双引号 img", `<img src="https://cdn.example.com/2.webp">`, []string{"https://cdn.example.com/2.webp"}},
|
||
{"单引号 img", `<img src='https://cdn.example.com/3.webp'/>`, []string{"https://cdn.example.com/3.webp"}},
|
||
{"属性顺序无关", `<img data-x="1" src="https://cdn.example.com/4.webp" class="a">`, []string{"https://cdn.example.com/4.webp"}},
|
||
{"一行多个标签", `<p><img src="/a/1.webp"><img src='/a/2.webp'></p>`, []string{"/a/1.webp", "/a/2.webp"}},
|
||
{"img 没有 src 时跳过", `<img alt="空">`, nil},
|
||
{"非 img 的 HTML 片段跳过", `</div>`, nil},
|
||
}
|
||
for _, c := range cases {
|
||
got := imageRefsInLine(c.line)
|
||
if strings.Join(got, ",") != strings.Join(c.want, ",") {
|
||
t.Fatalf("%s: got %q want %q", c.name, got, c.want)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 端到端：图片型书籍走 GetContentForBook 时，Images 必须是真正的图片地址，
|
||
// 不能把 <img> 标签本身当成图片地址（那会让代理取回一堆 HTML，前端全破图）。
|
||
func TestGetContentForBookExtractsImageSrc(t *testing.T) {
|
||
page := `<html><body><div class="pages">
|
||
<img src="/img/1.webp"><img src="/img/2.webp">
|
||
</div></body></html>`
|
||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
_, _ = w.Write([]byte(page))
|
||
}))
|
||
defer srv.Close()
|
||
|
||
svc, _ := newLoginTestService(t)
|
||
ctx := t.Context()
|
||
srcJSON := fmt.Sprintf(`{
|
||
"bookSourceUrl": %q,
|
||
"bookSourceName": "漫画测试源",
|
||
"bookSourceType": 64,
|
||
"ruleContent": { "content": "class.pages@html" }
|
||
}`, srv.URL)
|
||
sourceID := importTestSource(t, svc, srcJSON, srv.URL)
|
||
|
||
book := &model.ReaderBook{
|
||
UserID: "u1",
|
||
Origin: srv.URL,
|
||
OriginName: "漫画测试源",
|
||
BookURL: srv.URL + "/book/1",
|
||
Name: "航海王",
|
||
Type: 2, // 图片
|
||
}
|
||
if err := svc.repo.CreateBook(ctx, book); err != nil {
|
||
t.Fatalf("创建书籍失败: %v", err)
|
||
}
|
||
if err := svc.SaveChapters(ctx, book.ID, []ChapterInput{
|
||
{Index: 0, Title: "第1话", URL: srv.URL + "/book/1/c1.html"},
|
||
}); err != nil {
|
||
t.Fatalf("写入章节失败: %v", err)
|
||
}
|
||
_ = sourceID
|
||
|
||
out, err := svc.GetContentForBook(ctx, "u1", book.ID, 0)
|
||
if err != nil {
|
||
t.Fatalf("取正文失败: %v", err)
|
||
}
|
||
if out.Type != "image" {
|
||
t.Fatalf("类型 = %q，期望 image", out.Type)
|
||
}
|
||
if len(out.Images) != 2 {
|
||
t.Fatalf("图片数 = %d，期望 2（%v）", len(out.Images), out.Images)
|
||
}
|
||
for i, img := range out.Images {
|
||
// 书架维度返回的是签名代理地址（浏览器取图带不上防盗链头），
|
||
// 把里面的原始地址解出来核对，顺便确认没有被塞进 HTML 标签。
|
||
if !strings.HasPrefix(img, "/api/reader/media?") {
|
||
t.Fatalf("第 %d 张不是代理地址：%q", i, img)
|
||
}
|
||
encoded := img[strings.Index(img, "u=")+2:]
|
||
if end := strings.IndexByte(encoded, '&'); end >= 0 {
|
||
encoded = encoded[:end]
|
||
}
|
||
raw, err := base64.RawURLEncoding.DecodeString(encoded)
|
||
if err != nil {
|
||
t.Fatalf("第 %d 张地址解不开：%q", i, img)
|
||
}
|
||
want := fmt.Sprintf("%s/img/%d.webp", srv.URL, i+1)
|
||
if string(raw) != want {
|
||
t.Fatalf("第 %d 张解析出 %q，期望 %q", i, raw, want)
|
||
}
|
||
}
|
||
}
|