httptest 响应重新记录保留旧值

Question

我有一个需要测试的函数：handlerAuthenticationfunc handlerAuthentication(c *gin.Context) {    session := Session.GetSession(c)    var login Login    err := c.BindJSON(&login)    if err != nil {        c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})        return    }    client, err := initClient(c, login)    fmt.Println("Error: ",err)    if err != nil {        fmt.Println("There's an error !")        c.JSON(http.StatusUnauthorized, gin.H{"error": ErrorWrongLogin})        return    }    err = (*client).Logout(c)    if err != nil {        return    }    session.Set("username", login.Username)    session.Set("password", login.Password)    err = session.Save()    if err != nil {        c.JSON(http.StatusInternalServerError, gin.H{"error": "an error occurred during the save of the session:" + err.Error()})        return    }    c.JSON(http.StatusOK, "Connected")}为此，我做了这个：func TestHandlerAuthentication(t *testing.T) {    UrlOdoo = "https://isi.nc"    resp := httptest.NewRecorder()    gin.SetMode(gin.TestMode)    c, r := gin.CreateTestContext(resp)    r.POST("/test", func(c *gin.Context) {        handlerAuthentication(c)    })

Answer 1

我需要为每个测试初始化一个新测试。为此，我将初始化移动到函数：httptest.ResponseRecordert.Run(name,func(t *testing.T)

func TestHandlerAuthentication(t *testing.T) {

    UrlOdoo = "https://isi.nc"

    ctrl := gomock.NewController(t)

    defer ctrl.Finish()

    Odoo = OdooRPC{createMockOdooClient}

    client = mock_odoorpc.NewMockOdooClient(ctrl)

    client.EXPECT().Authenticate(gomock.Any(), gomock.Any(), invalidUsername, invalidPassword).AnyTimes().Return(fmt.Errorf("invalid login"))

    client.EXPECT().Authenticate(gomock.Any(), gomock.Any(), validUsername, validPassword).AnyTimes().Return(nil)

    client.EXPECT().Logout(gomock.Any()).AnyTimes().Return(nil)

    session = mock_session.NewMockSession(ctrl)

    Session = SessionGetter{createMockSession}

    session.EXPECT().Set("username", validUsername).AnyTimes().Return()

    session.EXPECT().Set("password", validPassword).AnyTimes().Return()

    session.EXPECT().Save().AnyTimes().Return(nil)

    for name, test := range map[string]struct {

        input Login

        want  int

    }{

        "valid login": {

            input: Login{

                Username: validUsername,

                Password: validPassword,

            },

            want: 200,

        },

        "invalid login": {

            input: Login{

                Username: invalidUsername,

                Password: invalidPassword,

            },

            want: 401,

        },

    } {

        t.Run(name, func(t *testing.T) {

            resp := httptest.NewRecorder()

            gin.SetMode(gin.TestMode)

            c, r := gin.CreateTestContext(resp)

            r.POST("/test", func(c *gin.Context) {

                handlerAuthentication(c)

            })

            body, _ := json.Marshal(test.input)

            c.Request, _ = http.NewRequest(http.MethodPost, "/test", strings.NewReader(string(body)))

            r.ServeHTTP(resp, c.Request)

            assert.Equal(t, test.want, resp.Code)

        })

    }

}