yjp 3 years ago
parent
commit
370829ec38
2 changed files with 8 additions and 8 deletions
  1. 4 4
      invoke/invoke.go
  2. 4 4
      test/invoke_test.go

+ 4 - 4
invoke/invoke.go

@@ -52,8 +52,8 @@ func (api *API) DeleteWithoutHeaders(methodName string) ([]byte, error) {
 	return api.Delete(methodName, nil)
 }
 
-func (api *API) GetWithoutHeaders(methodName string) ([]byte, error) {
-	return api.Get(methodName, nil)
+func (api *API) GetWithoutHeaders(methodName string, queryParams map[string]string) ([]byte, error) {
+	return api.Get(methodName, queryParams, nil)
 }
 
 func (api *API) Post(methodName string, headers map[string]string, data []byte) ([]byte, error) {
@@ -118,13 +118,13 @@ func (api *API) Put(methodName string, headers map[string]string, data []byte) (
 	return resp.Body(), nil
 }
 
-func (api *API) Get(methodName string, headers map[string]string) ([]byte, error) {
+func (api *API) Get(methodName string, queryParams map[string]string, headers map[string]string) ([]byte, error) {
 	invokeUrl, err := url.JoinPath(api.baseUrl, methodName)
 	if err != nil {
 		return nil, err
 	}
-
 	resp, err := api.client.R().
+		SetQueryParams(queryParams).
 		SetHeaders(headers).
 		Get(invokeUrl)
 	if err != nil {

+ 4 - 4
test/invoke_test.go

@@ -233,14 +233,14 @@ func TestGetWithoutHeaaders(t *testing.T) {
 	srv := initInvokeHttpServer(t, wgServer, wg)
 	wgServer.Wait()
 
-	response, err := api.GetWithoutHeaders(getJsonMethodName)
+	response, err := api.GetWithoutHeaders(getJsonMethodName, nil)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	assert.Equal(t, getJsonResponse, string(response), "响应不一致")
 
-	daprResponse, err := daprAPI.GetWithoutHeaders(getJsonMethodName)
+	daprResponse, err := daprAPI.GetWithoutHeaders(getJsonMethodName, nil)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -369,14 +369,14 @@ func TestGet(t *testing.T) {
 	srv := initInvokeHttpServer(t, wgServer, wg)
 	wgServer.Wait()
 
-	response, err := api.Get(getMethodName, contentTypeHeader)
+	response, err := api.Get(getMethodName, nil, contentTypeHeader)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	assert.Equal(t, getResponse, string(response), "响应不一致")
 
-	daprResponse, err := daprAPI.Get(getMethodName, contentTypeHeader)
+	daprResponse, err := daprAPI.Get(getMethodName, nil, contentTypeHeader)
 	if err != nil {
 		t.Fatal(err)
 	}