mirror of
https://github.com/edgelesssys/constellation.git
synced 2025-05-04 23:35:11 -04:00
Move cli/cmd into cli/internal
This commit is contained in:
parent
d71e97a940
commit
c3ebd3d3cd
34 changed files with 45 additions and 32 deletions
63
cli/internal/cmd/userinteraction_test.go
Normal file
63
cli/internal/cmd/userinteraction_test.go
Normal file
|
@ -0,0 +1,63 @@
|
|||
package cmd
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"io"
|
||||
"testing"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestAskToConfirm(t *testing.T) {
|
||||
// errAborted is an error where the user aborted the action.
|
||||
errAborted := errors.New("user aborted")
|
||||
|
||||
cmd := &cobra.Command{
|
||||
Use: "test",
|
||||
Args: cobra.NoArgs,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
ok, err := askToConfirm(cmd, "777")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !ok {
|
||||
return errAborted
|
||||
}
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
testCases := map[string]struct {
|
||||
input string
|
||||
wantErr error
|
||||
}{
|
||||
"user confirms": {"y\n", nil},
|
||||
"user confirms long": {"yes\n", nil},
|
||||
"user disagrees": {"n\n", errAborted},
|
||||
"user disagrees long": {"no\n", errAborted},
|
||||
"user is first unsure, but agrees": {"what?\ny\n", nil},
|
||||
"user is first unsure, but disagrees": {"wait.\nn\n", errAborted},
|
||||
"repeated invalid input": {"h\nb\nq\n", ErrInvalidInput},
|
||||
}
|
||||
|
||||
for name, tc := range testCases {
|
||||
t.Run(name, func(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
out := &bytes.Buffer{}
|
||||
cmd.SetOut(out)
|
||||
cmd.SetErr(&bytes.Buffer{})
|
||||
in := bytes.NewBufferString(tc.input)
|
||||
cmd.SetIn(in)
|
||||
|
||||
err := cmd.Execute()
|
||||
assert.ErrorIs(err, tc.wantErr)
|
||||
|
||||
output, err := io.ReadAll(out)
|
||||
assert.NoError(err)
|
||||
assert.Contains(string(output), "777")
|
||||
})
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue