test RTP timestamps

2026-08-05 12:06:13 -05:00
parent c3bcc2cdcc
commit 49604b8b53
14 changed files with 430 additions and 68 deletions
@@ -0,0 +1,44 @@
+//go:build overlord_webrtc
+
+package webrtcpub
+
+import (
+	"testing"
+	"time"
+
+	"github.com/pion/rtp"
+)
+
+type recordingRTPSink struct {
+	packets []*rtp.Packet
+}
+
+func (s *recordingRTPSink) WriteRTP(packet *rtp.Packet) error {
+	clone := *packet
+	clone.Payload = append([]byte(nil), packet.Payload...)
+	s.packets = append(s.packets, &clone)
+	return nil
+}
+
+func TestH264TrackWriterUsesCaptureClockForTimestamps(t *testing.T) {
+	sink := &recordingRTPSink{}
+	writer := newH264TrackWriterWithSink(sink, 1000)
+	start := time.Unix(100, 0)
+
+	if err := writer.WriteH264(annexBKey(1), start); err != nil {
+		t.Fatal(err)
+	}
+	firstCount := len(sink.packets)
+	if firstCount == 0 {
+		t.Fatal("first frame produced no RTP packets")
+	}
+	if err := writer.WriteH264(annexBDelta(2), start.Add(250*time.Millisecond)); err != nil {
+		t.Fatal(err)
+	}
+	if len(sink.packets) <= firstCount {
+		t.Fatal("second frame produced no RTP packets")
+	}
+	if got, want := sink.packets[firstCount].Timestamp-sink.packets[0].Timestamp, uint32(22500); got != want {
+		t.Fatalf("capture-clock RTP delta=%d, want %d", got, want)
+	}
+}