bilinovel-downloader/downloader/bilinovel/bilinovel_test.go

package bilinovel

import (
	"bilinovel-downloader/model"
	"strings"
	"testing"

	"github.com/PuerkitoBio/goquery"
)

func TestCleanChapterContentRemovesAdvertisementContainers(t *testing.T) {
	doc, err := goquery.NewDocumentFromReader(strings.NewReader(`
		<div id="acontent">
			<div class="cgo"><ins class="adsbygoogle" style="height: 280px"></ins></div>
			<div class="csgo"><ins class="adsbygoogle" style="height: 280px"></ins></div>
			<p>保留正文</p>
			<ins class="adsbygoogle" style="height: 280px"></ins>
			<div class="google-auto-placed"><iframe></iframe></div>
			<div class="co"><ins class="adsbygoogle" style="height: 280px"></ins></div>
		</div>
	`))
	if err != nil {
		t.Fatalf("failed to parse fixture: %v", err)
	}

	content := doc.Find("#acontent").First()
	cleanChapterContent(content)

	if remaining := content.Find(".cgo, .csgo, .co, .google-auto-placed, ins.adsbygoogle, iframe").Length(); remaining != 0 {
		t.Fatalf("cleaned content still contains %d advertisement elements", remaining)
	}
	html, err := content.Html()
	if err != nil {
		t.Fatalf("failed to render cleaned content: %v", err)
	}
	if !strings.Contains(html, "<p>保留正文</p>") {
		t.Fatalf("cleaned content lost chapter text: %s", html)
	}
}

func TestMergeDownloadedChapterPreservesVolumeTitle(t *testing.T) {
	testCases := []struct {
		name            string
		listedTitle     string
		downloadedTitle string
	}{
		{
			name:            "circle placeholder",
			listedTitle:     "～第三败～ 欲擒主将，可是马儿太强",
			downloadedTitle: "～第三败～ 〇擒主将，可是马儿太强",
		},
		{
			name:            "heart placeholder",
			listedTitle:     "番外篇 真心话",
			downloadedTitle: "番外篇 真♡话",
		},
	}

	for _, testCase := range testCases {
		t.Run(testCase.name, func(t *testing.T) {
			listedChapter := &model.Chapter{
				Title: testCase.listedTitle,
				Url:   "https://www.bilinovel.com/novel/3095/186113.html",
			}
			downloadedChapter := &model.Chapter{
				Id:    186113,
				Title: testCase.downloadedTitle,
				Url:   "https://www.bilinovel.com/novel/3095/186113_1.html",
				Content: &model.ChaperContent{
					Html: "<p>正文</p>",
				},
			}

			mergedChapter := mergeDownloadedChapter(listedChapter, downloadedChapter)

			if mergedChapter.Title != listedChapter.Title {
				t.Fatalf("merged title = %q, want volume title %q", mergedChapter.Title, listedChapter.Title)
			}
			if mergedChapter.Content != downloadedChapter.Content {
				t.Fatal("merged chapter did not retain downloaded content")
			}
			if mergedChapter.Url != listedChapter.Url {
				t.Fatalf("merged URL = %q, want volume URL %q", mergedChapter.Url, listedChapter.Url)
			}
			if mergedChapter.Id != downloadedChapter.Id {
				t.Fatalf("merged chapter id = %d, want downloaded id %d", mergedChapter.Id, downloadedChapter.Id)
			}
		})
	}
}