mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-09-24 10:15:47 +00:00
68092f27dd
* Move `file`, `ssh` and `user` packages to internal * Rename `SSHKey` to `(ssh.)UserKey` * Rename KeyValue / Publickey to PublicKey * Rename SSH key file from "debugd" to "ssh-keys" * Add CreateSSHUsers function to Core * Call CreateSSHUsers users on first control-plane node, when defined in config Tests: * Make StubUserCreator add entries to /etc/passwd * Add NewLinuxUserManagerFake for unit tests * Add unit tests & adjust existing ones to changes
71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"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"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func defaultConfigAsYAML(t *testing.T) string {
|
|
var readBuffer bytes.Buffer
|
|
require.NoError(t, yaml.NewEncoder(&readBuffer).Encode(config.Default()))
|
|
return readBuffer.String()
|
|
}
|
|
|
|
func TestConfigGenerateDefault(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
cmd := newConfigGenerateCmd()
|
|
|
|
require.NoError(configGenerate(cmd, fileHandler))
|
|
|
|
readYAML, err := fileHandler.Read(constants.ConfigFilename)
|
|
assert.NoError(err)
|
|
assert.Equal(defaultConfigAsYAML(t), string(readYAML))
|
|
}
|
|
|
|
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()
|
|
|
|
require.Error(configGenerate(cmd, fileHandler))
|
|
}
|
|
|
|
func TestConfigGenerateFileFlagRemoved(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
fileHandler := file.NewHandler(afero.NewMemMapFs())
|
|
cmd := newConfigGenerateCmd()
|
|
cmd.ResetFlags()
|
|
|
|
require.Error(configGenerate(cmd, fileHandler))
|
|
}
|
|
|
|
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", "-"))
|
|
|
|
require.NoError(configGenerate(cmd, fileHandler))
|
|
|
|
assert.Equal(defaultConfigAsYAML(t), outBuffer.String())
|
|
}
|