Do should work (#35)

cristalhq · Aug 23, 2022 · e0b9131 · e0b9131
1 parent 378a78b
commit e0b9131
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/acmd.go b/acmd.go
@@ -58,6 +58,9 @@ type Command struct {
 // simple way to get exec function
 func (cmd *Command) getExec() func(ctx context.Context, args []string) error {
 	switch {
+	case cmd.Do != nil:
+		cmd.ExecFunc = cmd.Do
+		fallthrough
 	case cmd.ExecFunc != nil:
 		return cmd.ExecFunc
 	case cmd.Exec != nil:

diff --git a/acmd_test.go b/acmd_test.go
@@ -354,6 +354,28 @@ func TestCommand_IsHidden(t *testing.T) {
 	}
 }
 
+func TestDoShouldWork(t *testing.T) {
+	var ok bool
+
+	cmds := []Command{
+		{
+			Name: "foo",
+			Do: func(ctx context.Context, args []string) error {
+				ok = true
+				return nil
+			},
+		},
+	}
+
+	r := RunnerOf(cmds, Config{
+		Args:    []string{"./someapp", "foo"},
+		AppName: "myapp",
+	})
+
+	failIfErr(t, r.Run())
+	mustEqual(t, ok, true)
+}
+
 func TestExit(t *testing.T) {
 	wantStatus := 42
 	wantOutput := "myapp: code 42\n"