如何为我的测试用例正确地在golang中创建模拟http.request?

时间:2019-03-30 00:13:10

标签: go routes

我想编写一个测试用例来验证我的参数解析器功能。 以下是我的模拟http.request

的示例代码
rawUrl := "http://localhost/search/content?query=test"

func createSearchRequest(rawUrl string) SearchRequest {
    api := NewWebService()

    req, err := http.NewRequest("POST", rawUrl, nil)
    if err != nil {
        logger.Fatal(err)
    }
    logger.Infof("%v", req)
    return api.searchRequest(req)
}

我的Web服务器使用 github.com/gorilla/mux 作为路由

router := mux.NewRouter()

router.HandleFunc("/search/{query_type}", searchApiService.Search).Methods("GET")

但是在我的测试用例中,我无法从模拟{query_type}中获得http.request

func (api WebService) searchRequest(req *http.Request){
    // skip ....

    vars := mux.Vars(req)
    queryType := vars["query_type"]
    logger.Infof("queryType:%v", queryType)

    //skip ....
}

如何在测试用例中获取多路复用器的path参数?

1 个答案:

答案 0 :(得分:1)

func TestMaincaller(t *testing.T) {
    r,_ := http.NewRequest("GET", "hello/1", nil)
    w := httptest.NewRecorder()
   //create a map of variable and set it into mux
    vars := map[string]string{
    "parameter_name": "parametervalue",
    }

   r = mux.SetURLVars(r, vars)
  callfun(w,r)
}