Merge pull request #208 from ddosify/develop

Check status code for remote file on multipart
getanteon · May 29, 2023 · ddf38c3 · ddf38c3
2 parents 7d1bec4 + 23dfd5d
commit ddf38c3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/config/json.go b/config/json.go
@@ -443,6 +443,12 @@ func prepareMultipartPayload(parts []multipartFormData) (body string, contentTyp
 					return "", "", err
 				}
 
+				if !(response.StatusCode >= 200 && response.StatusCode <= 299) {
+					multipartError.wrappedErr = fmt.Errorf("Multipart: request to remote url (%s) failed. Status code: %d", part.Value, response.StatusCode)
+					multipartError.msg = "Error while getting remote multipart file"
+					return "", "", multipartError
+				}
+
 				_, err = io.Copy(formPart, response.Body)
 				if err != nil {
 					multipartError.wrappedErr = err

diff --git a/main_benchmark_test.go b/main_benchmark_test.go
@@ -151,7 +151,7 @@ var table = []struct {
 	{
 		name:             "config_multipart_inject_1krps",
 		path:             "config/config_testdata/benchmark/config_multipart_inject_1krps.json",
-		cpuTimeThreshold: 180,
+		cpuTimeThreshold: 200,
 		maxMemThreshold:  10,
 		avgMemThreshold:  15,
 		testType:         Multipart,