unarr/internal/engine/vaapi_args_test.go

package engine

import (
	"strings"
	"testing"
)

func TestBuildHLSFFmpegArgsVAAPI(t *testing.T) {
	cfg := HLSSessionConfig{
		SessionID:  "test",
		SourcePath: "/tmp/test.mkv",
		Quality:    "720p",
		AudioIndex: 0,
		Transcode: TranscodeRuntime{
			FFmpegPath:  "/usr/bin/ffmpeg",
			FFprobePath: "/usr/bin/ffprobe",
			HWAccel:     HWAccelVAAPI,
		},
	}
	probe := &StreamProbe{Width: 1920, Height: 1080, DurationSec: 100}
	args := buildHLSFFmpegArgsAt(cfg, probe, "/tmp/tmpdir", 0, 0)
	got := strings.Join(args, " ")

	wants := []string{
		"-hwaccel vaapi",
		"-vaapi_device /dev/dri/renderD128",
		"-c:v h264_vaapi",
		"format=nv12",
		"hwupload",
	}
	for _, want := range wants {
		if !strings.Contains(got, want) {
			t.Errorf("argv missing %q\n%s", want, got)
		}
	}
	if strings.Contains(got, "scale_vaapi") {
		t.Errorf("argv unexpectedly contains scale_vaapi (mesa bug): %s", got)
	}
	if strings.Contains(got, "format=yuv420p") {
		t.Errorf("argv contains format=yuv420p (libx264 path) for VAAPI codec: %s", got)
	}
}

func TestBuildHLSFFmpegArgsLibx264NoRegression(t *testing.T) {
	cfg := HLSSessionConfig{
		SessionID:  "test",
		SourcePath: "/tmp/test.mkv",
		Quality:    "720p",
		AudioIndex: 0,
		Transcode: TranscodeRuntime{
			FFmpegPath:  "/usr/bin/ffmpeg",
			FFprobePath: "/usr/bin/ffprobe",
			HWAccel:     HWAccelNone,
		},
	}
	probe := &StreamProbe{Width: 1920, Height: 1080, DurationSec: 100}
	args := buildHLSFFmpegArgsAt(cfg, probe, "/tmp/tmpdir", 0, 0)
	got := strings.Join(args, " ")
	for _, want := range []string{"-c:v libx264", "format=yuv420p", "setparams=colorspace=bt709"} {
		if !strings.Contains(got, want) {
			t.Errorf("libx264 argv missing %q: %s", want, got)
		}
	}
	for _, bad := range []string{"-vaapi_device", "format=nv12", "hwupload"} {
		if strings.Contains(got, bad) {
			t.Errorf("libx264 argv unexpectedly contains %q: %s", bad, got)
		}
	}
}