fix handling flag without '='

task.go

@@ -45,8 +45,9 @@ func RunTasks(names ...string) {
 }
 
 // RunTasksFromArgs reads the command line and runs each named task.
-// A task name is an argument that does not start with a dash "-"
-// Make sure any flags are on the commandline are present before the task names.
+// A task name is an argument that does not start with a dash "-".
+// Flags can be set before or after the task names.
+// Boolean flags must have the form `-v=true`.
 func RunTasksFromArgs() {
 	defer deferList.run()
 	if len(os.Args) == 1 {
@@ -57,8 +58,12 @@ func RunTasksFromArgs() {
 	}
 	for i := 1; i < len(os.Args); i++ {
 		each := os.Args[i]
-		if !strings.HasPrefix(each, "-") {
-			RunTasks(os.Args[i])
+		if strings.HasPrefix(each, "-") {
+			if !strings.Contains(each, "=") {
+				i++ // next can be flag argument or a bool flag
+			}
+		} else {
+			RunTasks(each)
 		}
 	}
 }

task_test.go

@@ -0,0 +1,76 @@
+package muts
+
+import (
+	"os"
+	"testing"
+)
+
+func TestRunTasksFromArgs(t *testing.T) {
+	testHasRun := false
+	Task("test", func() {
+		testHasRun = true
+	})
+	os.Args = []string{"muts", "test"}
+	RunTasksFromArgs()
+	if !testHasRun {
+		t.Fail()
+	}
+}
+func TestRunTasksFromArgsWithFlag(t *testing.T) {
+	testHasRun := false
+	Task("test", func() {
+		testHasRun = true
+	})
+	os.Args = []string{"muts", "-flag", "flag", "test"}
+	RunTasksFromArgs()
+	if !testHasRun {
+		t.Fail()
+	}
+}
+func TestRunTasksFromArgsWithFlagAssign(t *testing.T) {
+	testHasRun := false
+	Task("test", func() {
+		testHasRun = true
+	})
+	os.Args = []string{"muts", "-flag=flag", "test"}
+	RunTasksFromArgs()
+	if !testHasRun {
+		t.Fail()
+	}
+}
+
+func TestRunTasksFromArgsWithFlagAssignWithEquals(t *testing.T) {
+	testHasRun := false
+	Task("test", func() {
+		testHasRun = true
+	})
+	os.Args = []string{"muts", "-flag=flag=1", "test"}
+	RunTasksFromArgs()
+	if !testHasRun {
+		t.Fail()
+	}
+}
+
+func TestRunTasksFromArgsWithFlagAtEnd(t *testing.T) {
+	testHasRun := false
+	Task("test", func() {
+		testHasRun = true
+	})
+	os.Args = []string{"muts", "test", "-flag", "flag"}
+	RunTasksFromArgs()
+	if !testHasRun {
+		t.Fail()
+	}
+}
+
+func TestRunTasksFromArgsWithBoolFlag(t *testing.T) {
+	testHasRun := false
+	Task("test", func() {
+		testHasRun = true
+	})
+	os.Args = []string{"muts", "-v=true", "test"}
+	RunTasksFromArgs()
+	if !testHasRun {
+		t.Fail()
+	}
+}