Sybren A. Stüvel
02fac6a4df
Change the package base name of the Go code, from `git.blender.org/flamenco` to `projects.blender.org/studio/flamenco`. The old location, `git.blender.org`, has no longer been use since the [migration to Gitea][1]. The new package names now reflect the actual location where Flamenco is hosted. [1]: https://code.blender.org/2023/02/new-blender-development-infrastructure/
111 lines
3.5 KiB
Go
111 lines
3.5 KiB
Go
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
package worker
|
|
|
|
import (
|
|
"context"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/rs/zerolog"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"projects.blender.org/studio/flamenco/pkg/api"
|
|
)
|
|
|
|
func TestCmdFramesToVideoSimplePosix(t *testing.T) {
|
|
// Windows and non-Windows platforms differ in how they communicate globs to FFmpeg.
|
|
if runtime.GOOS == "windows" {
|
|
t.Skipf("skipping POSIX test on %s", runtime.GOOS)
|
|
}
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
|
|
ce, mocks := testCommandExecutor(t, mockCtrl)
|
|
|
|
taskID := "1d54c6fe-1242-4c8f-bd63-5a09e358d7b6"
|
|
exe := `F:\software\tools\ffmpeg.exe` // Backslashes are tricksy, test with them on all platforms.
|
|
cmd := api.Command{
|
|
Name: "blender",
|
|
Parameters: map[string]interface{}{
|
|
"exe": exe,
|
|
"exeArgs": "-v quiet",
|
|
"argsBefore": []string{"-report"},
|
|
"inputGlob": "path/to/renders/*.png",
|
|
"fps": 10.0,
|
|
"args": []string{
|
|
"-c:v", "hevc",
|
|
"-crf", "31",
|
|
"-vf", "pad=ceil(iw/2)*2:ceil(ih/2)*2",
|
|
},
|
|
"outputFile": "path/to/renders/preview.mkv",
|
|
},
|
|
}
|
|
|
|
cliArgs := []string{
|
|
"-v", "quiet", // exeArgs
|
|
"-report", // argsBefore
|
|
"-r", "10", // input frame rate
|
|
"-pattern_type", "glob", "-i", "path/to/renders/*.png", // inputGlob
|
|
"-c:v", "hevc", "-crf", "31", "-vf", "pad=ceil(iw/2)*2:ceil(ih/2)*2", // args
|
|
"-r", "10", // output frame rate
|
|
"path/to/renders/preview.mkv", // outputFile
|
|
}
|
|
mocks.cli.EXPECT().CommandContext(gomock.Any(), exe, cliArgs).Return(nil)
|
|
|
|
err := ce.cmdFramesToVideo(context.Background(), zerolog.Nop(), taskID, cmd)
|
|
assert.Equal(t, ErrNoExecCmd, err, "nil *exec.Cmd should result in ErrNoExecCmd")
|
|
}
|
|
|
|
func TestCmdFramesToVideoSimpleWindows(t *testing.T) {
|
|
// Windows and non-Windows platforms differ in how they communicate globs to FFmpeg.
|
|
if runtime.GOOS != "windows" {
|
|
t.Skipf("skipping Windows test on %s", runtime.GOOS)
|
|
}
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
|
|
ce, mocks := testCommandExecutor(t, mockCtrl)
|
|
|
|
taskID := "1d54c6fe-1242-4c8f-bd63-5a09e358d7b6"
|
|
exe := `F:\software\tools\ffmpeg.exe`
|
|
cmd := api.Command{
|
|
Name: "blender",
|
|
Parameters: map[string]interface{}{
|
|
"exe": exe,
|
|
"exeArgs": "-v quiet",
|
|
"argsBefore": []string{"-report"},
|
|
// NOTE: these files MUST exist, otherwise the glob index file creation will fail.
|
|
"inputGlob": "command_ffmpeg_test_files/*.png",
|
|
"fps": 10.0,
|
|
"args": []string{
|
|
"-c:v", "hevc",
|
|
"-crf", "31",
|
|
"-vf", "pad=ceil(iw/2)*2:ceil(ih/2)*2",
|
|
},
|
|
"outputFile": "path/to/renders/preview.mkv",
|
|
},
|
|
}
|
|
|
|
cliArgs := []string{
|
|
"-v", "quiet", // exeArgs
|
|
"-report", // argsBefore
|
|
"-f", "concat", "-safe", "0", "-i", "command_ffmpeg_test_files\\ffmpeg-file-index.txt", // input glob
|
|
"-c:v", "hevc", "-crf", "31", "-vf", "pad=ceil(iw/2)*2:ceil(ih/2)*2", // args
|
|
"-r", "10", // fps
|
|
"path/to/renders/preview.mkv", // outputFile
|
|
}
|
|
mocks.cli.EXPECT().
|
|
CommandContext(gomock.Any(), exe, gomock.Any()).
|
|
DoAndReturn(func(ctx context.Context, name string, arg ...string) error {
|
|
assert.EqualValues(t, cliArgs, arg)
|
|
return nil // this is the nil *exec.Cmd referenced below
|
|
})
|
|
|
|
err := ce.cmdFramesToVideo(context.Background(), zerolog.Nop(), taskID, cmd)
|
|
assert.Equal(t, ErrNoExecCmd, err, "nil *exec.Cmd should result in ErrNoExecCmd")
|
|
|
|
assert.NoFileExists(t, "command_ffmpeg_test_files/ffmpeg-file-index.txt",
|
|
"the glob index file should have been removed")
|
|
}
|