2019-12-05 22:35:25 +00:00
|
|
|
package run
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/golang/mock/gomock"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2019-12-09 17:56:02 +00:00
|
|
|
func TestNewUpgrade(t *testing.T) {
|
2019-12-05 22:35:25 +00:00
|
|
|
ctrl := gomock.NewController(t)
|
|
|
|
defer ctrl.Finish()
|
|
|
|
|
|
|
|
mCmd := NewMockcmd(ctrl)
|
2019-12-09 18:52:41 +00:00
|
|
|
originalCommand := command
|
2019-12-05 22:35:25 +00:00
|
|
|
|
2019-12-09 18:52:41 +00:00
|
|
|
command = func(path string, args ...string) cmd {
|
|
|
|
assert.Equal(t, helmBin, path)
|
2019-12-09 17:56:02 +00:00
|
|
|
assert.Equal(t, []string{"upgrade", "--install", "jonas_brothers_only_human", "at40"}, args)
|
2019-12-05 22:35:25 +00:00
|
|
|
|
|
|
|
return mCmd
|
|
|
|
}
|
2019-12-09 18:52:41 +00:00
|
|
|
defer func() { command = originalCommand }()
|
2019-12-05 22:35:25 +00:00
|
|
|
|
|
|
|
mCmd.EXPECT().
|
|
|
|
Stdout(gomock.Any())
|
|
|
|
mCmd.EXPECT().
|
|
|
|
Stderr(gomock.Any())
|
|
|
|
mCmd.EXPECT().
|
|
|
|
Run().
|
|
|
|
Times(1)
|
|
|
|
|
2019-12-09 17:56:02 +00:00
|
|
|
u := NewUpgrade("jonas_brothers_only_human", "at40")
|
|
|
|
u.Run()
|
2019-12-05 22:35:25 +00:00
|
|
|
}
|