如何在杜松子酒的测试上下文中设置主机?

How to set a host on gin's test context?

我想为控制器编写单元测试,但我一直收到运行时错误。我发现这是由于请求上缺少 HostClientIP() 方法和请求主体。如何在测试上下文中设置它们?

这是我到目前为止得到的。它在 Host: c.Request.Host.

行失败

控制器:

type loggingControllers struct {
    LoggingService services.InterfaceLogging
}

func (l *loggingControllers) RegisterError(c *gin.Context) {
    errorEvent := models.Error{
        Badges:    map[string]string{},
        Host:      c.Request.Host,
        ClientIP:  c.ClientIP(),
        UserAgent: c.Request.UserAgent(),
        Count:     1,
        Timestamp: time.Now().Unix(),
    }

    err := json.NewDecoder(c.Request.Body).Decode(&errorEvent)
    if err != nil {
        utils.RespondWithError(c, http.StatusInternalServerError, err.Error())
        return
    }

    go l.LoggingService.SaveError(errorEvent)

    utils.RespondWithSuccess(c)
}

func GetLoggingControllerMock() loggingControllers {
    loggingServiceMock := services.GetLoggingServiceMock()

    return loggingControllers{
        LoggingService: &loggingServiceMock,
    }
}

单元测试:

func TestLoggingControllers(t *testing.T) {
    loggingControllers := GetLoggingControllerMock()

    w := httptest.NewRecorder()
    c, _ := gin.CreateTestContext(w)

    loggingControllers.RegisterError(c)
}

错误信息:

--- FAIL: TestLoggingControllers (0.00s)
panic: runtime error: invalid memory address or nil pointer dereference [recovered]
    panic: runtime error: invalid memory address or nil pointer dereference
[signal SIGSEGV: segmentation violation code=0x1 addr=0x38 pc=0x15238ec]

感谢 Adrian 的评论,我找到了解决方案。

func TestLoggingControllers(t *testing.T) {
    loggingControllers := GetLoggingControllerMock()

    w := httptest.NewRecorder()
    c, r := gin.CreateTestContext(w)

    r.POST("/", loggingControllers.RegisterError)

    c.Request, _ = http.NewRequest(http.MethodPost, "/", bytes.NewBuffer([]byte("{}")))

    r.ServeHTTP(w, c.Request)

    if w.Code != http.StatusOK {
        t.Errorf("Expected status %d, got %d", http.StatusOK, w.Code)
    }
}