fix invalid capacity bug for distinct user mode

getanteon · May 11, 2023 · 1b01d60 · 1b01d60
1 parent f835fec
commit 1b01d60
Show file tree

Hide file tree

Showing 4 changed files with 69 additions and 2 deletions.
diff --git a/config/config_testdata/benchmark/config_distinct_user.json b/config/config_testdata/benchmark/config_distinct_user.json
@@ -0,0 +1,25 @@
+{
+    "iteration_count": 100,
+    "engine_mode": "ddosify",
+    "load_type": "linear",
+    "duration": 10,
+    "steps": [
+            {
+              "id": 1,
+              "url": "{{HTTPBIN}}/json",
+              "name": "JSON",
+              "method": "GET",
+              "others": {
+                "h2": false,
+                "disable-redirect": true,
+                "disable-compression": false
+              }         
+            }
+    ],
+    "output": "stdout",
+    "env":{
+        "HTTPBIN" : "https://httpbin.ddosify.com"
+    },
+    "debug" : false,
+    "engine-mode": "distinct-user"
+}
diff --git a/config/config_testdata/benchmark/config_repeated_user.json b/config/config_testdata/benchmark/config_repeated_user.json
@@ -0,0 +1,25 @@
+{
+    "iteration_count": 100,
+    "engine_mode": "ddosify",
+    "load_type": "linear",
+    "duration": 10,
+    "steps": [
+            {
+              "id": 1,
+              "url": "{{HTTPBIN}}/json",
+              "name": "JSON",
+              "method": "GET",
+              "others": {
+                "h2": false,
+                "disable-redirect": true,
+                "disable-compression": false
+              }         
+            }
+    ],
+    "output": "stdout",
+    "env":{
+        "HTTPBIN" : "https://httpbin.ddosify.com"
+    },
+    "debug" : false,
+    "engine-mode": "distinct-user"
+}
diff --git a/core/scenario/service.go b/core/scenario/service.go
@@ -109,8 +109,8 @@ func (s *ScenarioService) Init(ctx context.Context, scenario types.Scenario,
 			initialCount = opts.MaxConcurrentIterCount
 			maxCount = opts.MaxConcurrentIterCount
 		} else if s.engineMode == types.EngineModeDistinctUser {
-			initialCount = opts.IterationCount
-			maxCount = opts.MaxConcurrentIterCount
+			initialCount = opts.MaxConcurrentIterCount
+			maxCount = opts.IterationCount
 		}
 		s.cPool, err = NewClientPool(initialCount, maxCount, s.engineMode, putInitialCookiesInJarFactory(s.engineMode, opts.InitialCookies), func(c *http.Client) { c.CloseIdleConnections() })
 	}

diff --git a/main_benchmark_test.go b/main_benchmark_test.go
@@ -47,6 +47,7 @@ type TestType string
 const (
 	Multipart   TestType = "multipart"
 	Correlation TestType = "correlation"
+	Basic       TestType = "basic"
 )
 
 var table = []struct {
@@ -59,6 +60,22 @@ var table = []struct {
 
 	testType TestType
 }{
+	{
+		name:             "config_distinct_user",
+		path:             "config/config_testdata/benchmark/config_distinct_user.json",
+		cpuTimeThreshold: 0.350,
+		maxMemThreshold:  1,
+		avgMemThreshold:  1,
+		testType:         Basic,
+	},
+	{
+		name:             "config_repeated_user",
+		path:             "config/config_testdata/benchmark/config_repeated_user.json",
+		cpuTimeThreshold: 0.350,
+		maxMemThreshold:  1,
+		avgMemThreshold:  1,
+		testType:         Basic,
+	},
 	{
 		name:             "config_correlation_load_1",
 		path:             "config/config_testdata/benchmark/config_correlation_load_1.json",