Mooncake - 一个简单易用的Golang Mock测试库

Mooncake是一个轻量级的Golang Mock测试库，它可以帮助开发者快速生成各种测试场景所需的Mock数据。下面我将介绍Mooncake的基本用法和示例代码。

安装Mooncake

go get github.com/smartystreets/mooncake

核心功能

Mooncake提供了以下几种Mock能力：

1. 随机数据生成
2. 结构化数据Mock
3. HTTP请求/响应Mock
4. 数据库操作Mock

基本使用示例

1. 随机数据生成

package main

import (
	"fmt"
	"github.com/smartystreets/mooncake"
)

func main() {
	m := mooncake.New()

	// 生成随机字符串
	fmt.Println(m.String(10)) // 生成10位随机字符串

	// 生成随机整数
	fmt.Println(m.Int(1, 100)) // 生成1-100之间的随机整数

	// 生成随机浮点数
	fmt.Println(m.Float(0, 1)) // 生成0-1之间的随机浮点数

	// 生成随机布尔值
	fmt.Println(m.Bool())

	// 生成随机时间
	fmt.Println(m.Time())
}

2. 结构化数据Mock

package main

import (
	"fmt"
	"github.com/smartystreets/mooncake"
)

type User struct {
	ID       int
	Name     string
	Email    string
	Age      int
	IsActive bool
}

func main() {
	m := mooncake.New()

	var user User
	m.Fill(&user) // 自动填充结构体

	fmt.Printf("%+v\n", user)
	// 输出示例: {ID:42 Name:JxZb Email:JxZb@example.com Age:28 IsActive:true}

	// 自定义填充规则
	m.WithRules(map[string]interface{}{
		"Name":  "固定名称",
		"Email": func() string { return "custom@email.com" },
	})

	var customUser User
	m.Fill(&customUser)
	fmt.Printf("%+v\n", customUser)
}

3. HTTP请求/响应Mock

package main

import (
	"fmt"
	"net/http"
	"net/http/httptest"
	"github.com/smartystreets/mooncake"
)

func main() {
	m := mooncake.New()

	// 创建Mock HTTP处理器
	handler := m.NewHTTPHandler()

	// 设置Mock路由和响应
	handler.GET("/users", func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
		w.Write([]byte(`[{"id":1,"name":"Mock User"}]`))
	})

	// 使用httptest测试
	server := httptest.NewServer(handler)
	defer server.Close()

	// 发起请求测试
	resp, err := http.Get(server.URL + "/users")
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()

	fmt.Println("Status:", resp.Status)
	// 输出: Status: 200 OK
}

4. 数据库操作Mock

package main

import (
	"database/sql"
	"fmt"
	"github.com/smartystreets/mooncake"
)

type UserRepository struct {
	db *sql.DB
}

func (r *UserRepository) GetUser(id int) (*User, error) {
	row := r.db.QueryRow("SELECT id, name, email FROM users WHERE id = ?", id)
	
	var user User
	err := row.Scan(&user.ID, &user.Name, &user.Email)
	if err != nil {
		return nil, err
	}
	
	return &user, nil
}

func main() {
	m := mooncake.New()

	// 创建Mock数据库
	mockDB := m.NewSQLMock()

	// 设置预期查询和返回结果
	mockDB.ExpectQuery("SELECT id, name, email FROM users WHERE id = ?").
		WithArgs(1).
		WillReturnRows(mockDB.NewRows([]string{"id", "name", "email"}).
			AddRow(1, "Mock User", "mock@example.com"))

	repo := &UserRepository{db: mockDB}
	user, err := repo.GetUser(1)
	if err != nil {
		panic(err)
	}

	fmt.Printf("%+v\n", user)
	// 输出: &{ID:1 Name:Mock User Email:mock@example.com Age:0 IsActive:false}
}

高级用法

自定义生成器

package main

import (
	"fmt"
	"github.com/smartystreets/mooncake"
)

func main() {
	m := mooncake.New()

	// 注册自定义生成器
	m.RegisterGenerator("phone", func() string {
		return "138" + m.String(8, mooncake.CharsetDigit)
	})

	// 使用自定义生成器
	fmt.Println(m.Generate("phone")) // 输出类似: 13812345678

	// 在结构体中使用
	type Contact struct {
		Phone string `mooncake:"phone"`
	}
	
	var c Contact
	m.Fill(&c)
	fmt.Println(c.Phone)
}

序列Mock

package main

import (
	"fmt"
	"github.com/smartystreets/mooncake"
)

func main() {
	m := mooncake.New()

	// 创建序列Mock
	seq := m.NewSequence().
		Push("first").
		Push("second").
		Push("third")

	// 按顺序返回值
	fmt.Println(seq.Next()) // first
	fmt.Println(seq.Next()) // second
	fmt.Println(seq.Next()) // third
}

总结

Mooncake提供了简单易用的Mock功能，特别适合以下场景：

1. 单元测试中需要模拟各种数据
2. 需要快速生成测试数据集
3. 需要模拟外部服务响应
4. 需要模拟数据库操作

通过合理使用Mooncake，可以显著提高测试代码的编写效率和质量。更多高级用法可以参考Mooncake的官方文档。