Added a basic transcoder test
This commit is contained in:
parent
a9a46a4493
commit
95fd8aed96
@ -24,6 +24,7 @@ type Transcoder struct {
|
||||
hlsPlaylistLength int
|
||||
segmentLengthSeconds int
|
||||
appendToStream bool
|
||||
ffmpegPath string
|
||||
}
|
||||
|
||||
// HLSVariant is a combination of settings that results in a single HLS stream
|
||||
@ -104,7 +105,7 @@ func (t *Transcoder) getString() string {
|
||||
|
||||
ffmpegFlags := []string{
|
||||
"cat", t.input, "|",
|
||||
config.Config.GetFFMpegPath(),
|
||||
t.ffmpegPath,
|
||||
"-hide_banner",
|
||||
"-i pipe:",
|
||||
t.getVariantsString(),
|
||||
@ -112,7 +113,7 @@ func (t *Transcoder) getString() string {
|
||||
// HLS Output
|
||||
"-f", "hls",
|
||||
"-hls_time", strconv.Itoa(t.segmentLengthSeconds), // Length of each segment
|
||||
"-hls_list_size", strconv.Itoa(config.Config.GetMaxNumberOfReferencedSegmentsInPlaylist()), // Max # in variant playlist
|
||||
"-hls_list_size", strconv.Itoa(t.hlsPlaylistLength), // Max # in variant playlist
|
||||
"-hls_delete_threshold", "10", // Start deleting files after hls_list_size + 10
|
||||
"-hls_flags", strings.Join(hlsOptionFlags, "+"), // Specific options in HLS generation
|
||||
|
||||
@ -175,6 +176,8 @@ func getVariantFromConfigQuality(quality config.StreamQuality, index int) HLSVar
|
||||
// NewTranscoder will return a new Transcoder, populated by the config
|
||||
func NewTranscoder() Transcoder {
|
||||
transcoder := new(Transcoder)
|
||||
transcoder.ffmpegPath = config.Config.GetFFMpegPath()
|
||||
transcoder.hlsPlaylistLength = config.Config.GetMaxNumberOfReferencedSegmentsInPlaylist()
|
||||
|
||||
var outputPath string
|
||||
if config.Config.S3.Enabled || config.Config.IPFS.Enabled {
|
||||
|
30
core/ffmpeg/transcoder_test.go
Normal file
30
core/ffmpeg/transcoder_test.go
Normal file
@ -0,0 +1,30 @@
|
||||
package ffmpeg
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestFFmpegCommand(t *testing.T) {
|
||||
transcoder := new(Transcoder)
|
||||
transcoder.ffmpegPath = "/fake/path/ffmpeg"
|
||||
transcoder.SetSegmentLength(4)
|
||||
transcoder.SetInput("fakecontent.flv")
|
||||
transcoder.SetOutputPath("fakeOutput")
|
||||
transcoder.SetHLSPlaylistLength(10)
|
||||
|
||||
variant := HLSVariant{}
|
||||
variant.videoBitrate = "1200k"
|
||||
variant.isAudioPassthrough = true
|
||||
variant.encoderPreset = "veryfast"
|
||||
variant.SetVideoFramerate(30)
|
||||
|
||||
transcoder.AddVariant(variant)
|
||||
|
||||
cmd := transcoder.getString()
|
||||
|
||||
expected := "cat fakecontent.flv | /fake/path/ffmpeg -hide_banner -i pipe: -map v:0 -c:v:0 libx264 -b:v:0 1200k -map a:0 -c:a:0 copy -r 30 -g 60 -keyint_min 60 -preset veryfast -var_stream_map \"v:0,a:0 \" -f hls -hls_time 4 -hls_list_size 10 -hls_delete_threshold 10 -hls_flags delete_segments+program_date_time+temp_file -tune zerolatency -sc_threshold 0 -master_pl_name stream.m3u8 -strftime 1 -hls_segment_filename fakeOutput/%v/stream-%s.ts -max_muxing_queue_size 400 fakeOutput/%v/stream.m3u8 2> transcoder.log"
|
||||
|
||||
if cmd != expected {
|
||||
t.Errorf("ffmpeg command does not match expected. Got %s, want: %s", cmd, expected)
|
||||
}
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user