From 481e87f0c58a622edcca376a88956d40d497fa01 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BE=90=E8=81=AA?= <xucong.053@bytedance.com>
Date: Tue, 4 Jan 2022 21:38:58 +0800
Subject: [PATCH] feat: data-driven.

---
 examples/parameters_test.yaml |  3 +--
 parser.go                     | 19 +++++++++++--------
 2 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/examples/parameters_test.yaml b/examples/parameters_test.yaml
index 0eff1384..0f3d8c9d 100644
--- a/examples/parameters_test.yaml
+++ b/examples/parameters_test.yaml
@@ -4,8 +4,7 @@ config:
     user_agent: [ "iOS/10.1", "iOS/10.2" ]
     username-password: ${parameterize(examples/account.csv)}
   parameters_setting:
-    strategy: ""
-    iteration: 10
+    strategy: ["random", "sequential"]
   variables:
     app_version: v1
     user_agent: iOS/10.3
diff --git a/parser.go b/parser.go
index 0d5c70e6..7180b3f8 100644
--- a/parser.go
+++ b/parser.go
@@ -614,6 +614,7 @@ func initParameterIterator(cfg *TConfig, mode string) (err error) {
 	rawValue := reflect.ValueOf(cfg.ParametersSetting.Strategy)
 	switch rawValue.Kind() {
 	case reflect.Slice:
+		// strategy: ["random", "sequential"]
 		if len(parameters) != rawValue.Len() {
 			return errors.New("parameters and strategy should have the same length")
 		} else {
@@ -625,6 +626,7 @@ func initParameterIterator(cfg *TConfig, mode string) (err error) {
 			}
 		}
 	case reflect.String:
+		// strategy: "random"
 		if len(rawValue.String()) == 0 {
 			cfg.ParametersSetting.Strategy = strategySequential
 		} else {
@@ -635,6 +637,7 @@ func initParameterIterator(cfg *TConfig, mode string) (err error) {
 			newIterator(genCartesianProduct(parameters), cfg.ParametersSetting.Strategy.(string), cfg.ParametersSetting.Iteration),
 		)
 	default:
+		// default strategy: sequential
 		cfg.ParametersSetting.Strategy = strategySequential
 		cfg.ParametersSetting.Iterator = append(
 			cfg.ParametersSetting.Iterator,
@@ -645,14 +648,14 @@ func initParameterIterator(cfg *TConfig, mode string) (err error) {
 }
 
 func newIterator(parameters paramsType, strategy string, iteration int) *Iterator {
-	it := parameters.Iterator()
-	it.strategy = strategy
+	iter := parameters.Iterator()
+	iter.strategy = strategy
 	if iteration > 0 {
-		it.iteration = iteration
-	} else if it.iteration == 0 {
-		it.iteration = 1
-	} else {
-		it.iteration = -1
+		iter.iteration = iteration
+	} else if iteration < 0 {
+		iter.iteration = -1
+	} else if iter.iteration == 0 {
+		iter.iteration = 1
 	}
-	return it
+	return iter
 }