-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
delete_test.go
70 lines (63 loc) · 1.9 KB
/
delete_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package delete
import (
"bytes"
"errors"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/spf13/cobra"
"github.com/weaveworks/eksctl/pkg/ctl/cmdutils"
)
var _ = Describe("delete", func() {
Describe("invalid-resource", func() {
It("with no flag", func() {
cmd := newDefaultCmd("invalid-resource")
_, err := cmd.execute()
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("Error: unknown command \"invalid-resource\" for \"delete\""))
Expect(err.Error()).To(ContainSubstring("usage"))
})
It("with invalid-resource and some flag", func() {
cmd := newDefaultCmd("invalid-resource", "--invalid-flag", "foo")
_, err := cmd.execute()
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("Error: unknown command \"invalid-resource\" for \"delete\""))
Expect(err.Error()).To(ContainSubstring("usage"))
})
It("with invalid-resource and additional argument", func() {
cmd := newDefaultCmd("invalid-resource", "foo")
_, err := cmd.execute()
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("Error: unknown command \"invalid-resource\" for \"delete\""))
Expect(err.Error()).To(ContainSubstring("usage"))
})
})
})
func newDefaultCmd(args ...string) *mockVerbCmd {
flagGrouping := cmdutils.NewGrouping()
cmd := Command(flagGrouping)
cmd.SetArgs(args)
return &mockVerbCmd{
parentCmd: cmd,
}
}
func newMockEmptyCmd(args ...string) *mockVerbCmd {
cmd := cmdutils.NewVerbCmd("delete", "Delete resource(s)", "")
cmd.SetArgs(args)
return &mockVerbCmd{
parentCmd: cmd,
}
}
type mockVerbCmd struct {
parentCmd *cobra.Command
}
func (c mockVerbCmd) execute() (string, error) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
c.parentCmd.SetOut(outBuf)
c.parentCmd.SetErr(errBuf)
err := c.parentCmd.Execute()
if err != nil {
err = errors.New(errBuf.String())
}
return outBuf.String(), err
}