support running single test-package

tools/flakeguard/cmd/run_tests.go

@@ -15,13 +15,21 @@ var RunTestsCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		repoPath, _ := cmd.Flags().GetString("repo-path")
 		testPackagesJson, _ := cmd.Flags().GetString("test-packages-json")
+		testPackage, _ := cmd.Flags().GetString("test-package")
 		count, _ := cmd.Flags().GetInt("count")
 		useRace, _ := cmd.Flags().GetBool("race")
 		failFast, _ := cmd.Flags().GetBool("fail-fast")
 
 		var testPackages []string
-		if err := json.Unmarshal([]byte(testPackagesJson), &testPackages); err != nil {
-			fmt.Fprintf(os.Stderr, "Error decoding test packages JSON: %s\n", err)
+		if testPackagesJson != "" {
+			if err := json.Unmarshal([]byte(testPackagesJson), &testPackages); err != nil {
+				fmt.Fprintf(os.Stderr, "Error decoding test packages JSON: %s\n", err)
+				os.Exit(1)
+			}
+		} else if testPackage != "" {
+			testPackages = append(testPackages, testPackage)
+		} else {
+			fmt.Fprintf(os.Stderr, "Error: must specify either --test-packages-json or --test-package\n")
 			os.Exit(1)
 		}
 
@@ -43,6 +51,7 @@ var RunTestsCmd = &cobra.Command{
 func init() {
 	RunTestsCmd.Flags().StringP("repo-path", "r", ".", "Path to the Git repository")
 	RunTestsCmd.Flags().String("test-packages-json", "", "JSON-encoded string of test packages")
+	RunTestsCmd.Flags().String("test-package", "", "Single test package to run")
 	RunTestsCmd.Flags().IntP("count", "c", 1, "Number of times to run the tests")
 	RunTestsCmd.Flags().Bool("race", false, "Enable the race detector")
 	RunTestsCmd.Flags().Bool("fail-fast", false, "Stop on the first test failure")