test: add the field to http header of existing tests

Signed-off-by: 170210 <j170210@icloud.com>
Finschia · Jul 21, 2023 · 9dac9e3 · 9dac9e3
1 parent 535d824
commit 9dac9e3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/rpc/jsonrpc/server/common_test.go b/rpc/jsonrpc/server/common_test.go
@@ -28,6 +28,8 @@ var (
 	TestFuncMap   = map[string]*RPCFunc{"c": TestRPCFunc}
 	TestGoodBody  = `{"jsonrpc": "2.0", "method": "c", "id": "0", "params": null}`
 	TestBadParams = `{"jsonrpc": "2.0", "method": "c", "id": "0", "params": "s=a,i=b"}`
+
+	TestMaxRequestBatchRequest = "10"
 )
 
 type FailManager struct {

diff --git a/rpc/jsonrpc/server/http_json_handler_test.go b/rpc/jsonrpc/server/http_json_handler_test.go
@@ -61,6 +61,7 @@ func TestRPCParams(t *testing.T) {
 
 	for i, tt := range tests {
 		req, _ := http.NewRequest("POST", "http://localhost/", strings.NewReader(tt.payload))
+		req.Header.Set("MaxRequestBatchRequest", TestMaxRequestBatchRequest)
 		rec := httptest.NewRecorder()
 		mux.ServeHTTP(rec, req)
 		res := rec.Result()
@@ -109,6 +110,7 @@ func TestJSONRPCID(t *testing.T) {
 
 	for i, tt := range tests {
 		req, _ := http.NewRequest("POST", "http://localhost/", strings.NewReader(tt.payload))
+		req.Header.Set("MaxRequestBatchRequest", TestMaxRequestBatchRequest)
 		rec := httptest.NewRecorder()
 		mux.ServeHTTP(rec, req)
 		res := rec.Result()
@@ -138,6 +140,7 @@ func TestRPCNotification(t *testing.T) {
 	mux := testMux()
 	body := strings.NewReader(`{"jsonrpc": "2.0"}`)
 	req, _ := http.NewRequest("POST", "http://localhost/", body)
+	req.Header.Set("MaxRequestBatchRequest", TestMaxRequestBatchRequest)
 	rec := httptest.NewRecorder()
 	mux.ServeHTTP(rec, req)
 	res := rec.Result()
@@ -175,6 +178,7 @@ func TestRPCNotificationInBatch(t *testing.T) {
 	}
 	for i, tt := range tests {
 		req, _ := http.NewRequest("POST", "http://localhost/", strings.NewReader(tt.payload))
+		req.Header.Set("MaxRequestBatchRequest", TestMaxRequestBatchRequest)
 		rec := httptest.NewRecorder()
 		mux.ServeHTTP(rec, req)
 		res := rec.Result()
@@ -230,7 +234,7 @@ func TestTooManyRPCNotificationInBatch_error(t *testing.T) {
 	// execute the batch request
 	mux := testMux()
 	req, _ := http.NewRequest("POST", "http://localhost/", strings.NewReader(string(jsonData)))
-	req.Header.Set("MaxRequestBatchRequest", "10")
+	req.Header.Set("MaxRequestBatchRequest", TestMaxRequestBatchRequest)
 	rec := httptest.NewRecorder()
 	mux.ServeHTTP(rec, req)
 	res := rec.Result()
@@ -280,6 +284,7 @@ func TestMakeJSONRPCHandler_Unmarshal_WriteRPCResponseHTTPError_error(t *testing
 func TestMakeJSONRPCHandler_last_WriteRPCResponseHTTP_error(t *testing.T) {
 	handlerFunc := makeJSONRPCHandler(TestFuncMap, log.TestingLogger())
 	req, _ := http.NewRequest("GET", "http://localhost/", strings.NewReader(TestGoodBody))
+	req.Header.Set("MaxRequestBatchRequest", TestMaxRequestBatchRequest)
 	// WriteRPCResponseHTTP error
 	rec := NewFailedWriteResponseWriter()
 	handlerFunc.ServeHTTP(rec, req)