video_test.go

package youtube

import (
	"io"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func ExampleClient_GetStream() {
	client := Client{Debug: true}

	video, err := client.GetVideo("https://www.youtube.com/watch?v=9_MbW9FK1fA")
	if err != nil {
		panic(err)
	}

	// Typically youtube only provides separate streams for video and audio.
	// If you want audio and video combined, take a look a the downloader package.
	format := video.Formats.FindByQuality("medium")
	reader, _, err := client.GetStream(video, format)
	if err != nil {
		panic(err)
	}

	// do something with the reader

	reader.Close()
}

func TestSimpleTest(t *testing.T) {
	client := Client{Debug: true, ChunkSize: Size10Mb}

	video, err := client.GetVideo("https://www.youtube.com/watch?v=9_MbW9FK1fA")
	require.NoError(t, err, "get body")

	_, err = client.GetTranscript(video)
	require.NoError(t, err, "get transcript")

	// Typically youtube only provides separate streams for video and audio.
	// If you want audio and video combined, take a look a the downloader package.
	format := video.Formats.FindByQuality("hd1080")

	start := time.Now()
	reader, _, err := client.GetStream(video, format)
	require.NoError(t, err, "get stream")

	t.Log("Duration Milliseconds: ", time.Since(start).Milliseconds())

	// do something with the reader
	b, err := io.ReadAll(reader)
	require.NoError(t, err, "read body")

	t.Log("Downloaded ", len(b))
}

func TestDownload_Regular(t *testing.T) {

	testcases := []struct {
		name       string
		url        string
		outputFile string
		itagNo     int
		quality    string
	}{
		{
			// Video from issue #25
			name:       "default",
			url:        "https://www.youtube.com/watch?v=54e6lBE3BoQ",
			outputFile: "default_test.mp4",
			quality:    "",
		},
		{
			// Video from issue #25
			name:       "quality:medium",
			url:        "https://www.youtube.com/watch?v=54e6lBE3BoQ",
			outputFile: "medium_test.mp4",
			quality:    "medium",
		},
		{
			name: "without-filename",
			url:  "https://www.youtube.com/watch?v=n3kPvBCYT3E",
		},
		{
			name:       "Format",
			url:        "https://www.youtube.com/watch?v=54e6lBE3BoQ",
			outputFile: "muxedstream_test.mp4",
			itagNo:     18,
		},
		{
			name:       "AdaptiveFormat_video",
			url:        "https://www.youtube.com/watch?v=54e6lBE3BoQ",
			outputFile: "adaptiveStream_video_test.m4v",
			itagNo:     134,
		},
		{
			name:       "AdaptiveFormat_audio",
			url:        "https://www.youtube.com/watch?v=54e6lBE3BoQ",
			outputFile: "adaptiveStream_audio_test.m4a",
			itagNo:     140,
		},
		{
			// Video from issue #138
			name:       "NotPlayableInEmbed",
			url:        "https://www.youtube.com/watch?v=gr-IqFcNExY",
			outputFile: "not_playable_in_embed.mp4",
		},
	}
	for _, tc := range testcases {
		t.Run(tc.name, func(t *testing.T) {
			require := require.New(t)

			video, err := testClient.GetVideo(tc.url)
			require.NoError(err)

			var format *Format
			if tc.itagNo > 0 {
				format = video.Formats.FindByItag(tc.itagNo)
				require.NotNil(format)
			} else {
				format = &video.Formats[0]
			}

			url, err := testClient.GetStreamURL(video, format)
			require.NoError(err)
			require.NotEmpty(url)
		})
	}
}

func TestDownload_WhenPlayabilityStatusIsNotOK(t *testing.T) {
	testcases := []struct {
		issue   string
		videoID string
		err     string
	}{
		{
			issue:   "issue#65",
			videoID: "9ja-K2FslBU",
			err:     `status: ERROR`,
		},
		{
			issue:   "issue#59",
			videoID: "nINQjT7Zr9w",
			err:     ErrVideoPrivate.Error(),
		},
	}

	for _, tc := range testcases {
		t.Run(tc.issue, func(t *testing.T) {
			_, err := testClient.GetVideo(tc.videoID)
			require.Error(t, err)
			require.Contains(t, err.Error(), tc.err)
		})
	}
}

// See https://github.com/kkdai/youtube/pull/238
func TestDownload_SensitiveContent(t *testing.T) {
	_, err := testClient.GetVideo("MS91knuzoOA")
	require.EqualError(t, err, "can't bypass age restriction: embedding of this video has been disabled")
}