v0.11.2

- 修复未设置redirect时无法请求的问题 - 修复动画命中存在无法作为文件夹名字符的问题 - 优化部分单元测量mock方式
wetor · Feb 3, 2024 · 8f5dad7 · 8f5dad7
1 parent d261a39
commit 8f5dad7
Show file tree

Hide file tree

Showing 11 changed files with 297 additions and 108 deletions.
diff --git a/Changelog.md b/Changelog.md
@@ -4,6 +4,8 @@
 - **配置文件版本号为`1.7.1`**
 - 更新文档
 - 修复设置Mikan的redirect时，RSS订阅中未进行替换的问题
+- 修复动画命中存在无法作为文件夹名字符的问题
+- 优化配置文件升级代码逻辑
 
 ### v0.11.1
 - 优化代码，调整项目结构

diff --git a/internal/animego/anisource/anisource_test.go b/internal/animego/anisource/anisource_test.go
@@ -1,11 +1,9 @@
 package anisource_test
 
 import (
+	"context"
 	"fmt"
-	"io"
-	"net/url"
 	"os"
-	"path"
 	"sync"
 	"testing"
 
@@ -17,61 +15,23 @@ import (
 	"github.com/wetor/AnimeGo/internal/animego/anisource/mikan"
 	"github.com/wetor/AnimeGo/internal/animego/anisource/themoviedb"
 	"github.com/wetor/AnimeGo/internal/api"
+	"github.com/wetor/AnimeGo/internal/constant"
 	"github.com/wetor/AnimeGo/internal/exceptions"
 	"github.com/wetor/AnimeGo/internal/models"
 	"github.com/wetor/AnimeGo/internal/pkg/request"
 	"github.com/wetor/AnimeGo/internal/plugin"
 	"github.com/wetor/AnimeGo/internal/wire"
 	"github.com/wetor/AnimeGo/pkg/cache"
-	"github.com/wetor/AnimeGo/pkg/json"
 	"github.com/wetor/AnimeGo/pkg/log"
 	"github.com/wetor/AnimeGo/pkg/utils"
-	"github.com/wetor/AnimeGo/pkg/xpath"
 	"github.com/wetor/AnimeGo/test"
 )
 
 var (
 	mikanSource api.AniSource
+	ctx, cancel = context.WithCancel(context.Background())
 )
 
-func HookGetWriter(uri string, w io.Writer) error {
-	log.Infof("Mock HTTP GET %s", uri)
-	id := path.Base(xpath.P(uri))
-	jsonData, err := test.GetData("mikan", id)
-	if err != nil {
-		return err
-	}
-	_, err = w.Write(jsonData)
-	if err != nil {
-		return err
-	}
-	return nil
-}
-
-func HookGet(uri string, body interface{}) error {
-	log.Infof("Mock HTTP GET %s", uri)
-	u, err := url.Parse(uri)
-	if err != nil {
-		return err
-	}
-	id := u.Query().Get("with_text_query")
-	if len(id) == 0 {
-		id = path.Base(xpath.P(u.Path))
-	}
-
-	p := test.GetDataPath("themoviedb", id)
-	if !utils.IsExist(p) {
-		p = test.GetDataPath("bangumi", id)
-	}
-
-	jsonData, err := os.ReadFile(p)
-	if err != nil {
-		return err
-	}
-	_ = json.Unmarshal(jsonData, body)
-	return nil
-}
-
 func TestMain(m *testing.M) {
 	fmt.Println("begin")
 	_ = utils.CreateMutiDir("data")
@@ -104,7 +64,33 @@ func TestMain(m *testing.M) {
 		Cache:     b,
 		CacheTime: int64(7 * 24 * 60 * 60),
 	})
+
+	bangumiHost := test.MockBangumiStart(ctx)
+	mikanHost := test.MockMikanStart(ctx)
+	themoviedbHost := test.MockThemoviedbStart(ctx)
+	request.Init(&request.Options{
+		Host: map[string]*request.HostOptions{
+			constant.BangumiHost: {
+				Redirect: bangumiHost,
+			},
+			constant.MikanHost: {
+				Redirect: mikanHost,
+				Cookie: map[string]string{
+					constant.MikanAuthCookie: "MikanAuthCookie",
+				},
+			},
+			constant.ThemoviedbHost: {
+				Redirect: themoviedbHost,
+				Params: map[string]string{
+					constant.ThemoviedbApiKey: "123456",
+				},
+			},
+		},
+	})
+
 	m.Run()
+
+	cancel()
 	b.Close()
 	bangumiCache.Close()
 	_ = log.Close()
@@ -177,9 +163,6 @@ func TestMikan_Parse(t *testing.T) {
 		},
 	}
 
-	test.Hook(request.GetWriter, HookGetWriter)
-	test.Hook(request.Get, HookGet)
-	defer test.UnHook()
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			gotAnime, err := mikanSource.Parse(tt.args.opts)

diff --git a/internal/animego/anisource/bangumi/bangumi_test.go b/internal/animego/anisource/bangumi/bangumi_test.go
@@ -1,6 +1,7 @@
 package bangumi_test
 
 import (
+	"context"
 	"fmt"
 	"os"
 	"sync"
@@ -10,6 +11,7 @@ import (
 	"github.com/stretchr/testify/assert"
 
 	"github.com/wetor/AnimeGo/internal/animego/anisource/bangumi"
+	"github.com/wetor/AnimeGo/internal/constant"
 	"github.com/wetor/AnimeGo/internal/exceptions"
 	"github.com/wetor/AnimeGo/internal/pkg/request"
 	"github.com/wetor/AnimeGo/pkg/cache"
@@ -18,10 +20,9 @@ import (
 	"github.com/wetor/AnimeGo/test"
 )
 
-const testdata = "bangumi"
-
 var (
 	bangumiInst *bangumi.Bangumi
+	ctx, cancel = context.WithCancel(context.Background())
 )
 
 func TestMain(m *testing.M) {
@@ -31,11 +32,14 @@ func TestMain(m *testing.M) {
 		File:  "data/log.log",
 		Debug: true,
 	})
+	host := test.MockBangumiStart(ctx)
 	request.Init(&request.Options{
-		Debug: true,
+		Host: map[string]*request.HostOptions{
+			constant.BangumiHost: {
+				Redirect: host,
+			},
+		},
 	})
-	test.HookAll(testdata, nil)
-	defer test.UnHook()
 	mutex := sync.Mutex{}
 
 	db := cache.NewBolt()
@@ -50,6 +54,7 @@ func TestMain(m *testing.M) {
 	})
 	m.Run()
 
+	cancel()
 	db.Close()
 	bangumiCache.Close()
 	_ = log.Close()

diff --git a/internal/animego/anisource/mikan/mikan_test.go b/internal/animego/anisource/mikan/mikan_test.go
@@ -1,6 +1,7 @@
 package mikan_test
 
 import (
+	"context"
 	"fmt"
 	"os"
 	"testing"
@@ -9,16 +10,17 @@ import (
 	"github.com/stretchr/testify/assert"
 
 	"github.com/wetor/AnimeGo/internal/animego/anisource/mikan"
+	"github.com/wetor/AnimeGo/internal/constant"
 	"github.com/wetor/AnimeGo/internal/exceptions"
+	"github.com/wetor/AnimeGo/internal/pkg/request"
 	"github.com/wetor/AnimeGo/pkg/cache"
 	"github.com/wetor/AnimeGo/pkg/log"
 	"github.com/wetor/AnimeGo/test"
 )
 
-const testdata = "mikan"
-
 var (
-	mikanInst *mikan.Mikan
+	mikanInst   *mikan.Mikan
+	ctx, cancel = context.WithCancel(context.Background())
 )
 
 func TestMain(m *testing.M) {
@@ -27,8 +29,18 @@ func TestMain(m *testing.M) {
 		File:  "data/test.log",
 		Debug: true,
 	})
-	test.HookAll(testdata, nil)
-	defer test.UnHook()
+
+	host := test.MockMikanStart(ctx)
+	request.Init(&request.Options{
+		Host: map[string]*request.HostOptions{
+			constant.MikanHost: {
+				Redirect: host,
+				Cookie: map[string]string{
+					constant.MikanAuthCookie: "MikanAuthCookie",
+				},
+			},
+		},
+	})
 
 	db := cache.NewBolt()
 	db.Open("data/bolt.db")
@@ -38,6 +50,7 @@ func TestMain(m *testing.M) {
 	})
 	m.Run()
 
+	cancel()
 	db.Close()
 	_ = log.Close()
 	_ = os.RemoveAll("data")

diff --git a/internal/animego/anisource/themoviedb/themoviedb_test.go b/internal/animego/anisource/themoviedb/themoviedb_test.go
@@ -1,28 +1,26 @@
 package themoviedb_test
 
 import (
+	"context"
 	"fmt"
-	"net/url"
 	"os"
-	"path"
 	"testing"
 
 	"github.com/pkg/errors"
 	"github.com/stretchr/testify/assert"
 
 	"github.com/wetor/AnimeGo/internal/animego/anisource/themoviedb"
+	"github.com/wetor/AnimeGo/internal/constant"
 	"github.com/wetor/AnimeGo/internal/exceptions"
 	"github.com/wetor/AnimeGo/internal/pkg/request"
 	"github.com/wetor/AnimeGo/pkg/cache"
 	"github.com/wetor/AnimeGo/pkg/log"
-	"github.com/wetor/AnimeGo/pkg/xpath"
 	"github.com/wetor/AnimeGo/test"
 )
 
-const testdata = "themoviedb"
-
 var (
-	tmdbInst *themoviedb.Themoviedb
+	tmdbInst    *themoviedb.Themoviedb
+	ctx, cancel = context.WithCancel(context.Background())
 )
 
 func TestMain(m *testing.M) {
@@ -36,23 +34,20 @@ func TestMain(m *testing.M) {
 	tmdbInst = themoviedb.NewThemoviedb(&themoviedb.Options{
 		Cache: db,
 	})
+	host := test.MockThemoviedbStart(ctx)
 	request.Init(&request.Options{
-		Debug: true,
-	})
-	test.HookGet(testdata, func(uri string) string {
-		u, err := url.Parse(uri)
-		if err != nil {
-			return ""
-		}
-		id := u.Query().Get("with_text_query")
-		if len(id) == 0 {
-			id = path.Base(xpath.P(u.Path))
-		}
-		return id
+		Host: map[string]*request.HostOptions{
+			constant.ThemoviedbHost: {
+				Redirect: host,
+				Params: map[string]string{
+					constant.ThemoviedbApiKey: "123456",
+				},
+			},
+		},
 	})
-	defer test.UnHook()
 	m.Run()
 
+	cancel()
 	db.Close()
 	_ = log.Close()
 	_ = os.RemoveAll("data")

diff --git a/internal/animego/parser/parser_test.go b/internal/animego/parser/parser_test.go
@@ -1,10 +1,9 @@
 package parser_test
 
 import (
+	"context"
 	"fmt"
-	"net/url"
 	"os"
-	"path"
 	"testing"
 
 	"github.com/pkg/errors"
@@ -14,20 +13,24 @@ import (
 	"github.com/wetor/AnimeGo/internal/animego/anisource"
 	"github.com/wetor/AnimeGo/internal/animego/anisource/mikan"
 	"github.com/wetor/AnimeGo/internal/animego/parser"
+	"github.com/wetor/AnimeGo/internal/constant"
 	"github.com/wetor/AnimeGo/internal/exceptions"
 	"github.com/wetor/AnimeGo/internal/models"
+	"github.com/wetor/AnimeGo/internal/pkg/request"
 	"github.com/wetor/AnimeGo/internal/pkg/torrent"
 	"github.com/wetor/AnimeGo/internal/plugin"
 	"github.com/wetor/AnimeGo/pkg/cache"
 	pkgExceptions "github.com/wetor/AnimeGo/pkg/exceptions"
 	"github.com/wetor/AnimeGo/pkg/log"
-	"github.com/wetor/AnimeGo/pkg/xpath"
 	"github.com/wetor/AnimeGo/test"
 )
 
 const testdata = "parser"
 
-var mgr *parser.Manager
+var (
+	mgr         *parser.Manager
+	ctx, cancel = context.WithCancel(context.Background())
+)
 
 func TestMain(m *testing.M) {
 	fmt.Println("begin")
@@ -39,17 +42,17 @@ func TestMain(m *testing.M) {
 	torrent.Init(&torrent.Options{
 		TempPath: "data",
 	})
-	test.HookGetWriter(testdata, nil)
-	test.HookGet(testdata, func(uri string) string {
-		u, err := url.Parse(uri)
-		if err != nil {
-			return ""
-		}
-		id := u.Query().Get("with_text_query")
-		if len(id) == 0 {
-			id = path.Base(xpath.P(u.Path))
-		}
-		return id
+	host := test.MockThemoviedbStart(ctx)
+	request.Init(&request.Options{
+		Host: map[string]*request.HostOptions{
+			constant.ThemoviedbHost: {
+				Redirect: host,
+				Params: map[string]string{
+					constant.ThemoviedbApiKey: "123456",
+					"testdata":                testdata,
+				},
+			},
+		},
 	})
 	test.Hook(torrent.LoadUri, HookLoadUri)
 	defer test.UnHook()
@@ -84,6 +87,7 @@ func TestMain(m *testing.M) {
 
 	m.Run()
 
+	cancel()
 	b.Close()
 	_ = log.Close()
 	_ = os.RemoveAll("data")