message: add GobDecode test

gortc · Nov 1, 2020 · ba5d9a7 · ba5d9a7
1 parent a2af8aa
commit ba5d9a7
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/message_test.go b/message_test.go
@@ -1037,3 +1037,32 @@ func TestMessage_MarshalBinary(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestMessage_GobDecode(t *testing.T) {
+	m := MustBuild(
+		NewSoftware("software"),
+		&XORMappedAddress{
+			IP: net.IPv4(213, 1, 223, 5),
+		},
+	)
+	data, err := m.GobEncode()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Reset m.Raw to check retention.
+	for i := range m.Raw {
+		m.Raw[i] = 0
+	}
+	if err := m.GobDecode(data); err != nil {
+		t.Fatal(err)
+	}
+
+	// Reset data to check retention.
+	for i := range data {
+		data[i] = 0
+	}
+	if err := m.Decode(); err != nil {
+		t.Fatal(err)
+	}
+}