mirror of
https://github.com/xfrr/goffmpeg.git
synced 2025-10-06 15:16:49 +08:00
ec40467798
* cleanup readme & examples * upgrade go version * add makefile with basic commands * add e2e test * add gha * update .gitignore
105 lines
1.9 KiB
Go
Executable File
105 lines
1.9 KiB
Go
Executable File
package test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os/exec"
|
|
"path"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/xfrr/goffmpeg/transcoder"
|
|
)
|
|
|
|
const (
|
|
fixturePath = "./fixtures"
|
|
resultsPath = "./test_results"
|
|
)
|
|
|
|
var (
|
|
// Input files
|
|
input3gp = path.Join(fixturePath, "input.3gp")
|
|
)
|
|
|
|
func TestInputNotFound(t *testing.T) {
|
|
createResultsDir(t)
|
|
var outputPath = path.Join(resultsPath, "notfound.mp4")
|
|
|
|
trans := new(transcoder.Transcoder)
|
|
|
|
err := trans.Initialize("notfound.3gp", outputPath)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestTranscodingProgress(t *testing.T) {
|
|
createResultsDir(t)
|
|
|
|
outputPath := path.Join(resultsPath, "progress.mp4")
|
|
trans := new(transcoder.Transcoder)
|
|
|
|
err := trans.Initialize(input3gp, outputPath)
|
|
assert.Nil(t, err)
|
|
|
|
errCh := trans.Run(true)
|
|
|
|
progress := []transcoder.Progress{}
|
|
for val := range trans.Output() {
|
|
progress = append(progress, val)
|
|
}
|
|
err = <-errCh
|
|
assert.Nil(t, err)
|
|
assert.GreaterOrEqual(t, len(progress), 1)
|
|
checkFileExists(t, outputPath)
|
|
}
|
|
|
|
func TestTranscodePipes(t *testing.T) {
|
|
createResultsDir(t)
|
|
|
|
c1 := exec.Command("cat", input3gp)
|
|
|
|
trans := new(transcoder.Transcoder)
|
|
|
|
err := trans.InitializeEmptyTranscoder()
|
|
assert.Nil(t, err)
|
|
|
|
w, err := trans.CreateInputPipe()
|
|
assert.Nil(t, err)
|
|
c1.Stdout = w
|
|
|
|
r, err := trans.CreateOutputPipe("mp4")
|
|
assert.Nil(t, err)
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
go func() {
|
|
_, err := ioutil.ReadAll(r)
|
|
assert.Nil(t, err)
|
|
|
|
r.Close()
|
|
wg.Done()
|
|
}()
|
|
|
|
go func() {
|
|
err := c1.Run()
|
|
assert.Nil(t, err)
|
|
w.Close()
|
|
}()
|
|
done := trans.Run(false)
|
|
err = <-done
|
|
assert.Nil(t, err)
|
|
|
|
wg.Wait()
|
|
}
|
|
|
|
func createResultsDir(t *testing.T) {
|
|
err := exec.Command("mkdir", "-p", resultsPath).Run()
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func checkFileExists(t *testing.T, filepath string) {
|
|
res, err := exec.Command("cat", filepath).Output()
|
|
assert.Nil(t, err)
|
|
assert.Greater(t, len(res), 0)
|
|
}
|