34 lines
603 B
Go
34 lines
603 B
Go
package run
|
|
|
|
import (
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestUpgrade(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
|
|
mCmd := NewMockcmd(ctrl)
|
|
originalCommand := Command
|
|
|
|
Command = func(path string, args ...string) cmd {
|
|
assert.Equal(t, HELM_BIN, path)
|
|
assert.Equal(t, []string{"upgrade", "arg1", "arg2"}, args)
|
|
|
|
return mCmd
|
|
}
|
|
defer func() { Command = originalCommand }()
|
|
|
|
mCmd.EXPECT().
|
|
Stdout(gomock.Any())
|
|
mCmd.EXPECT().
|
|
Stderr(gomock.Any())
|
|
mCmd.EXPECT().
|
|
Run().
|
|
Times(1)
|
|
|
|
Upgrade("arg1", "arg2")
|
|
}
|