
Rename the file containing the command executor from `command_exe.go` to `cmd_executor.go`), to distinguish it from the implementation of the `exec` command (`command_exec.go`). No functional changes.
73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package worker
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/benbjohnson/clock"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"projects.blender.org/studio/flamenco/internal/worker/mocks"
|
|
"projects.blender.org/studio/flamenco/pkg/api"
|
|
)
|
|
|
|
type CommandExecutorMocks struct {
|
|
cli *mocks.MockCommandLineRunner
|
|
listener *mocks.MockCommandListener
|
|
clock *clock.Mock
|
|
}
|
|
|
|
func testCommandExecutor(t *testing.T, mockCtrl *gomock.Controller) (*CommandExecutor, *CommandExecutorMocks) {
|
|
cli := mocks.NewMockCommandLineRunner(mockCtrl)
|
|
listener := mocks.NewMockCommandListener(mockCtrl)
|
|
clock := mockedClock(t)
|
|
|
|
ce := NewCommandExecutor(cli, listener, clock)
|
|
mocks := CommandExecutorMocks{
|
|
cli: cli,
|
|
listener: listener,
|
|
clock: clock,
|
|
}
|
|
|
|
return ce, &mocks
|
|
}
|
|
|
|
func TestCmdSettingAsStrings(t *testing.T) {
|
|
cmd := api.Command{
|
|
Name: "test",
|
|
Parameters: map[string]interface{}{
|
|
"strings": []string{"a", "b"},
|
|
"ints": []int{3, 4},
|
|
"floats": []float64{0.47, 0.327},
|
|
"mixed": []interface{}{"a", 47, 0.327},
|
|
},
|
|
}
|
|
|
|
{
|
|
slice, ok := cmdParameterAsStrings(cmd, "strings")
|
|
if ok {
|
|
assert.Equal(t, []string{"a", "b"}, slice)
|
|
} else {
|
|
t.Error("not ok")
|
|
}
|
|
}
|
|
{
|
|
_, ok := cmdParameterAsStrings(cmd, "ints")
|
|
assert.False(t, ok, "only []string or []interface{} are expected to work")
|
|
}
|
|
{
|
|
_, ok := cmdParameterAsStrings(cmd, "floats")
|
|
assert.False(t, ok, "only []string or []interface{} are expected to work")
|
|
}
|
|
{
|
|
slice, ok := cmdParameterAsStrings(cmd, "mixed")
|
|
if ok {
|
|
assert.Equal(t, []string{"a", "47", "0.327"}, slice)
|
|
} else {
|
|
t.Error("not ok")
|
|
}
|
|
}
|
|
}
|