wip tags support

Author: flowerinthenight

main.go

@@ -43,6 +43,7 @@ var (
 
 	files []string
 	dir   string
+	tags  []string
 
 	slack   string
 	verbose bool
@@ -344,6 +345,7 @@ func init() {
 	rootcmd.PersistentFlags().StringVarP(&dir, "dir", "d", dir, "root directory for scenario file[s]")
 	rootcmd.PersistentFlags().StringVar(&slack, "report-slack", slack, "slack url for notification")
 	rootcmd.Flags().StringSliceVarP(&files, "scenarios", "s", files, "scenario file[s] to run, comma-separated, or multiple -s")
+	rootcmd.Flags().StringSliceVarP(&tags, "tags", "t", tags, "key=value labels in scenario files that are allowed to run, empty means all")
 	rootcmd.AddCommand(runCmd())
 }
 

scenario.go

@@ -40,6 +40,7 @@ type Run struct {
 
 // Scenario reprents a single scenario file to run.
 type Scenario struct {
+	Tags    map[string]string `yaml:"tags"`
 	Env     map[string]string `yaml:"env"`
 	Prepare string            `yaml:"prepare"`
 	Run     []Run             `yaml:"run"`
@@ -150,6 +151,28 @@ type doScenarioInput struct {
 	Verbose       bool
 }
 
+func isAllowed(s *Scenario) bool {
+	if len(tags) == 0 {
+		return true
+	}
+
+	var matched int
+	for _, t := range tags {
+		tt := strings.Split(t, "=")
+		if len(tt) != 2 {
+			continue
+		}
+
+		for k, v := range s.Tags {
+			if k == tt[0] && v == tt[1] {
+				matched++
+			}
+		}
+	}
+
+	return matched == len(tags)
+}
+
 func doScenario(in *doScenarioInput) error {
 	for _, f := range in.ScenarioFiles {
 		yml, err := ioutil.ReadFile(f)
@@ -163,6 +186,11 @@ func doScenario(in *doScenarioInput) error {
 			continue
 		}
 
+		if !isAllowed(&s) {
+			log.Printf("%v is not allowed by tags", f)
+			continue
+		}
+
 		s.me = &s    // self-reference for our LoggerReporter functions
 		s.input = in // our copy
 		log.Printf("scenario: %v", f)