Add args to Execute

Author: bep

cobrakai.go

@@ -11,8 +11,8 @@ type root struct {
 	c *Commandeer
 }
 
-// Execute executes the root command and returns the commandeer that was executed.
-func (r *root) Execute(ctx context.Context) (*Commandeer, error) {
+func (r *root) Execute(ctx context.Context, args []string) (*Commandeer, error) {
+	r.c.CobraCommand.SetArgs(args)
 	cobraCommand, err := r.c.CobraCommand.ExecuteContextC(ctx)
 	if err != nil {
 		return nil, err
@@ -69,8 +69,9 @@ func (c *Commandeer) compile() error {
 }
 
 // Executer is the execution entry point.
+// The args are usually filled with os.Args[1:].
 type Executer interface {
-	Execute(ctx context.Context) (*Commandeer, error)
+	Execute(ctx context.Context, args []string) (*Commandeer, error)
 }
 
 // Commander is the interface that must be implemented by all commands.

��cobrakai_test.go

@@ -3,7 +3,6 @@ package cobrakai_test
 import (
 	"context"
 	"fmt"
-	"os"
 	"testing"
 
 	ck "github.com/bep/cobrakai"
@@ -19,7 +18,6 @@ func TestCobraKai(t *testing.T) {
 		fooBazCommand = &testComand2{name: "foo_baz"}
 	)
 
-	os.Args = []string{"hugo", "foo", "--localFlagName", "foo_local", "--persistentFlagName", "foo_persistent"}
 	c := qt.New(t)
 	r, err := ck.R(
 		&testComand1{name: "hugo"},
@@ -33,17 +31,18 @@ func TestCobraKai(t *testing.T) {
 	// This can be anything, just used to make sure the same context is passed all the way.
 	type key string
 	ctx := context.WithValue(context.Background(), key("foo"), "bar")
-	cdeer, err := r.Execute(ctx)
+	args := []string{"foo", "--localFlagName", "foo_local", "--persistentFlagName", "foo_persistent"}
+	cdeer, err := r.Execute(ctx, args)
 	c.Assert(err, qt.IsNil)
 	c.Assert(cdeer.Command.Name(), qt.Equals, "foo")
 	tc := cdeer.Command.(*testComand1)
 	c.Assert(tc.ctx, qt.Equals, ctx)
 	c.Assert(tc.localFlagName, qt.Equals, "foo_local")
 	c.Assert(tc.persistentFlagName, qt.Equals, "foo_persistent")
 
-	os.Args = []string{"hugo", "foo", "foo_baz", "--localFlagName", "foo_local2", "--persistentFlagName", "foo_persistent2"}
+	args = []string{"foo", "foo_baz", "--localFlagName", "foo_local2", "--persistentFlagName", "foo_persistent2"}
 	ctx = context.WithValue(context.Background(), key("bar"), "baz")
-	cdeer2, err := r.Execute(ctx)
+	cdeer2, err := r.Execute(ctx, args)
 	c.Assert(err, qt.IsNil)
 	c.Assert(cdeer2.Command.Name(), qt.Equals, "foo_baz")
 	tc2 := cdeer2.Command.(*testComand2)