node build fixed

seanime-2.9.10/internal/library/scanner/media_tree_analysis_test.go

@@ -0,0 +1,170 @@
+package scanner
+
+import (
+	"context"
+	"github.com/davecgh/go-spew/spew"
+	"github.com/stretchr/testify/assert"
+	"seanime/internal/api/anilist"
+	"seanime/internal/api/metadata"
+	"seanime/internal/test_utils"
+	"seanime/internal/util/limiter"
+	"testing"
+	"time"
+)
+
+func TestMediaTreeAnalysis(t *testing.T) {
+	test_utils.InitTestProvider(t, test_utils.Anilist())
+
+	anilistClient := anilist.TestGetMockAnilistClient()
+
+	anilistRateLimiter := limiter.NewAnilistLimiter()
+	tree := anilist.NewCompleteAnimeRelationTree()
+
+	metadataProvider := metadata.GetMockProvider(t)
+
+	tests := []struct {
+		name                          string
+		mediaId                       int
+		absoluteEpisodeNumber         int
+		expectedRelativeEpisodeNumber int
+	}{
+		{
+			name:                          "Media Tree Analysis for 86 - Eighty Six Part 2",
+			mediaId:                       131586, // 86 - Eighty Six Part 2
+			absoluteEpisodeNumber:         23,
+			expectedRelativeEpisodeNumber: 12,
+		},
+		{
+			name:                          "Oshi no Ko Season 2",
+			mediaId:                       150672, // 86 - Eighty Six Part 2
+			absoluteEpisodeNumber:         12,
+			expectedRelativeEpisodeNumber: 1,
+		},
+		{
+			name:                          "Re:zero",
+			mediaId:                       21355, // Re:Zero kara Hajimeru Isekai Seikatsu
+			absoluteEpisodeNumber:         51,
+			expectedRelativeEpisodeNumber: 1,
+		},
+	}
+
+	for _, tt := range tests {
+
+		t.Run(tt.name, func(t *testing.T) {
+
+			mediaF, err := anilistClient.BaseAnimeByID(context.Background(), &tt.mediaId)
+			if err != nil {
+				t.Fatal("expected media, got not found")
+			}
+			media := mediaF.GetMedia()
+
+			// +---------------------+
+			// |     MediaTree       |
+			// +---------------------+
+
+			err = media.FetchMediaTree(
+				anilist.FetchMediaTreeAll,
+				anilistClient,
+				anilistRateLimiter,
+				tree,
+				anilist.NewCompleteAnimeCache(),
+			)
+
+			if err != nil {
+				t.Fatal("expected media tree, got error:", err.Error())
+			}
+
+			// +---------------------+
+			// |  MediaTreeAnalysis  |
+			// +---------------------+
+
+			mta, err := NewMediaTreeAnalysis(&MediaTreeAnalysisOptions{
+				tree:             tree,
+				metadataProvider: metadataProvider,
+				rateLimiter:      limiter.NewLimiter(time.Minute, 25),
+			})
+			if err != nil {
+				t.Fatal("expected media tree analysis, got error:", err.Error())
+			}
+
+			// +---------------------+
+			// |  Relative Episode   |
+			// +---------------------+
+
+			relEp, _, ok := mta.getRelativeEpisodeNumber(tt.absoluteEpisodeNumber)
+
+			if assert.Truef(t, ok, "expected relative episode number %v for absolute episode number %v, nothing found", tt.expectedRelativeEpisodeNumber, tt.absoluteEpisodeNumber) {
+
+				assert.Equal(t, tt.expectedRelativeEpisodeNumber, relEp)
+
+			}
+
+		})
+
+	}
+
+}
+
+func TestMediaTreeAnalysis2(t *testing.T) {
+
+	anilistClient := anilist.TestGetMockAnilistClient()
+	anilistRateLimiter := limiter.NewAnilistLimiter()
+	tree := anilist.NewCompleteAnimeRelationTree()
+
+	metadataProvider := metadata.GetMockProvider(t)
+
+	tests := []struct {
+		name    string
+		mediaId int
+	}{
+		{
+			name:    "Media Tree Analysis",
+			mediaId: 375, // Soreyuke! Uchuu Senkan Yamamoto Yohko
+		},
+	}
+
+	for _, tt := range tests {
+
+		t.Run(tt.name, func(t *testing.T) {
+
+			media, err := anilistClient.BaseAnimeByID(context.Background(), &tt.mediaId)
+			if err != nil {
+				t.Fatal("expected media, got error:", err.Error())
+			}
+
+			// +---------------------+
+			// |     MediaTree       |
+			// +---------------------+
+
+			err = media.GetMedia().FetchMediaTree(
+				anilist.FetchMediaTreeAll,
+				anilistClient,
+				anilistRateLimiter,
+				tree,
+				anilist.NewCompleteAnimeCache(),
+			)
+
+			if err != nil {
+				t.Fatal("expected media tree, got error:", err.Error())
+			}
+
+			// +---------------------+
+			// |  MediaTreeAnalysis  |
+			// +---------------------+
+
+			mta, err := NewMediaTreeAnalysis(&MediaTreeAnalysisOptions{
+				tree:             tree,
+				metadataProvider: metadataProvider,
+				rateLimiter:      limiter.NewLimiter(time.Minute, 25),
+			})
+			if err != nil {
+				t.Fatal("expected media tree analysis, got error:", err.Error())
+			}
+
+			t.Log(spew.Sdump(mta))
+
+		})
+
+	}
+
+}