add h264 tests (client and server), hide options that won't work for mac + linux and give warning for mac under certain circumstances (fuck ass file viewing).

2026-03-06 18:32:16 -06:00
parent 67323752cc
commit cc135acd36
4 changed files with 163 additions and 0 deletions
@@ -0,0 +1,82 @@
+package capture
+
+import (
+	"image"
+	"os"
+	"strings"
+	"sync"
+	"testing"
+)
+
+func resetCodecSelectionForTest() {
+	blockCodecOnce = sync.Once{}
+	cachedBlockCodec = ""
+	overrideCodec.Store("")
+}
+
+func TestSetQualityAndCodec_H264FallbackDependsOnAvailability(t *testing.T) {
+	t.Cleanup(resetCodecSelectionForTest)
+
+	SetQualityAndCodec(80, "h264")
+	got := blockCodec()
+
+	if h264Available() {
+		if got != "h264" {
+			t.Fatalf("expected h264 codec when available, got %q", got)
+		}
+		return
+	}
+
+	if got != "jpeg" {
+		t.Fatalf("expected jpeg fallback when h264 unavailable, got %q", got)
+	}
+}
+
+func TestSetQualityAndCodec_InvalidCodecForcesJpeg(t *testing.T) {
+	t.Cleanup(resetCodecSelectionForTest)
+
+	SetQualityAndCodec(75, "invalid-codec")
+	got := blockCodec()
+	if got != "jpeg" {
+		t.Fatalf("expected invalid codec to force jpeg, got %q", got)
+	}
+}
+
+func TestH264AvailabilityDetail_NotEmpty(t *testing.T) {
+	detail := strings.TrimSpace(h264AvailabilityDetail())
+	if detail == "" {
+		t.Fatal("expected h264 availability detail to be non-empty")
+	}
+}
+
+func TestEncodeH264Frame_WhenUnavailableReturnsError(t *testing.T) {
+	if h264Available() {
+		t.Skip("h264 is available in this build; unavailable-path assertion does not apply")
+	}
+
+	img := image.NewRGBA(image.Rect(0, 0, 2, 2))
+	out, err := encodeH264Frame(img)
+	if err == nil {
+		t.Fatal("expected an error when h264 is unavailable")
+	}
+	if out != nil {
+		t.Fatalf("expected nil output when h264 is unavailable, got %d bytes", len(out))
+	}
+}
+
+func TestBlockCodec_UsesEnvWhenNoOverride(t *testing.T) {
+	prev := os.Getenv(blockCodecEnv)
+	if err := os.Setenv(blockCodecEnv, "raw"); err != nil {
+		t.Fatalf("setenv failed: %v", err)
+	}
+	t.Cleanup(func() {
+		_ = os.Setenv(blockCodecEnv, prev)
+		resetCodecSelectionForTest()
+	})
+
+	resetCodecSelectionForTest()
+	got := blockCodec()
+	if got != "raw" {
+		t.Fatalf("expected env codec raw, got %q", got)
+	}
+}