feat: add allanime seasonal queries and provider show model
This commit is contained in:
90
integrations/playback/allanime/sequels_test.go
Normal file
90
integrations/playback/allanime/sequels_test.go
Normal file
@@ -0,0 +1,90 @@
|
||||
package allanime
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDirectSequelsReturnsOnlyPlayableSequels(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_, _ = w.Write([]byte(`{"data":{"show":{"relatedShows":[
|
||||
{"relation":"sequel","showId":"season-2"},
|
||||
{"relation":"side story","showId":"ova"}
|
||||
]}}}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider := NewAllAnimeProvider()
|
||||
provider.httpClient = server.Client()
|
||||
provider.baseURL = server.URL
|
||||
providerShow := ProviderShow{ID: "season-1"}
|
||||
|
||||
got, err := provider.DirectSequels(context.Background(), providerShow)
|
||||
if err != nil {
|
||||
t.Fatalf("DirectSequels: %v", err)
|
||||
}
|
||||
if len(got) != 1 || got[0] != "season-2" {
|
||||
t.Fatalf("DirectSequels = %v, want [season-2]", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetProviderShowKeepsOnlyPositiveIntegerEpisodes(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_, _ = w.Write([]byte(`{"data":{"show":{"_id":"season-2","name":"Example Season 2","englishName":"Example Season 2","malId":"42","status":"Finished","episodeCount":"2","availableEpisodesDetail":{"sub":["2","1","0","1.5"],"dub":["1"],"raw":[]}}}}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider := NewAllAnimeProvider()
|
||||
provider.httpClient = server.Client()
|
||||
provider.baseURL = server.URL
|
||||
|
||||
got, err := provider.GetProviderShow(context.Background(), "season-2")
|
||||
if err != nil {
|
||||
t.Fatalf("GetProviderShow: %v", err)
|
||||
}
|
||||
if got.MalID != 42 || len(got.SubEpisodes) != 2 || len(got.DubEpisodes) != 1 {
|
||||
t.Fatalf("GetProviderShow = %+v", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSeasonalShowsReturnsPlayableTVAnime(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var request struct {
|
||||
Variables struct {
|
||||
Page int `json:"page"`
|
||||
} `json:"variables"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&request); err != nil {
|
||||
t.Fatalf("decode request: %v", err)
|
||||
}
|
||||
edges := make([]map[string]any, 0)
|
||||
switch request.Variables.Page {
|
||||
case 1:
|
||||
for i := 0; i < 40; i++ {
|
||||
edges = append(edges, map[string]any{"_id": "empty", "malId": "12", "type": "TV", "availableEpisodesDetail": map[string]any{"sub": []string{}, "dub": []string{}}})
|
||||
}
|
||||
case 2:
|
||||
edges = append(edges, map[string]any{"_id": "tv", "name": "Summer Show", "malId": "10", "type": "TV", "season": map[string]any{"quarter": "Summer", "year": 2026}, "availableEpisodesDetail": map[string]any{"sub": []string{"1"}, "dub": []string{}}})
|
||||
}
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{"data": map[string]any{"shows": map[string]any{"edges": edges}}})
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider := NewAllAnimeProvider()
|
||||
provider.httpClient = server.Client()
|
||||
provider.baseURL = server.URL
|
||||
|
||||
got, err := provider.SeasonalShows(context.Background(), "summer", 2026)
|
||||
if err != nil {
|
||||
t.Fatalf("SeasonalShows: %v", err)
|
||||
}
|
||||
if len(got) != 1 || got[0].MalID != 10 {
|
||||
t.Fatalf("SeasonalShows = %+v", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user