improve fuzz tests (#168)

bluenviron · Aug 1, 2024 · 307ccc2 · 307ccc2
1 parent 1518119
commit 307ccc2
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/pkg/playlist/media_test.go b/pkg/playlist/media_test.go
@@ -318,6 +318,9 @@ func FuzzMediaUnmarshal(f *testing.F) {
 
 	f.Fuzz(func(_ *testing.T, a string) {
 		var m Media
-		m.Unmarshal([]byte(a)) //nolint:errcheck
+		err := m.Unmarshal([]byte(a))
+		if err == nil {
+			m.Marshal() //nolint:errcheck
+		}
 	})
 }
diff --git a/pkg/playlist/multivariant_test.go b/pkg/playlist/multivariant_test.go
@@ -879,6 +879,9 @@ func FuzzMultivariantUnmarshal(f *testing.F) {
 
 	f.Fuzz(func(_ *testing.T, a string) {
 		var m Multivariant
-		m.Unmarshal([]byte(a)) //nolint:errcheck
+		err := m.Unmarshal([]byte(a))
+		if err == nil {
+			m.Marshal() //nolint:errcheck
+		}
 	})
 }
diff --git a/pkg/playlist/playlist.go b/pkg/playlist/playlist.go
@@ -13,8 +13,10 @@ const (
 
 // Playlist is either Media or Multivariant.
 type Playlist interface {
-	isPlaylist()
 	Unmarshal([]byte) error
+	Marshal() ([]byte, error)
+
+	isPlaylist()
 }
 
 func findType(buf []byte) (Playlist, error) {

diff --git a/pkg/playlist/playlist_test.go b/pkg/playlist/playlist_test.go
@@ -34,6 +34,9 @@ func FuzzPlaylistUnmarshal(f *testing.F) {
 	}
 
 	f.Fuzz(func(_ *testing.T, a string) {
-		Unmarshal([]byte(a)) //nolint:errcheck
+		pl, err := Unmarshal([]byte(a))
+		if err == nil {
+			pl.Marshal() //nolint:errcheck
+		}
 	})
 }