mirror of
https://github.com/Luzifer/ansible-role-version.git
synced 2024-12-24 11:31:21 +00:00
145 lines
4.6 KiB
Go
145 lines
4.6 KiB
Go
package doc
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var flagb1, flagb2, flagb3, flagbr, flagbp bool
|
|
var flags1, flags2a, flags2b, flags3 string
|
|
var flagi1, flagi2, flagi3, flagir int
|
|
|
|
const strtwoParentHelp = "help message for parent flag strtwo"
|
|
const strtwoChildHelp = "help message for child flag strtwo"
|
|
|
|
var cmdEcho = &cobra.Command{
|
|
Use: "echo [string to echo]",
|
|
Aliases: []string{"say"},
|
|
Short: "Echo anything to the screen",
|
|
Long: `an utterly useless command for testing.`,
|
|
Example: "Just run cobra-test echo",
|
|
}
|
|
|
|
var cmdEchoSub = &cobra.Command{
|
|
Use: "echosub [string to print]",
|
|
Short: "second sub command for echo",
|
|
Long: `an absolutely utterly useless command for testing gendocs!.`,
|
|
Run: func(cmd *cobra.Command, args []string) {},
|
|
}
|
|
|
|
var cmdDeprecated = &cobra.Command{
|
|
Use: "deprecated [can't do anything here]",
|
|
Short: "A command which is deprecated",
|
|
Long: `an absolutely utterly useless command for testing deprecation!.`,
|
|
Deprecated: "Please use echo instead",
|
|
}
|
|
|
|
var cmdTimes = &cobra.Command{
|
|
Use: "times [# times] [string to echo]",
|
|
SuggestFor: []string{"counts"},
|
|
Short: "Echo anything to the screen more times",
|
|
Long: `a slightly useless command for testing.`,
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {},
|
|
Run: func(cmd *cobra.Command, args []string) {},
|
|
}
|
|
|
|
var cmdPrint = &cobra.Command{
|
|
Use: "print [string to print]",
|
|
Short: "Print anything to the screen",
|
|
Long: `an absolutely utterly useless command for testing.`,
|
|
}
|
|
|
|
var cmdRootNoRun = &cobra.Command{
|
|
Use: "cobra-test",
|
|
Short: "The root can run its own function",
|
|
Long: "The root description for help",
|
|
}
|
|
|
|
var cmdRootSameName = &cobra.Command{
|
|
Use: "print",
|
|
Short: "Root with the same name as a subcommand",
|
|
Long: "The root description for help",
|
|
}
|
|
|
|
var cmdRootWithRun = &cobra.Command{
|
|
Use: "cobra-test",
|
|
Short: "The root can run its own function",
|
|
Long: "The root description for help",
|
|
}
|
|
|
|
var cmdSubNoRun = &cobra.Command{
|
|
Use: "subnorun",
|
|
Short: "A subcommand without a Run function",
|
|
Long: "A long output about a subcommand without a Run function",
|
|
}
|
|
|
|
var cmdVersion1 = &cobra.Command{
|
|
Use: "version",
|
|
Short: "Print the version number",
|
|
Long: `First version of the version command`,
|
|
}
|
|
|
|
var cmdVersion2 = &cobra.Command{
|
|
Use: "version",
|
|
Short: "Print the version number",
|
|
Long: `Second version of the version command`,
|
|
}
|
|
|
|
func flagInit() {
|
|
cmdEcho.ResetFlags()
|
|
cmdPrint.ResetFlags()
|
|
cmdTimes.ResetFlags()
|
|
cmdRootNoRun.ResetFlags()
|
|
cmdRootSameName.ResetFlags()
|
|
cmdRootWithRun.ResetFlags()
|
|
cmdSubNoRun.ResetFlags()
|
|
cmdRootNoRun.PersistentFlags().StringVarP(&flags2a, "strtwo", "t", "two", strtwoParentHelp)
|
|
cmdEcho.Flags().IntVarP(&flagi1, "intone", "i", 123, "help message for flag intone")
|
|
cmdTimes.Flags().IntVarP(&flagi2, "inttwo", "j", 234, "help message for flag inttwo")
|
|
cmdPrint.Flags().IntVarP(&flagi3, "intthree", "i", 345, "help message for flag intthree")
|
|
cmdEcho.PersistentFlags().StringVarP(&flags1, "strone", "s", "one", "help message for flag strone")
|
|
cmdEcho.PersistentFlags().BoolVarP(&flagbp, "persistentbool", "p", false, "help message for flag persistentbool")
|
|
cmdTimes.PersistentFlags().StringVarP(&flags2b, "strtwo", "t", "2", strtwoChildHelp)
|
|
cmdPrint.PersistentFlags().StringVarP(&flags3, "strthree", "s", "three", "help message for flag strthree")
|
|
cmdEcho.Flags().BoolVarP(&flagb1, "boolone", "b", true, "help message for flag boolone")
|
|
cmdTimes.Flags().BoolVarP(&flagb2, "booltwo", "c", false, "help message for flag booltwo")
|
|
cmdPrint.Flags().BoolVarP(&flagb3, "boolthree", "b", true, "help message for flag boolthree")
|
|
cmdVersion1.ResetFlags()
|
|
cmdVersion2.ResetFlags()
|
|
}
|
|
|
|
func initializeWithRootCmd() *cobra.Command {
|
|
cmdRootWithRun.ResetCommands()
|
|
flagInit()
|
|
cmdRootWithRun.Flags().BoolVarP(&flagbr, "boolroot", "b", false, "help message for flag boolroot")
|
|
cmdRootWithRun.Flags().IntVarP(&flagir, "introot", "i", 321, "help message for flag introot")
|
|
return cmdRootWithRun
|
|
}
|
|
|
|
func checkStringContains(t *testing.T, found, expected string) {
|
|
if !strings.Contains(found, expected) {
|
|
logErr(t, found, expected)
|
|
}
|
|
}
|
|
|
|
func checkStringOmits(t *testing.T, found, expected string) {
|
|
if strings.Contains(found, expected) {
|
|
logErr(t, found, expected)
|
|
}
|
|
}
|
|
|
|
func logErr(t *testing.T, found, expected string) {
|
|
out := new(bytes.Buffer)
|
|
|
|
_, _, line, ok := runtime.Caller(2)
|
|
if ok {
|
|
fmt.Fprintf(out, "Line: %d ", line)
|
|
}
|
|
fmt.Fprintf(out, "Unexpected response.\nExpecting to contain: \n %q\nGot:\n %q\n", expected, found)
|
|
t.Errorf(out.String())
|
|
}
|