mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
da7a870f54
The flag can be used to specify a Kubernetes version in format MAJOR.MINOR and let the CLI extend the value with the patch version.
142 lines
3.7 KiB
Go
142 lines
3.7 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/cloud/cloudprovider"
|
|
"github.com/edgelesssys/constellation/v2/internal/config"
|
|
"github.com/edgelesssys/constellation/v2/internal/constants"
|
|
"github.com/edgelesssys/constellation/v2/internal/file"
|
|
"github.com/edgelesssys/constellation/v2/internal/logger"
|
|
"github.com/edgelesssys/constellation/v2/internal/versions"
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/mod/semver"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func TestConfigGenerateKubernetesVersion(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
version string
|
|
wantErr bool
|
|
}{
|
|
"success": {
|
|
version: semver.MajorMinor(string(versions.Default)),
|
|
},
|
|
"no semver": {
|
|
version: "asdf",
|
|
wantErr: true,
|
|
},
|
|
"not supported": {
|
|
version: "1111",
|
|
wantErr: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
cmd := newConfigGenerateCmd()
|
|
err := cmd.Flags().Set("kubernetes", tc.version)
|
|
require.NoError(err)
|
|
|
|
cg := &configGenerateCmd{log: logger.NewTest(t)}
|
|
err = cg.configGenerate(cmd, fileHandler, cloudprovider.Unknown)
|
|
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
return
|
|
}
|
|
assert.NoError(err)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestConfigGenerateDefault(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
cmd := newConfigGenerateCmd()
|
|
|
|
cg := &configGenerateCmd{log: logger.NewTest(t)}
|
|
require.NoError(cg.configGenerate(cmd, fileHandler, cloudprovider.Unknown))
|
|
|
|
var readConfig config.Config
|
|
err := fileHandler.ReadYAML(constants.ConfigFilename, &readConfig)
|
|
assert.NoError(err)
|
|
assert.Equal(*config.Default(), readConfig)
|
|
}
|
|
|
|
func TestConfigGenerateDefaultGCPSpecific(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
cmd := newConfigGenerateCmd()
|
|
|
|
wantConf := config.Default()
|
|
wantConf.RemoveProviderExcept(cloudprovider.GCP)
|
|
|
|
cg := &configGenerateCmd{log: logger.NewTest(t)}
|
|
require.NoError(cg.configGenerate(cmd, fileHandler, cloudprovider.GCP))
|
|
|
|
var readConfig config.Config
|
|
err := fileHandler.ReadYAML(constants.ConfigFilename, &readConfig)
|
|
assert.NoError(err)
|
|
assert.Equal(*wantConf, readConfig)
|
|
}
|
|
|
|
func TestConfigGenerateDefaultExists(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
require.NoError(fileHandler.Write(constants.ConfigFilename, []byte("foobar"), file.OptNone))
|
|
cmd := newConfigGenerateCmd()
|
|
|
|
cg := &configGenerateCmd{log: logger.NewTest(t)}
|
|
require.Error(cg.configGenerate(cmd, fileHandler, cloudprovider.Unknown))
|
|
}
|
|
|
|
func TestConfigGenerateFileFlagRemoved(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
cmd := newConfigGenerateCmd()
|
|
cmd.ResetFlags()
|
|
|
|
cg := &configGenerateCmd{log: logger.NewTest(t)}
|
|
require.Error(cg.configGenerate(cmd, fileHandler, cloudprovider.Unknown))
|
|
}
|
|
|
|
func TestConfigGenerateStdOut(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
|
|
var outBuffer bytes.Buffer
|
|
cmd := newConfigGenerateCmd()
|
|
cmd.SetOut(&outBuffer)
|
|
require.NoError(cmd.Flags().Set("file", "-"))
|
|
|
|
cg := &configGenerateCmd{log: logger.NewTest(t)}
|
|
require.NoError(cg.configGenerate(cmd, fileHandler, cloudprovider.Unknown))
|
|
|
|
var readConfig config.Config
|
|
require.NoError(yaml.NewDecoder(&outBuffer).Decode(&readConfig))
|
|
|
|
assert.Equal(*config.Default(), readConfig)
|
|
}
|