Avoid usage of OBUElements in av1_packet_test

It's deprecated.
pion · Apr 15, 2024 · 242f5de · 242f5de
1 parent 1a1a9c0
commit 242f5de
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 10 deletions.
diff --git a/codecs/av1_packet.go b/codecs/av1_packet.go
@@ -171,6 +171,8 @@ func (p *AV1Packet) Unmarshal(payload []byte) ([]byte, error) {
 			return nil, err
 		}
 		p.OBUElements = elts
+	} else {
+		p.OBUElements = nil
 	}
 
 	return payload[1:], nil

diff --git a/codecs/av1_packet_test.go b/codecs/av1_packet_test.go
@@ -275,20 +275,32 @@ func TestAV1_Unmarshal(t *testing.T) {
 	}
 
 	av1Pkt := &AV1Packet{}
-	if _, err := av1Pkt.Unmarshal(av1Payload); err != nil {
+	av1Pkt.SetDontParseBody(true)
+	payload, err := av1Pkt.Unmarshal(av1Payload)
+	if err != nil {
 		t.Fatal(err)
 	}
 
+	t.Log(*av1Pkt)
 	if !reflect.DeepEqual(av1Pkt, &AV1Packet{
-		Z: false,
-		Y: true,
-		W: 2,
-		N: true,
-		OBUElements: [][]byte{
-			av1Payload[2:14],
-			av1Payload[14:],
-		},
+		Z:             false,
+		Y:             true,
+		W:             2,
+		N:             true,
+		dontParseBody: true,
 	}) {
-		t.Fatal("AV1 Unmarshal didn't store the expected results in the packet")
+		t.Error("AV1 Unmarshal didn't store the expected results in the packet")
+	}
+
+	obus, err := av1Pkt.ParseBody(payload)
+	if err != nil {
+		t.Fatal("ParseOBUElements: ", err)
+	}
+
+	if !reflect.DeepEqual(obus, [][]byte{
+		av1Payload[2:14],
+		av1Payload[14:],
+	}) {
+		t.Fatal("Unexpected value ParseOBUElements")
 	}
 }