126 lines
2.1 KiB
Go
126 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResolveCmdName(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
have *cobra.Command
|
|
expected string
|
|
}{
|
|
{
|
|
"ShouldResolveRootCmd",
|
|
newRootCmd(),
|
|
"authelia-gen",
|
|
},
|
|
{
|
|
"ShouldResolveDocsCmd",
|
|
newDocsCmd(),
|
|
"docs",
|
|
},
|
|
{
|
|
"ShouldResolveDocsSubCmd",
|
|
newRootCmd().Commands()[0].Commands()[0],
|
|
"code.keys",
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
assert.Equal(t, tc.expected, resolveCmdName(tc.have))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRootCmdGetArgs(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
have func() *cobra.Command
|
|
args []string
|
|
expected []string
|
|
}{
|
|
{
|
|
"ShouldReturnRootCmdArgs",
|
|
func() *cobra.Command {
|
|
cmd := newRootCmd()
|
|
|
|
cmd.SetArgs([]string{"a", "b"})
|
|
|
|
return cmd.Commands()[0]
|
|
},
|
|
[]string{"c", "d"},
|
|
[]string{"authelia-gen", "code", "c", "d"},
|
|
},
|
|
{
|
|
"ShouldReturnRootCmdWithoutArgs",
|
|
func() *cobra.Command {
|
|
cmd := newRootCmd()
|
|
|
|
return cmd.Commands()[0]
|
|
},
|
|
nil,
|
|
[]string{"authelia-gen", "code"},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
assert.Equal(t, tc.expected, rootCmdGetArgs(tc.have(), tc.args))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSortCmds(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
have *cobra.Command
|
|
expected []string
|
|
}{
|
|
{
|
|
"ShouldSortRootCmd",
|
|
newRootCmd(),
|
|
[]string{"code", "commit-lint", "github", "locales", "docs"},
|
|
},
|
|
{
|
|
"ShouldSortDocsCmd",
|
|
newDocsCmd(),
|
|
[]string{"cli", "data", "date"},
|
|
},
|
|
{
|
|
"ShouldSortGitHubCmd",
|
|
newGitHubCmd(),
|
|
[]string{"issue-templates"},
|
|
},
|
|
{
|
|
"ShouldSortLocalesCmd",
|
|
newLocalesCmd(),
|
|
nil,
|
|
},
|
|
{
|
|
"ShouldSortDocsDataCmd",
|
|
newDocsDataCmd(),
|
|
[]string{"keys", "misc"},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
actual := sortCmds(tc.have)
|
|
|
|
n := len(tc.expected)
|
|
|
|
require.Len(t, actual, n)
|
|
|
|
for i := 0; i < n; i++ {
|
|
assert.Equal(t, tc.expected[i], actual[i].Use)
|
|
}
|
|
})
|
|
}
|
|
}
|