Simplified tests

Author: djthorpe

decoder_test.go

@@ -101,15 +101,25 @@ func Test_decoder_003(t *testing.T) {
 	// Decode video frames and resize them
 	assert := assert.New(t)
 
+	// Open the file
 	manager := NewManager()
 	media, err := manager.Open("./etc/test/sample.mp4", nil)
 	if !assert.NoError(err) {
 		t.SkipNow()
 	}
 	defer media.Close()
 
+	// Initialize
+	n := 0
+	// tmpdir := t.TempDir()
+	tmpdir, err := os.MkdirTemp("", "media_test")
+	if !assert.NoError(err) {
+		t.SkipNow()
+	}
+
+	// Create a decoder which resizes the video frames
 	decoder, err := media.Decoder(func(stream Stream) (Parameters, error) {
-		// Make greyscale images
+		// Rescale the video
 		if stream.Type() == VIDEO {
 			return manager.VideoParameters(640, 480, "yuv420p")
 		}
@@ -120,23 +130,17 @@ func Test_decoder_003(t *testing.T) {
 		t.SkipNow()
 	}
 
-	// Frame function
-	n := 0
-	// tmpdir := t.TempDir()
-	tmpdir, err := os.MkdirTemp("", "media_test")
-	if !assert.NoError(err) {
-		t.SkipNow()
-	}
+	// This is the function which processes the frames
 	framefn := func(frame Frame) error {
-		if frame.Type() != VIDEO {
-			return nil
-		}
+		// Create an output file
 		filename := filepath.Join(tmpdir, fmt.Sprintf("frame%03d.jpg", n))
 		w, err := os.Create(filename)
 		if err != nil {
 			return err
 		}
 		defer w.Close()
+
+		// Decode the frame into an image, save as JPEG
 		if image, err := frame.Image(); err != nil {
 			return err
 		} else if err := jpeg.Encode(w, image, nil); err != nil {
@@ -145,6 +149,7 @@ func Test_decoder_003(t *testing.T) {
 			t.Logf("Frame %d: %dx%d (%q) => %s", n, frame.Width(), frame.Height(), frame.PixelFormat(), filename)
 			n++
 		}
+
 		// Stop after 10 frames
 		if n >= 10 {
 			return io.EOF
@@ -153,6 +158,41 @@ func Test_decoder_003(t *testing.T) {
 		}
 	}
 
-	// decode frames from the stream
+	// Finally, this is where we actually decode frames from the stream
+	assert.NoError(decoder.Decode(context.Background(), framefn))
+}
+
+func Test_decoder_004(t *testing.T) {
+	// Decode audio frames
+	assert := assert.New(t)
+
+	// Open the file
+	manager := NewManager()
+	media, err := manager.Open("./etc/test/sample.mp4", nil)
+	if !assert.NoError(err) {
+		t.SkipNow()
+	}
+	defer media.Close()
+
+	// Create a decoder to decompress the audio
+	decoder, err := media.Decoder(func(stream Stream) (Parameters, error) {
+		// Audio - pass through
+		if stream.Type() == AUDIO {
+			return stream.Parameters(), nil
+		}
+		// Ignore other streams
+		return nil, nil
+	})
+	if !assert.NoError(err) {
+		t.SkipNow()
+	}
+
+	// This is the function which processes the audio frames
+	framefn := func(frame Frame) error {
+		t.Log(frame)
+		return nil
+	}
+
+	// Finally, this is where we actually decode frames from the stream
 	assert.NoError(decoder.Decode(context.Background(), framefn))
 }