Skip to content

Instantly share code, notes, and snippets.

@mattli001

mattli001/ffmpeg_test.go

Created May 19, 2017 01:56
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save mattli001/012709aa0be1fb0c173f2a7fbce6295d to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save mattli001/012709aa0be1fb0c173f2a7fbce6295d to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. mattli001 created this gist May 19, 2017.
    125 changes: 125 additions & 0 deletions ffmpeg_test.go
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,125 @@
    package ffmpeg

    import (
    "bytes"
    "context"
    "io"
    "os"
    "os/exec"
    "strings"
    "testing"

    "github.com/wader/ydls/leaktest"
    )

    var testFfmpeg = os.Getenv("TEST_FFMPEG") != ""

    func dummyFile(t *testing.T, format string, acodec string, vcodec string) io.Reader {
    var err error

    // file with black screen and no sound
    dummyFileCmd := exec.Command(
    "ffmpeg",
    "-f", "lavfi", "-i", "color=s=cga:d=1",
    "-f", "lavfi", "-i", "anullsrc",
    "-map", "0:0", "-acodec", acodec,
    "-map", "1:0", "-vcodec", vcodec,
    "-shortest",
    "-f", format,
    "-",
    )

    stdoutBuf := &bytes.Buffer{}
    stderrBuf := &bytes.Buffer{}
    dummyFileCmd.Stdout = stdoutBuf
    dummyFileCmd.Stderr = stderrBuf

    if err = dummyFileCmd.Run(); err != nil {
    t.Logf("cmd failed: %s", strings.Join(dummyFileCmd.Args, " "))
    t.Log(string(stderrBuf.Bytes()))
    t.Fatal(err)
    }

    return stdoutBuf
    }

    func TestProbe(t *testing.T) {
    if !testFfmpeg {
    t.Skip("TEST_FFMPEG env not set")
    }

    defer leaktest.Check(t)()

    pi, probeErr := Probe(context.Background(), dummyFile(t, "matroska", "mp3", "h264"), nil, nil)
    if probeErr != nil {
    t.Error(probeErr)
    }

    if pi.FormatName() != "matroska" {
    t.Fatalf("FormatName should be matroska, is %s", pi.FormatName())
    }
    if pi.ACodec() != "mp3" {
    t.Fatalf("ACodec should be mp3, is %s", pi.ACodec())
    }
    if pi.VCodec() != "h264" {
    t.Fatalf("VCodec should be h264, is %s", pi.VCodec())
    }
    }

    type closeBuffer struct {
    bytes.Buffer
    }

    func (closeBuffer) Close() error {
    return nil
    }

    func TestStart(t *testing.T) {
    if !testFfmpeg {
    t.Skip("TEST_FFMPEG env not set")
    }

    defer leaktest.Check(t)()

    file := dummyFile(t, "matroska", "mp3", "h264")
    output := &closeBuffer{}

    ffmpegP := &FFmpeg{
    StreamMaps: []StreamMap{
    StreamMap{
    Reader: file,
    Specifier: "a:0",
    Codec: "acodec:vorbis",
    },
    StreamMap{
    Reader: file,
    Specifier: "v:0",
    Codec: "vcodec:vp8",
    },
    },
    Format: Format{Name: "matroska"},
    DebugLog: nil, // log.New(os.Stdout, "debug> ", 0),
    Stderr: nil, // writelogger.New(log.New(os.Stdout, "stderr> ", 0), ""),
    Stdout: output,
    }

    if err := ffmpegP.Start(context.Background()); err != nil {
    t.Fatal(err)
    }
    ffmpegP.Wait()

    pi, piErr := Probe(context.Background(), bytes.NewBuffer(output.Bytes()), nil, nil)
    if piErr != nil {
    t.Fatal(piErr)
    }

    if pi.FormatName() != "matroska" {
    t.Fatalf("FormatName should be matroska, is %s", pi.FormatName())
    }
    if pi.ACodec() != "vorbis" {
    t.Fatalf("ACodec should be vorbis, is %s", pi.ACodec())
    }
    if pi.VCodec() != "vp8" {
    t.Fatalf("VCodec should be vp8, is %s", pi.VCodec())
    }
    }