如何在Golang框架中编写可测试的代码？

在 golang 框架中编写可测试的代码至关重要，可通过以下最佳实践实现：使用接口定义行为约定，以便轻松模拟依赖项。采用依赖注入模式，允许注入依赖项以方便测试。利用模拟技术创建假依赖项实例，以控制其行为并测试特定交互。

如何在 Golang 框架中编写可测试的代码

在 Go 框架中编写可测试的代码对于确保代码的可靠性和可维护性至关重要。遵循一些简单的最佳实践，可以轻松地编写可测试的代码。

使用接口

立即学习“go语言免费学习笔记（深入）”；

接口是编写可测试代码的关键部分。它们允许您定义行为约定，而无需具体实现。通过使用接口，您可以轻松地模拟依赖项，而无需修改实际代码。

// 这是一个定义了一个 `GetUser` 方法的 `UserService` 接口。
type UserService interface {
    GetUser(id int64) (*User, error)
}

依赖注入

SiteDynamic企业网站管理系统1.7.7 标准版

SiteDynamic企业网站管理系统采用较为成熟的ASP+ACCESS编写，是迄今为止国内较先进的ASP语言企业网站管理系统。系统为企业级网站提供一个框架，能满足企业的基本应用，同时系统开放全部源码，用户可以根据自己的需求扩展出自己需求的模块，如：单页面、新闻、产品展示、下载、友情链接、电子商务、广告、会员、在线支付、人才招聘等。整套系统的设计构造，完全考虑大中小企业类网站的功能要求，网站的后台

下载

依赖注入是一种设计模式，允许您将依赖项注入函数或结构。这使您可以轻松地交换依赖项，以进行测试和其他目的。

// 这个函数使用 `UserService` 接口获取 `User`。
func GetUser(userId int64, userService UserService) (*User, error) {
    return userService.GetUser(userId)
}

模拟

模拟是一种创建依赖项假实例的技术，能够控制其行为。这对于测试函数或结构与依赖项交互的方式非常有用。

import "testing"

func TestGetUser(t *testing.T) {
    // 创建一个模拟的 `UserService`。
    userService := &UserServiceMock{}
    userService.On("GetUser").Return(&User{Name: "John"}, nil)

    // 使用模拟的 `UserService` 调用 `GetUser`。
    user, err := GetUser(1, userService)

    if err != nil {
        t.Errorf("GetUser() returned unexpected error: %v", err)
    }

    // 断言返回的用户与预期的一致。
    if user.Name != "John" {
        t.Errorf("GetUser() returned unexpected user: %v", user)
    }
}

实战案例

考虑一个简单的 Gin 路由，它接受一个 Post 请求，并使用 userService 从数据库中获取用户。

package main

import (
    "github.com/gin-gonic/gin"
)

type User struct {
    ID   int64
    Name string
}

type UserService interface {
    GetUser(id int64) (*User, error)
}

func main() {
    userService := &UserServiceImpl{}
    router := gin.Default()
    router.POST("/users/:id", getUser(userService))
}

func getUser(userService UserService) gin.HandlerFunc {
    return func(c *gin.Context) {
        id, _ := strconv.ParseInt(c.Param("id"), 10, 64)
        user, err := userService.GetUser(id)
        if err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
            return
        }
        c.JSON(http.StatusOK, user)
    }
}

我们可以使用上面讨论的技术测试这个路由：

import (
    "bytes"
    "encoding/json"
    "fmt"
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/gin-gonic/gin"
)

func TestGetUser(t *testing.T) {
    userService := &UserServiceMock{}
    userService.On("GetUser").Return(&User{ID: 1, Name: "John"}, nil)

    router := gin.Default()
    router.POST("/users/:id", getUser(userService))

    // 创建一个 HTTP 请求。
    url := fmt.Sprintf("http://localhost:8080/users/%d", 1)
    req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer([]byte("")))
    if err != nil {
        t.Fatal(err)
    }

    // 执行 HTTP 请求。
    rr := httptest.NewRecorder()
    router.ServeHTTP(rr, req)

    // 断言 HTTP 响应状态。
    if status := rr.Code; status != http.StatusOK {
        t.Errorf("Unexpected HTTP status code: %d", status)
    }

    // 解析 HTTP 响应正文。
    var user User
    if err := json.NewDecoder(rr.Body).Decode(&user); err != nil {
        t.Fatal(err)
    }

    // 断言返回的用户与预期的一致。
    if user.ID != 1 || user.Name != "John" {
        t.Errorf("Unexpected user: %v", user)
    }
}