Function TestUpgradeRelease_RollbackOnFailure

pkg/action/upgrade_test.go:159–212

(t *testing.T)

}

func TestUpgradeRelease_RollbackOnFailure(t *testing.T) {
	is := assert.New(t)
	req := require.New(t)

	t.Run("rollback-on-failure rollback succeeds", func(t *testing.T) {
		upAction := upgradeAction(t)

		rel := releaseStub()
		rel.Name = "nuketown"
		rel.Info.Status = common.StatusDeployed
		require.NoError(t, upAction.cfg.Releases.Create(rel))

		failer := upAction.cfg.KubeClient.(*kubefake.FailingKubeClient)
		// We can't make Update error because then the rollback won't work
		failer.WatchUntilReadyError = errors.New("arming key removed")
		upAction.cfg.KubeClient = failer
		upAction.RollbackOnFailure = true
		vals := map[string]any{}

		resi, err := upAction.Run(rel.Name, buildChart(), vals)
		req.Error(err)
		is.Contains(err.Error(), "arming key removed")
		is.Contains(err.Error(), "rollback-on-failure")
		res, err := releaserToV1Release(resi)
		is.NoError(err)

		// Now make sure it is actually upgraded
		updatedResi, err := upAction.cfg.Releases.Get(res.Name, 3)
		is.NoError(err)
		updatedRes, err := releaserToV1Release(updatedResi)
		is.NoError(err)
		// Should have rolled back to the previous
		is.Equal(updatedRes.Info.Status, common.StatusDeployed)
	})

	t.Run("rollback-on-failure uninstall fails", func(t *testing.T) {
		upAction := upgradeAction(t)
		rel := releaseStub()
		rel.Name = "fallout"
		rel.Info.Status = common.StatusDeployed
		require.NoError(t, upAction.cfg.Releases.Create(rel))

		failer := upAction.cfg.KubeClient.(*kubefake.FailingKubeClient)
		failer.UpdateError = errors.New("update fail")
		upAction.cfg.KubeClient = failer
		upAction.RollbackOnFailure = true
		vals := map[string]any{}

		_, err := upAction.Run(rel.Name, buildChart(), vals)
		req.Error(err)
		is.Contains(err.Error(), "update fail")
		is.Contains(err.Error(), "an error occurred while rolling back the release")
	})
}

func TestUpgradeRelease_ReuseValues(t *testing.T) {
	is := assert.New(t)