mux.Vars 无法从 httpTest 请求中检索 var

Question

我正在为未来的项目编写一个简单的 rest 样板，我目前正在为我的控制器进行一些测试，我正在尝试通过它的 Id 检索待办事项，这/todo/{id}是处理程序。func (t TodoController) GetById(w http.ResponseWriter, r *http.Request) {    params := mux.Vars(r)    id, err := strconv.Atoi(params["id"])    if err != nil {        w.WriteHeader(http.StatusBadRequest)        return    }    todo, err := t.todoService.GetById(id)    if err != nil {        w.WriteHeader(http.StatusNotFound)        return    }    helpers.SendResponse(http.StatusOK, todo, w)}这是此控制器的测试。var (    todoController TodoController    recorder       *httptest.ResponseRecorder    todos          []models.Todo)func setup() {    todoController = *NewTodoController(services.NewTodoService())    recorder = httptest.NewRecorder()    todos = []models.Todo{        {            ID:        0,            Todo:      "Buy milk",            Completed: false,        },        {            ID:        1,            Todo:      "Buy cheese",            Completed: false,        },        {            ID:        2,            Todo:      "Buy eggs",            Completed: false,        },    }}func TestGetById(t *testing.T)  {    // Arrange    setup()    request := httptest.NewRequest(http.MethodGet, "/todo/1", nil)    var response models.Todo    // Act    todoController.GetById(recorder, request)    result := recorder.Result()    defer result.Body.Close()    data, err := ioutil.ReadAll(result.Body)    err = json.Unmarshal(data, &response)    // Assert    if err != nil {        t.Errorf("Expected error to be nil but got %v", err)    }    assert.Equal(t, result.StatusCode, http.StatusOK, "Response should have been 200 Ok")    assert.Equal(t, response, todos[1], "Response did not match the expected result")}看起来当向/todo/1mux 发送请求时无法检索 Id，因此它最终返回 BadRequest 错误。这是此回购协议的链接：https ://github.com/Je12emy/rest_boiler

Answer 1

Mux 提供了两种注入请求参数的方法——SetURLVars帮助器和使用mux.Router包装器而不是直接调用处理程序。

SetURLVars做的正是你想要的——将否则无法访问的参数注入到 HTTP 请求中。

这种方法很简单，但有一个问题。使用的 URL 和注入的参数不同步。

None 禁止开发人员编写此代码：

// we use 2 in request path

request := httptest.NewRequest(http.MethodGet, "/todo/2", nil)

// and we use 1 in request param

request = SetURLVars(request, map[string]string{"id":"1"})

这不是很干净的测试实践。

我们不测试路由中的 var 名称是否正确。我们可以item_id在 router 中使用而不是id并且测试没有捕获到它。

不仅如此，我们还可以在路由器定义中使用错误的路径并映射不同的处理程序。如果我们犯了那个错误，客户可以删除order而不是项目。todo

如果我们在测试中使用我们的生产路由器，这可以解决。

假设这是我们的生产代码：

func InitRouter(t TodoController) http.handler

    r := mux.NewRouter()

    r.HandleFunc("/todo/{id}", t.GetById).Methods(http.MethodGet)

    return r

在测试中，我们可以GetById通过我们在InitRouter函数中创建的路由器进行测试：

func TestGetById(t *testing.T)  {

    // Arrange

    setup()

    request := httptest.NewRequest(http.MethodGet, "/todo/1", nil)

    var response models.Todo

    // added setup

    sut := InitRouter(todoController)

    // Act

    // changed act - calling GetById through production router

    sut.ServeHTTP(recorder, request)

    // no chnages after that

    result := recorder.Result()

    defer result.Body.Close()

    ...

}