Fix unit tests

livepeer · Sep 6, 2024 · bf833e1 · bf833e1
1 parent a01169d
commit bf833e1
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/core/transcoder_test.go b/core/transcoder_test.go
@@ -178,14 +178,24 @@ func TestProfilesToTranscodeOptions(t *testing.T) {
 	}
 	defer func() { common.RandomIDGenerator = oldRandIDFunc }()
 
+	makeMeta := func(p []ffmpeg.VideoProfile, c bool) *SegTranscodingMetadata {
+		return &SegTranscodingMetadata{
+			Profiles:           p,
+			CalcPerceptualHash: c,
+			Metadata: map[string]string{
+				"meta": "data",
+			},
+		}
+	}
+
 	// Test 0 profiles
 	profiles := []ffmpeg.VideoProfile{}
-	opts := profilesToTranscodeOptions(workDir, ffmpeg.Software, profiles, false, nil)
+	opts := profilesToTranscodeOptions(workDir, ffmpeg.Software, makeMeta(profiles, false))
 	assert.Equal(0, len(opts))
 
 	// Test 1 profile
 	profiles = []ffmpeg.VideoProfile{ffmpeg.P144p30fps16x9}
-	opts = profilesToTranscodeOptions(workDir, ffmpeg.Software, profiles, false, nil)
+	opts = profilesToTranscodeOptions(workDir, ffmpeg.Software, makeMeta(profiles, false))
 	assert.Equal(1, len(opts))
 	assert.Equal("foo/out_bar.tempfile", opts[0].Oname)
 	assert.Equal(ffmpeg.Software, opts[0].Accel)
@@ -194,22 +204,25 @@ func TestProfilesToTranscodeOptions(t *testing.T) {
 
 	// Test > 1 profile
 	profiles = []ffmpeg.VideoProfile{ffmpeg.P144p30fps16x9, ffmpeg.P240p30fps16x9}
-	opts = profilesToTranscodeOptions(workDir, ffmpeg.Software, profiles, false, nil)
+	opts = profilesToTranscodeOptions(workDir, ffmpeg.Software, makeMeta(profiles, false))
 	assert.Equal(2, len(opts))
 
 	for i, p := range profiles {
 		assert.Equal("foo/out_bar.tempfile", opts[i].Oname)
 		assert.Equal(ffmpeg.Software, opts[i].Accel)
 		assert.Equal(p, opts[i].Profile)
 		assert.Equal("copy", opts[i].AudioEncoder.Name)
+		assert.Equal(opts[i].Metadata, map[string]string{
+			"meta": "data",
+		})
 	}
 
 	// Test different acceleration value
-	opts = profilesToTranscodeOptions(workDir, ffmpeg.Nvidia, profiles, false, nil)
+	opts = profilesToTranscodeOptions(workDir, ffmpeg.Nvidia, makeMeta(profiles, false))
 	assert.Equal(2, len(opts))
 
 	// Test signature calculation
-	opts = profilesToTranscodeOptions(workDir, ffmpeg.Nvidia, profiles, true, nil)
+	opts = profilesToTranscodeOptions(workDir, ffmpeg.Nvidia, makeMeta(profiles, true))
 	assert.True(opts[0].CalcSign)
 	assert.True(opts[1].CalcSign)