AB#2309 constellation upgrade execute (#2)

Signed-off-by: Daniel Weiße <dw@edgeless.systems>

cli/internal/cmd/upgradeexecute_test.go

@@ -0,0 +1,56 @@
+package cmd
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/edgelesssys/constellation/internal/config"
+	"github.com/edgelesssys/constellation/internal/constants"
+	"github.com/edgelesssys/constellation/internal/file"
+	"github.com/spf13/afero"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestUpgradeExecute(t *testing.T) {
+	testCases := map[string]struct {
+		upgrader stubUpgrader
+		wantErr  bool
+	}{
+		"success": {
+			upgrader: stubUpgrader{},
+		},
+		"upgrade error": {
+			upgrader: stubUpgrader{err: errors.New("error")},
+			wantErr:  true,
+		},
+	}
+
+	for name, tc := range testCases {
+		t.Run(name, func(t *testing.T) {
+			assert := assert.New(t)
+			require := require.New(t)
+			cmd := newUpgradeExecuteCmd()
+			cmd.Flags().String("config", constants.ConfigFilename, "") // register persistent flag manually
+
+			handler := file.NewHandler(afero.NewMemMapFs())
+			require.NoError(handler.WriteYAML(constants.ConfigFilename, config.Default()))
+
+			err := upgradeExecute(cmd, tc.upgrader, handler)
+			if tc.wantErr {
+				assert.Error(err)
+			} else {
+				assert.NoError(err)
+			}
+		})
+	}
+}
+
+type stubUpgrader struct {
+	err error
+}
+
+func (u stubUpgrader) Upgrade(context.Context, string, map[uint32][]byte) error {
+	return u.err
+}