add synchronous video frame bridge

2026-08-27 12:44:18 +03:00
parent 4ffbf0266c
commit 7a6da099df
2 changed files with 198 additions and 0 deletions
@@ -0,0 +1,138 @@
+package playback
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+)
+
+const videoBridgeTestTimeout = time.Second
+
+func TestVideoBridgeDeliversFrameAndCompletionResult(t *testing.T) {
+	bridge := NewVideoBridge()
+	wantErr := errors.New("stage frame")
+	wantFrame := VideoFrame{
+		Index:   42,
+		Width:   1920,
+		Height:  1080,
+		Stride:  7680,
+		Payload: []byte{1, 2, 3},
+	}
+	consumeResult := make(chan error, 1)
+
+	go func() {
+		consumeResult <- bridge.ConsumeVideo(context.Background(), wantFrame)
+	}()
+
+	ctx, cancel := context.WithTimeout(context.Background(), videoBridgeTestTimeout)
+	defer cancel()
+	pending, err := bridge.Next(ctx)
+	if err != nil {
+		t.Fatalf("Next() error = %v", err)
+	}
+	if pending.Frame.Index != wantFrame.Index {
+		t.Fatalf("Next() frame index = %d, want %d", pending.Frame.Index, wantFrame.Index)
+	}
+	if &pending.Frame.Payload[0] != &wantFrame.Payload[0] {
+		t.Fatal("Next() copied the borrowed payload")
+	}
+
+	pending.Complete(wantErr)
+	select {
+	case err := <-consumeResult:
+		if !errors.Is(err, wantErr) {
+			t.Fatalf("ConsumeVideo() error = %v, want %v", err, wantErr)
+		}
+	case <-time.After(videoBridgeTestTimeout):
+		t.Fatal("ConsumeVideo() did not return after completion")
+	}
+}
+
+func TestVideoBridgeConsumeHonorsCancellationBeforeDelivery(t *testing.T) {
+	bridge := NewVideoBridge()
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	err := bridge.ConsumeVideo(ctx, VideoFrame{})
+	if !errors.Is(err, context.Canceled) {
+		t.Fatalf("ConsumeVideo() error = %v, want %v", err, context.Canceled)
+	}
+}
+
+func TestVideoBridgeConsumeWaitsForCompletionAfterDelivery(t *testing.T) {
+	bridge := NewVideoBridge()
+	ctx, cancel := context.WithCancel(context.Background())
+	consumeResult := make(chan error, 1)
+
+	go func() {
+		consumeResult <- bridge.ConsumeVideo(ctx, VideoFrame{Index: 7})
+	}()
+
+	nextCtx, nextCancel := context.WithTimeout(context.Background(), videoBridgeTestTimeout)
+	defer nextCancel()
+	pending, err := bridge.Next(nextCtx)
+	if err != nil {
+		t.Fatalf("Next() error = %v", err)
+	}
+	cancel()
+
+	select {
+	case err := <-consumeResult:
+		t.Fatalf("ConsumeVideo() returned before completion: %v", err)
+	case <-time.After(20 * time.Millisecond):
+	}
+
+	pending.Complete(nil)
+	select {
+	case err := <-consumeResult:
+		if err != nil {
+			t.Fatalf("ConsumeVideo() error = %v, want nil", err)
+		}
+	case <-time.After(videoBridgeTestTimeout):
+		t.Fatal("ConsumeVideo() did not return after completion")
+	}
+}
+
+func TestPendingVideoFrameCompleteIsIdempotent(t *testing.T) {
+	bridge := NewVideoBridge()
+	consumeResult := make(chan error, 1)
+
+	go func() {
+		consumeResult <- bridge.ConsumeVideo(context.Background(), VideoFrame{})
+	}()
+
+	ctx, cancel := context.WithTimeout(context.Background(), videoBridgeTestTimeout)
+	defer cancel()
+	pending, err := bridge.Next(ctx)
+	if err != nil {
+		t.Fatalf("Next() error = %v", err)
+	}
+
+	firstErr := errors.New("first")
+	pending.Complete(firstErr)
+	pending.Complete(errors.New("second"))
+
+	select {
+	case err := <-consumeResult:
+		if !errors.Is(err, firstErr) {
+			t.Fatalf("ConsumeVideo() error = %v, want %v", err, firstErr)
+		}
+	case <-time.After(videoBridgeTestTimeout):
+		t.Fatal("ConsumeVideo() did not return")
+	}
+}
+
+func TestVideoBridgeNextHonorsCancellation(t *testing.T) {
+	bridge := NewVideoBridge()
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	pending, err := bridge.Next(ctx)
+	if pending != nil {
+		t.Fatalf("Next() pending = %#v, want nil", pending)
+	}
+	if !errors.Is(err, context.Canceled) {
+		t.Fatalf("Next() error = %v, want %v", err, context.Canceled)
+	}
+}