Golang随机生成测试数据和结构体的高效插件库Faker的使用

Faker是一个用于Go语言的随机假数据和结构体生成库，具有以下特点：

• 提供100多种生成器函数
• 支持结构体生成
• 支持唯一数据生成
• 内置类型支持
• 易于自定义
• 零依赖
• 递归无限循环检测器

安装

go get github.com/pioz/faker

基础使用示例

faker.SetSeed(623)

fmt.Println(faker.Username())
fmt.Println(faker.String())
fmt.Println(faker.IntInRange(1, 10))
fmt.Println(faker.CurrencyName())
// Output: spicule
// gzaazJyRGt3jDnVh6ik7R9FO0AU1HcOzdOXbmNVBRQ5pq8n9tHf9B21PIFozLEzsoY4wILvZjTxSLQmD3UOAamDgVR411T3YHleDTgLuz90XSO3NFZm1AnaJiJamVRcNGD2zmi4qWkcjKF3E4JKgn1DiCeC3eSb5WELsw8XqRzlvJqG
// 10
// Myanmar Kyat

结构体生成示例

faker.SetSeed(622)

// 定义一个新的生成器
colorBuilder := func(params ...string) (interface{}, error) {
  return faker.Pick("Red", "Yellow", "Blue", "Black", "White"), nil
}

// 注册一个名为"color"的字符串类型生成器
err := faker.RegisterBuilder("color", "string", colorBuilder)
if err != nil {
  panic(err)
}

type Animal struct {
  Name  string `faker:"username"`
  Color string `faker:"color"` // 使用自定义颜色生成器
}

type Person struct {
  FirstName string            `faker:"firstName"`         // 任何可用的函数(不区分大小写)
  LastName  *string           `faker:"lastName"`          // 指针也支持
  Age       int               `faker:"intinrange(0,120)"` // 可以带参数调用
  UUID      string            `faker:"uuid;unique"`       // 保证唯一值
  Number    int               `faker:"-"`                 // 跳过此字段
  Code      string            // 无标签则使用该字段类型的默认生成器
  Pet       Animal            // 递归填充此结构体
  Nicknames []string          `faker:"username;len=3"`          // 使用faker.Username函数生成大小为3的数组
  Extra     map[string]string `faker:"stringWithSize(3);len=2"` // 支持map
}

p := Person{}
err = faker.Build(&p)
if err != nil {
  panic(err)
}
fmt.Println(p.FirstName)
fmt.Println(*p.LastName)
fmt.Println(p.Age)
fmt.Println(p.UUID)
fmt.Println(p.Number)
fmt.Println(p.Code)
fmt.Println(p.Pet.Name)
fmt.Println(p.Pet.Color)
fmt.Println(len(p.Nicknames))
fmt.Println(p.Nicknames[0])
fmt.Println(p.Nicknames[1])
fmt.Println(p.Nicknames[2])
fmt.Println(p.Extra)
// Output: Wilber
// Gutkowski
// 25
// ff8d6917-b920-46e6-b1be-dc2d48becfcb
// 0
// z
// honegger
// Red
// 3
// teagan
// polypeptide
// chinfest
// map[70w:3F6 gQS:isq]

工厂模式使用

Faker也可以作为工厂库使用。当调用faker.Build函数时，如果值不为零则不会修改，保留原始值。

faker.SetSeed(623)

type User struct {
    Username string `faker:"username"`
    Email    string `faker:"email"`
    Country  string `faker:"CountryAlpha2"`
}

italianUserFactory := func() *User {
    u := &User{Country: "IT"}
    faker.Build(u)
    return u
}

italianUser := italianUserFactory()
fmt.Println(italianUser)
// Output: &{spicule hoag@ornamented.biz IT}

自定义生成器

生成器是一个可变参数函数，可以接受任意数量的字符串作为参数，并返回一个接口或错误。

faker.SetSeed(1802)

// 定义一个新的生成器
builder := func(params ...string) (interface{}, error) {
    if len(params) > 0 && params[0] == "melee" {
        return faker.Pick("Barbarian", "Bard", "Fighter", "Monk", "Paladin", "Rogue"), nil
    }
    return faker.Pick("Cleric", "Druid", "Ranger", "Sorcerer", "Warlock", "Wizard"), nil
}

// 注册一个名为"dndClass"的字符串类型生成器
err := faker.RegisterBuilder("dndClass", "string", builder)
if err != nil {
    panic(err)
}

player := &struct {
    Class string `faker:"dndClass(melee)"`
    // other fields ...
}{}

// 用假数据构建结构体
faker.Build(&player)

fmt.Println(player.Class)
// Output: Paladin

Faker是一个功能强大且灵活的库，可以大大简化测试数据的生成工作。通过自定义生成器和标签系统，可以轻松地为各种复杂结构生成随机数据。