使用gofuzz生成随机测试数据

gofuzz是一个Go语言的库，用于自动生成随机测试数据，非常适合用于单元测试和模糊测试(fuzz testing)。下面我将详细介绍gofuzz的使用方法。

安装gofuzz

首先安装gofuzz库：

go get github.com/google/gofuzz

基本用法

1. 基本类型随机生成

package main

import (
	"fmt"
	"github.com/google/gofuzz"
)

func main() {
	var i int
	var s string
	var b bool

	f := gofuzz.New()

	f.Fuzz(&i)
	f.Fuzz(&s)
	f.Fuzz(&b)

	fmt.Printf("随机整数: %d\n", i)
	fmt.Printf("随机字符串: %s\n", s)
	fmt.Printf("随机布尔值: %t\n", b)
}

2. 结构体随机生成

package main

import (
	"fmt"
	"github.com/google/gofuzz"
	"time"
)

type User struct {
	ID        int
	Name      string
	Email     string
	CreatedAt time.Time
	IsActive  bool
}

func main() {
	var user User
	f := gofuzz.New().NilChance(0) // 设置nil几率为0，即所有字段都会填充

	f.Fuzz(&user)

	fmt.Printf("随机用户: %+v\n", user)
}

高级用法

1. 自定义随机规则

package main

import (
	"fmt"
	"github.com/google/gofuzz"
)

type Product struct {
	ID     int
	Name   string
	Price  float64
	InStock bool
}

func main() {
	var product Product
	f := gofuzz.New()

	// 自定义规则
	f.NumElements(5, 10) // 设置字符串长度范围
	f.Fuzz(&product)

	fmt.Printf("随机产品: %+v\n", product)
}

2. 切片和map的随机生成

package main

import (
	"fmt"
	"github.com/google/gofuzz"
)

func main() {
	var intSlice []int
	var stringMap map[string]int

	f := gofuzz.New().NilChance(0)

	// 设置切片元素数量范围
	f.NumElements(3, 7).Fuzz(&intSlice)
	
	// 设置map元素数量范围
	f.NumElements(2, 5).Fuzz(&stringMap)

	fmt.Printf("随机整数切片: %v\n", intSlice)
	fmt.Printf("随机map: %v\n", stringMap)
}

3. 结合测试使用

package user_test

import (
	"testing"
	"github.com/google/gofuzz"
	"your/package/path/user"
)

func TestUserValidation(t *testing.T) {
	f := gofuzz.New().NilChance(0)
	
	for i := 0; i < 100; i++ { // 运行100次随机测试
		var u user.User
		f.Fuzz(&u)
		
		err := u.Validate()
		if err != nil {
			t.Errorf("验证失败，用户: %+v, 错误: %v", u, err)
		}
	}
}

配置选项

gofuzz提供了一些配置选项来控制随机生成行为：

f := gofuzz.New().
	NilChance(0.1).          // 10%的几率为nil
	NumElements(5, 20).      // 切片/数组/字符串长度范围
	MaxDepth(5).             // 最大嵌套深度
	Funcs(funcs)             // 自定义函数

自定义生成器

你可以为特定类型定义自己的随机生成逻辑：

package main

import (
	"fmt"
	"github.com/google/gofuzz"
	"time"
)

type CustomType struct {
	Timestamp time.Time
}

func main() {
	var ct CustomType
	
	f := gofuzz.New().Funcs(
		func(t *time.Time, c gofuzz.Continue) {
			*t = time.Now().Add(time.Duration(c.Rand.Intn(1000)) * time.Hour)
		},
	)
	
	f.Fuzz(&ct)
	fmt.Printf("自定义时间: %v\n", ct.Timestamp)
}

注意事项

1. gofuzz生成的随机数据可能不总是有效的，需要根据业务逻辑进行验证
2. 对于复杂结构，可能需要自定义生成器以确保数据有效性
3. 随机测试不能替代正常的单元测试，应该结合使用

gofuzz是一个非常强大的工具，特别适合用于：

• 模糊测试
• 边界条件测试
• 数据库填充测试数据
• API压力测试

希望这些示例能帮助你开始使用gofuzz进行随机测试数据生成！