mirror of
https://github.com/mxpv/podsync.git
synced 2024-05-11 05:55:04 +00:00
ligang945
103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
package builder
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/mxpv/podsync/pkg/feed"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mxpv/podsync/pkg/model"
|
|
)
|
|
|
|
var (
|
|
testCtx = context.Background()
|
|
ytKey = os.Getenv("YOUTUBE_TEST_API_KEY")
|
|
)
|
|
|
|
func TestYT_QueryChannel(t *testing.T) {
|
|
if ytKey == "" {
|
|
t.Skip("YouTube API key is not provided")
|
|
}
|
|
|
|
builder, err := NewYouTubeBuilder(ytKey)
|
|
require.NoError(t, err)
|
|
|
|
channel, err := builder.listChannels(testCtx, model.TypeChannel, "UC2yTVSttx7lxAOAzx1opjoA", "id")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "UC2yTVSttx7lxAOAzx1opjoA", channel.Id)
|
|
|
|
channel, err = builder.listChannels(testCtx, model.TypeUser, "fxigr1", "id")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "UCr_fwF-n-2_olTYd-m3n32g", channel.Id)
|
|
}
|
|
|
|
func TestYT_BuildFeed(t *testing.T) {
|
|
if ytKey == "" {
|
|
t.Skip("YouTube API key is not provided")
|
|
}
|
|
|
|
builder, err := NewYouTubeBuilder(ytKey)
|
|
require.NoError(t, err)
|
|
|
|
urls := []string{
|
|
"https://www.youtube.com/channel/UCupvZG-5ko_eiXAupbDfxWw",
|
|
"https://www.youtube.com/playlist?list=PLF7tUDhGkiCk_Ne30zu7SJ9gZF9R9ZruE",
|
|
"https://www.youtube.com/channel/UCK9lZ2lHRBgx2LOcqPifukA",
|
|
"https://youtube.com/user/WylsaLive",
|
|
"https://www.youtube.com/playlist?list=PLUVl5pafUrBydT_gsCjRGeCy0hFHloec8",
|
|
}
|
|
|
|
for _, addr := range urls {
|
|
t.Run(addr, func(t *testing.T) {
|
|
_feed, err := builder.Build(testCtx, &feed.Config{URL: addr})
|
|
require.NoError(t, err)
|
|
|
|
assert.NotEmpty(t, _feed.Title)
|
|
assert.NotEmpty(t, _feed.Description)
|
|
assert.NotEmpty(t, _feed.Author)
|
|
assert.NotEmpty(t, _feed.ItemURL)
|
|
|
|
assert.NotZero(t, len(_feed.Episodes))
|
|
|
|
for _, item := range _feed.Episodes {
|
|
assert.NotEmpty(t, item.Title)
|
|
assert.NotEmpty(t, item.VideoURL)
|
|
assert.NotZero(t, item.Duration)
|
|
|
|
assert.NotEmpty(t, item.Title)
|
|
assert.NotEmpty(t, item.Thumbnail)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestYT_GetVideoCount(t *testing.T) {
|
|
if ytKey == "" {
|
|
t.Skip("YouTube API key is not provided")
|
|
}
|
|
|
|
builder, err := NewYouTubeBuilder(ytKey)
|
|
require.NoError(t, err)
|
|
|
|
feeds := []*model.Info{
|
|
{Provider: model.ProviderYoutube, LinkType: model.TypeUser, ItemID: "fxigr1"},
|
|
{Provider: model.ProviderYoutube, LinkType: model.TypeChannel, ItemID: "UCupvZG-5ko_eiXAupbDfxWw"},
|
|
{Provider: model.ProviderYoutube, LinkType: model.TypePlaylist, ItemID: "PLF7tUDhGkiCk_Ne30zu7SJ9gZF9R9ZruE"},
|
|
{Provider: model.ProviderYoutube, LinkType: model.TypeChannel, ItemID: "UCK9lZ2lHRBgx2LOcqPifukA"},
|
|
{Provider: model.ProviderYoutube, LinkType: model.TypeUser, ItemID: "WylsaLive"},
|
|
{Provider: model.ProviderYoutube, LinkType: model.TypePlaylist, ItemID: "PLUVl5pafUrBydT_gsCjRGeCy0hFHloec8"},
|
|
}
|
|
|
|
for _, _feed := range feeds {
|
|
feedCopy := _feed
|
|
t.Run(_feed.ItemID, func(t *testing.T) {
|
|
count, err := builder.GetVideoCount(testCtx, feedCopy)
|
|
assert.NoError(t, err)
|
|
assert.NotZero(t, count)
|
|
})
|
|
}
|
|
}
|