navidrome/engine/media_streamer_test.go

package engine

import (
	"context"
	"io"
	"io/ioutil"
	"net/http"
	"os"
	"strings"

	"github.com/deluan/navidrome/conf"
	"github.com/deluan/navidrome/log"
	"github.com/deluan/navidrome/model"
	"github.com/deluan/navidrome/persistence"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	"gopkg.in/djherbis/fscache.v0"
)

var _ = Describe("MediaStreamer", func() {

	var streamer MediaStreamer
	var ds model.DataStore
	ctx := log.NewContext(nil)

	BeforeEach(func() {
		conf.Server.EnableDownsampling = true
		fs := fscache.NewMemFs()
		cache, _ := fscache.NewCache(fs, nil)
		ds = &persistence.MockDataStore{}
		ds.MediaFile(ctx).(*persistence.MockMediaFile).SetData(`[{"id": "123", "path": "tests/fixtures/test.mp3", "bitRate": 128}]`, 1)
		streamer = NewMediaStreamer(ds, &fakeFFmpeg{}, cache)
	})

	getFile := func(id string, maxBitRate int, format string) (http.File, error) {
		fs, _ := streamer.NewFileSystem(ctx, maxBitRate, format)
		return fs.Open(id)
	}

	Context("NewFileSystem", func() {
		It("returns a File if format is 'raw'", func() {
			Expect(getFile("123", 0, "raw")).To(BeAssignableToTypeOf(&os.File{}))
		})
		It("returns a File if maxBitRate is 0", func() {
			Expect(getFile("123", 0, "mp3")).To(BeAssignableToTypeOf(&os.File{}))
		})
		It("returns a File if maxBitRate is higher than file bitRate", func() {
			Expect(getFile("123", 256, "mp3")).To(BeAssignableToTypeOf(&os.File{}))
		})
		It("returns a transcodingFile if maxBitRate is lower than file bitRate", func() {
			s, err := getFile("123", 64, "mp3")
			Expect(err).To(BeNil())
			Expect(s).To(BeAssignableToTypeOf(&transcodingFile{}))
			Expect(s.(*transcodingFile).bitRate).To(Equal(64))
		})
	})
})

type fakeFFmpeg struct {
}

func (ff *fakeFFmpeg) StartTranscoding(ctx context.Context, path string, maxBitRate int, format string) (f io.ReadCloser, err error) {
	return ioutil.NopCloser(strings.NewReader("fake data")), nil
}