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")
|
||
|
}
|