From 7d8e3706d180b6259c63d80a8e216cf56c698ee7 Mon Sep 17 00:00:00 2001
From: JaySon-Huang <tshent@qq.com>
Date: Fri, 3 Dec 2021 02:08:41 +0800
Subject: [PATCH] Add test cases

Signed-off-by: JaySon-Huang <tshent@qq.com>
---
 command_test.go | 90 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)

diff --git a/command_test.go b/command_test.go
index 583cb023..5b2da63a 100644
--- a/command_test.go
+++ b/command_test.go
@@ -2058,3 +2058,93 @@ func TestFParseErrWhitelistSiblingCommand(t *testing.T) {
 	}
 	checkStringContains(t, output, "unknown flag: --unknown")
 }
+
+func TestFCommandFindSubCommand(t *testing.T) {
+	root := &Command{
+		Use: "root",
+		Run: emptyRun,
+	}
+	type testOpts struct {
+		a bool
+		b bool
+	}
+	var opt testOpts
+	child := &Command{
+		Use: "child",
+		RunE: func(cmd *Command, args []string) error {
+			fmt.Printf("Done! a=%v b=%v\n", opt.a, opt.b)
+			return nil
+		},
+	}
+	child.Flags().BoolVar(&opt.a, "a", false, "a")
+	child.Flags().BoolVar(&opt.b, "b", false, "b")
+	flag := child.Flags().Lookup("a")
+	if flag != nil {
+		if flag.NoOptDefVal != "true" {
+			t.Errorf("ffff %v", flag.NoOptDefVal)
+		}
+	}
+	root.AddCommand(child)
+
+	argsToTest := [][]string{
+		{"--a", "--b", "child"},
+		{"child", "--a"},
+		{"child", "--b"},
+		{"--a=false", "child"},
+		{"--a=true", "child"},
+		{"--a", "child"},
+		{"--b", "child"},
+	}
+	for _, args := range argsToTest {
+		cmdFound, _, err := root.Find(args)
+		if err != nil {
+			t.Errorf("Unexpected error: %v with args: %v", err, args)
+		}
+		if cmdFound.Name() != "child" {
+			t.Errorf("expected found 'child' command with args: %v", args)
+		}
+	}
+}
+
+func TestFFindSubCommand(t *testing.T) {
+	root := &Command{
+		Use: "root",
+		Run: emptyRun,
+	}
+	type testOpts struct {
+		a bool
+		b bool
+	}
+	var opt testOpts
+	child := &Command{
+		Use: "child",
+		RunE: func(cmd *Command, args []string) error {
+			fmt.Printf("Done! a=%v b=%v\n", opt.a, opt.b)
+			return nil
+		},
+	}
+	child.Flags().BoolVar(&opt.a, "a", false, "a")
+	child.Flags().BoolVar(&opt.b, "b", false, "b")
+	root.AddCommand(child)
+
+	// ok
+	output, err := executeCommand(root, "--a", "child")
+	if err == nil {
+		t.Error("expected unknown flag error")
+	}
+	checkStringContains(t, output, "unknown flag: --a")
+
+	// ok
+	output, err = executeCommand(root, "--b", "child")
+	if err == nil {
+		t.Error("expected unknown flag error")
+	}
+	checkStringContains(t, output, "unknown flag: --b")
+
+	// fail?
+	output, err = executeCommand(root, "--a", "--b", "child")
+	if err == nil {
+		t.Error("expected unknown flag error")
+	}
+	checkStringContains(t, output, "unknown flag: --a")
+}