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

学习Golang要努力，但是不要急！今天的这篇文章《mux.Vars 无法从 httpTest 请求中检索 var》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

我正在为未来的项目编写一个简单的休息样板，我目前正在对我的控制器进行一些测试，我正在尝试通过它的 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/1 发送请求时，mux 无法检索 id，因此最终返回 badrequest 错误。

这是此存储库的链接：https://github.com/je12emy/rest_boiler

正确答案

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

SetURLVars 完全符合您的要求 - 将其他无法访问的参数注入到 http 请求中。

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

没有任何内容禁止开发人员编写此代码：

// 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"})

这不是非常干净的测试实践。

我们不测试路由中的变量名称是否正确。我们可以在路由器中使用 item_id 而不是 id 并且测试没有捕获到这一点。

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

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

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

func initrouter(t todocontroller) http.handler
    r := mux.newrouter()
    r.handlefunc("/todo/{id}", t.getbyid).methods(http.methodget)
    return r

在测试中，我们可以通过在initrouter函数中创建的路由器来测试getbyid：

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()

    ...
}

到这里，我们也就讲完了《mux.Vars 无法从 httpTest 请求中检索 var》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！