add skip-crds flag support

Signed-off-by: George Kaz <egeorgekaz@gmail.com>
This commit is contained in:
George Kaz
2021-07-20 19:21:44 +01:00
parent 8dba329407
commit e482a144c1
5 changed files with 32 additions and 1 deletions

View File

@@ -24,6 +24,7 @@ type Upgrade struct {
cleanupOnFail bool
certs *repoCerts
createNamespace bool
skipCrds bool
cmd cmd
}
@@ -47,6 +48,7 @@ func NewUpgrade(cfg env.Config) *Upgrade {
cleanupOnFail: cfg.CleanupOnFail,
certs: newRepoCerts(cfg),
createNamespace: cfg.CreateNamespace,
skipCrds: cfg.SkipCrds,
}
}
@@ -100,6 +102,9 @@ func (u *Upgrade) Prepare() error {
if u.createNamespace {
args = append(args, "--create-namespace")
}
if u.skipCrds {
args = append(args, "--skip-crds")
}
for _, vFile := range u.valuesFiles {
args = append(args, "--values", vFile)
}

View File

@@ -220,3 +220,27 @@ func (suite *UpgradeTestSuite) TestPrepareDebugFlag() {
suite.Equal(want, stderr.String())
suite.Equal("", stdout.String())
}
func (suite *UpgradeTestSuite) TestPrepareSkipCrdsFlag() {
defer suite.ctrl.Finish()
cfg := env.Config{
Chart: "at40",
Release: "cabbages_smell_great",
SkipCrds: true,
}
u := NewUpgrade(cfg)
command = func(path string, args ...string) cmd {
suite.Equal(helmBin, path)
suite.Equal([]string{"upgrade", "--install", "--skip-crds", "cabbages_smell_great", "at40"}, args)
return suite.mockCmd
}
suite.mockCmd.EXPECT().Stdout(gomock.Any())
suite.mockCmd.EXPECT().Stderr(gomock.Any())
err := u.Prepare()
suite.Require().Nil(err)
}