clipper/backend/youtube/helpers_test.go

package youtube_test

import (
	"testing"

	"git.netflux.io/rob/clipper/youtube"
	youtubev2 "github.com/kkdai/youtube/v2"
	"github.com/stretchr/testify/assert"
)

func TestSortAudio(t *testing.T) {
	formats := []youtubev2.Format{
		{
			MimeType:        `audio/webm; codecs="opus"`,
			Bitrate:         350_000,
			AudioChannels:   2,
			AudioSampleRate: "16000",
		},
		{
			MimeType:        `audio/webm; codecs="opus"`,
			Bitrate:         350_000,
			AudioChannels:   2,
			AudioSampleRate: "48000",
		},
		{
			MimeType:        `audio/mp4; codecs="mp4a.40.2"`,
			Bitrate:         250_000,
			AudioChannels:   2,
			AudioSampleRate: "48000",
		},
		{
			MimeType:        `audio/webm; codecs="opus"`,
			Bitrate:         125_000,
			AudioChannels:   2,
			AudioSampleRate: "48000",
		},
	}

	sortedFormats := youtube.SortAudio(formats)

	assert.Equal(t, formats[1], sortedFormats[0])
	assert.Equal(t, formats[3], sortedFormats[1])
	assert.Equal(t, formats[0], sortedFormats[2])
	assert.Equal(t, formats[2], sortedFormats[3])
}

func TestSortVideo(t *testing.T) {
	formats := []youtubev2.Format{
		{
			MimeType:      `video/mp4; codecs="avc1.42001E, mp4a.40.2"`,
			QualityLabel:  "240p",
			FPS:           30,
			AudioChannels: 2,
		},
		{
			MimeType:      `audio/webm; codecs="opus"`,
			QualityLabel:  "",
			FPS:           0,
			AudioChannels: 2,
		},
		{
			MimeType:      `video/mp4; codecs="avc1.42001E, mp4a.40.2"`,
			QualityLabel:  "720p",
			FPS:           30,
			AudioChannels: 2,
		},
		{
			MimeType:      `video/mp4; codecs="avc1.42001E, mp4a.40.2"`,
			QualityLabel:  "360p",
			FPS:           30,
			AudioChannels: 2,
		},
	}

	sortedFormats := youtube.SortVideo(formats)

	assert.Len(t, sortedFormats, 3)
	assert.Equal(t, formats[3], sortedFormats[0])
	assert.Equal(t, formats[0], sortedFormats[1])
	assert.Equal(t, formats[2], sortedFormats[2])
}