73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
|
package media_test
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"database/sql"
|
||
|
"errors"
|
||
|
"io"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"git.netflux.io/rob/clipper/config"
|
||
|
"git.netflux.io/rob/clipper/generated/mocks"
|
||
|
"git.netflux.io/rob/clipper/generated/store"
|
||
|
"git.netflux.io/rob/clipper/media"
|
||
|
"github.com/google/uuid"
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
"github.com/stretchr/testify/require"
|
||
|
"go.uber.org/zap"
|
||
|
)
|
||
|
|
||
|
func TestGetVideoFromFileStore(t *testing.T) {
|
||
|
ctx := context.Background()
|
||
|
logger := zap.NewNop().Sugar()
|
||
|
|
||
|
videoID := "video001"
|
||
|
mediaSetID := uuid.New()
|
||
|
mediaSet := store.MediaSet{
|
||
|
ID: mediaSetID,
|
||
|
YoutubeID: videoID,
|
||
|
VideoS3UploadedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||
|
VideoS3Key: sql.NullString{String: "videos/myvideo", Valid: true},
|
||
|
}
|
||
|
|
||
|
t.Run("NOK,ErrorFetchingMediaSet", func(t *testing.T) {
|
||
|
var mockStore mocks.Store
|
||
|
mockStore.On("GetMediaSet", ctx, mediaSetID).Return(store.MediaSet{}, errors.New("database fail"))
|
||
|
|
||
|
service := media.NewMediaSetService(&mockStore, nil, nil, nil, config.Config{}, logger)
|
||
|
_, err := service.GetVideo(ctx, mediaSetID)
|
||
|
require.EqualError(t, err, "error getting media set: database fail")
|
||
|
})
|
||
|
|
||
|
t.Run("NOK,ErrorGettingObjectURL", func(t *testing.T) {
|
||
|
var mockStore mocks.Store
|
||
|
mockStore.On("GetMediaSet", ctx, mediaSetID).Return(mediaSet, nil)
|
||
|
|
||
|
var fileStore mocks.FileStore
|
||
|
fileStore.On("GetURL", ctx, "videos/myvideo").Return("", errors.New("key missing"))
|
||
|
|
||
|
service := media.NewMediaSetService(&mockStore, nil, &fileStore, nil, config.Config{}, logger)
|
||
|
_, err := service.GetVideo(ctx, mediaSetID)
|
||
|
require.EqualError(t, err, "error generating presigned URL: key missing")
|
||
|
})
|
||
|
|
||
|
t.Run("OK", func(t *testing.T) {
|
||
|
var mockStore mocks.Store
|
||
|
mockStore.On("GetMediaSet", ctx, mediaSetID).Return(mediaSet, nil)
|
||
|
|
||
|
const url = "https://www.example.com/audio"
|
||
|
var fileStore mocks.FileStore
|
||
|
fileStore.On("GetURL", ctx, "videos/myvideo").Return(url, nil)
|
||
|
|
||
|
service := media.NewMediaSetService(&mockStore, nil, &fileStore, nil, config.Config{}, logger)
|
||
|
stream, err := service.GetVideo(ctx, mediaSetID)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
progress, err := stream.Next()
|
||
|
assert.Equal(t, float32(100), progress.PercentComplete)
|
||
|
assert.Equal(t, url, progress.URL)
|
||
|
assert.Equal(t, io.EOF, err)
|
||
|
})
|
||
|
}
|