Golang单元测试中模拟数据库操作示例-Golang-PHP中文网

Golang单元测试中模拟数据库操作示例

P粉602998670

发布： 2025-09-21 19:29:01

原创

634人浏览过

通过接口抽象和模拟实现，Go语言单元测试可避免直接操作数据库。首先定义UserDB接口规范数据库操作，UserService服务层依赖该接口实现业务逻辑；接着创建MockUserDB结构体模拟数据存储，实现相同接口；最后在测试中注入模拟对象，验证GetUserInfo和RegisterUser等方法行为，确保逻辑正确且不依赖真实数据库，提升测试速度与稳定性。

golang单元测试中模拟数据库操作示例

在Go语言的单元测试中，直接操作真实数据库会影响测试速度和稳定性。为了隔离依赖，通常使用模拟（mock）技术来替代真实的数据库操作。下面通过一个简单的用户服务示例，展示如何用 接口抽象 + 模拟实现 的方式，在单元测试中避免实际访问数据库。

1. 定义数据库接口

首先将数据库操作抽象为接口，便于替换实现：

type UserDB interface {
    GetUserByID(id int) (*User, error)
    CreateUser(name string, email string) error
}

type User struct {
    ID    int
    Name  string
    Email string
}

登录后复制

2. 实现业务逻辑服务

服务层依赖接口，不关心具体是真实数据库还是模拟数据：

type UserService struct {
    db UserDB
}

func NewUserService(db UserDB) *UserService {
    return &UserService{db: db}
}

func (s *UserService) GetUserInfo(id int) (string, error) {
    user, err := s.db.GetUserByID(id)
    if err != nil {
        return "", err
    }
    return "Name: " + user.Name + ", Email: " + user.Email, nil
}

func (s *UserService) RegisterUser(name, email string) error {
    if name == "" || email == "" {
        return fmt.Errorf("name and email required")
    }
    return s.db.CreateUser(name, email)
}

登录后复制

3. 编写模拟数据库

在测试包中创建一个模拟的数据库结构体，实现 UserDB 接口：

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

千面数字人

千面 Avatar 系列：音频转换让静图随声动起来，动作模仿让动漫复刻真人动作，操作简单，满足多元创意需求。

156

查看详情

type MockUserDB struct {
    users map[int]*User
}

func NewMockUserDB() *MockUserDB {
    return &MockUserDB{
        users: make(map[int]*User),
    }
}

func (m *MockUserDB) GetUserByID(id int) (*User, error) {
    user, exists := m.users[id]
    if !exists {
        return nil, fmt.Errorf("user not found")
    }
    return user, nil
}

func (m *MockUserDB) CreateUser(name, email string) error {
    for _, u := range m.users {
        if u.Email == email {
            return fmt.Errorf("email already exists")
        }
    }
    newID := len(m.users) + 1
    m.users[newID] = &User{ID: newID, Name: name, Email: email}
    return nil
}

登录后复制

4. 编写单元测试

使用模拟数据库进行测试，无需启动任何数据库服务：

func TestGetUserInfo(t *testing.T) {
    mockDB := NewMockUserDB()
    mockDB.users[1] = &User{ID: 1, Name: "Alice", Email: "alice@example.com"}

    service := NewUserService(mockDB)
    info, err := service.GetUserInfo(1)

    if err != nil {
        t.Fatalf("expected no error, got %v", err)
    }
    expected := "Name: Alice, Email: alice@example.com"
    if info != expected {
        t.Errorf("got %s, want %s", info, expected)
    }
}

func TestRegisterUser_InvalidInput(t *testing.T) {
    mockDB := NewMockUserDB()
    service := NewUserService(mockDB)

    err := service.RegisterUser("", "bob@example.com")
    if err == nil {
        t.Fatal("expected error for empty name")
    }
}

func TestRegisterUser_Success(t *testing.T) {
    mockDB := NewMockUserDB()
    service := NewUserService(mockDB)

    err := service.RegisterUser("Bob", "bob@example.com")
    if err != nil {
        t.Fatalf("expected no error, got %v", err)
    }

    // 验证用户是否被创建（可通过 mockDB 状态检查）
    user, err := mockDB.GetUserByID(1)
    if err != nil || user.Name != "Bob" {
        t.Error("user should have been created")
    }
}

登录后复制

这种方式的核心思想是：通过接口解耦，让测试可以注入模拟对象。不需要引入外部库如 sqlmock 或 testify/mock，适合中小型项目快速实现隔离测试。

基本上就这些。只要设计时注意依赖抽象，Go 的单元测试完全可以干净、高效地覆盖数据库相关逻辑。

以上就是Golang单元测试中模拟数据库操作示例的详细内容，更多请关注php中文网其它相关文章！