photoprism/internal/config/config_ffmpeg_test.go
Lukas 83d10ea00e
Config: Add video transcoding resolution option (#3498)
* Adds resolution limit

* Fixes 2 video filters and makes the env variable work

* Adds FFMPEG_RESOLUTION to all the docker-compose files
2023-06-28 15:22:52 +02:00

102 lines
3.1 KiB
Go

package config
import (
"testing"
"github.com/photoprism/photoprism/internal/ffmpeg"
"github.com/stretchr/testify/assert"
)
func TestConfig_FFmpegEncoder(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, ffmpeg.SoftwareEncoder, c.FFmpegEncoder())
c.options.FFmpegEncoder = "nvidia"
assert.Equal(t, ffmpeg.NvidiaEncoder, c.FFmpegEncoder())
c.options.FFmpegEncoder = "intel"
assert.Equal(t, ffmpeg.IntelEncoder, c.FFmpegEncoder())
c.options.FFmpegEncoder = "xxx"
assert.Equal(t, ffmpeg.SoftwareEncoder, c.FFmpegEncoder())
c.options.FFmpegEncoder = ""
assert.Equal(t, ffmpeg.SoftwareEncoder, c.FFmpegEncoder())
}
func TestConfig_FFmpegEnabled(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, true, c.FFmpegEnabled())
c.options.DisableFFmpeg = true
assert.Equal(t, false, c.FFmpegEnabled())
}
func TestConfig_FFmpegBitrate(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, 50, c.FFmpegBitrate())
c.options.FFmpegBitrate = 1000
assert.Equal(t, 960, c.FFmpegBitrate())
c.options.FFmpegBitrate = -5
assert.Equal(t, 50, c.FFmpegBitrate())
c.options.FFmpegBitrate = 800
assert.Equal(t, 800, c.FFmpegBitrate())
}
func TestConfig_FFmpegResolution(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, 144, c.FFmpegResolution())
c.options.FFmpegResolution = 1920
assert.Equal(t, 1920, c.FFmpegResolution())
c.options.FFmpegResolution = 8640
assert.Equal(t, 4320, c.FFmpegResolution())
}
func TestConfig_FFmpegBitrateExceeded(t *testing.T) {
c := NewConfig(CliTestContext())
c.options.FFmpegBitrate = 0
assert.False(t, c.FFmpegBitrateExceeded(0.95))
assert.False(t, c.FFmpegBitrateExceeded(1.05))
assert.False(t, c.FFmpegBitrateExceeded(2.05))
c.options.FFmpegBitrate = 1
assert.False(t, c.FFmpegBitrateExceeded(0.95))
assert.False(t, c.FFmpegBitrateExceeded(1.0))
assert.True(t, c.FFmpegBitrateExceeded(1.05))
assert.True(t, c.FFmpegBitrateExceeded(2.05))
c.options.FFmpegBitrate = 50
assert.False(t, c.FFmpegBitrateExceeded(0.95))
assert.False(t, c.FFmpegBitrateExceeded(1.05))
assert.False(t, c.FFmpegBitrateExceeded(2.05))
c.options.FFmpegBitrate = -5
assert.False(t, c.FFmpegBitrateExceeded(0.95))
assert.False(t, c.FFmpegBitrateExceeded(1.05))
assert.False(t, c.FFmpegBitrateExceeded(2.05))
}
func TestConfig_FFmpegMapVideo(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, ffmpeg.MapVideoDefault, c.FFmpegMapVideo())
}
func TestConfig_FFmpegMapAudio(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, ffmpeg.MapAudioDefault, c.FFmpegMapAudio())
}
func TestConfig_FFmpegOptions(t *testing.T) {
c := NewConfig(CliTestContext())
bitrate := "25M"
resolution := "1080"
opt, err := c.FFmpegOptions(ffmpeg.SoftwareEncoder, bitrate, resolution)
assert.NoError(t, err)
assert.Equal(t, c.FFmpegBin(), opt.Bin)
assert.Equal(t, ffmpeg.SoftwareEncoder, opt.Encoder)
assert.Equal(t, bitrate, opt.Bitrate)
assert.Equal(t, ffmpeg.MapVideoDefault, opt.MapVideo)
assert.Equal(t, ffmpeg.MapAudioDefault, opt.MapAudio)
assert.Equal(t, c.FFmpegMapVideo(), opt.MapVideo)
assert.Equal(t, c.FFmpegMapAudio(), opt.MapAudio)
assert.Equal(t, resolution, opt.Resolution)
}