Go testing/quick 遇到不可生成的结构体时怎么提供自定义 Generator

用 testing/quick 做属性测试时，最容易卡在“这个结构体不能自动生成”。典型例子是 time.Time 或项目里封装了私有字段的值对象：业务代码可以正常接收它们，但 quick.Value 不能替你绕过不可导出的字段。

官方文档把 Generator 定义为“生成自身类型随机值”的接口，接口方法是 Generate(rand *rand.Rand, size int) reflect.Value。下面只围绕 Generator、不可直接构造的字段和 Config.Values 展开，不把问题扩展到 fuzzing。

为什么 quick 生成结构体时会卡住

quick.Check 会反复调用属性函数，并为每个参数生成任意值。没有自定义生成器时，quick.Value 会按类型反射构造；对于结构体，它要求字段可导出。私有字段不是“随机性不够”，而是反射层没有安全的通用写入入口。

因此，遇到失败时先看测试函数的参数类型：如果参数是自己的类型，可以给它实现 Generator；如果参数是 time.Time 这类外部类型，就不能在当前包给它新增方法，需要在本次检查的配置里提供值。

本地类型：实现 Generator 返回自己的类型

先看一个可以控制内部字段的测试类型。这里用定义类型包住时间值，生成器返回的仍然是 QuickTime，而不是底层的 time.Time。这个类型一致性是最容易写错的地方。

package quickdemo

import (
    "math/rand"
    "reflect"
    "testing"
    "testing/quick"
    "time"
)

// QuickTime 是测试专用类型，可以为它声明 Generator 方法。
type QuickTime time.Time

// Generate 生成 QuickTime，而不是直接返回 time.Time。
func (QuickTime) Generate(r *rand.Rand, size int) reflect.Value {
    // size 只是范围提示，这里把它限制在合理的测试窗口内。
    days := int64(size)
    if days  365 {
        days = 365
    }

    // 生成 Unix 时间，避免依赖 time.Time 的私有内部字段。
    seconds := r.Int63n(days * 24 * 60 * 60)
    value := QuickTime(time.Unix(seconds, 0).UTC())
    return reflect.ValueOf(value)
}

func TestQuickTime(t *testing.T) {
    property := func(value QuickTime) bool {
        // 这里测试的是业务不变量，示例只要求值落在测试窗口内。
        tm := time.Time(value)
        return tm.Year() >= 1970
    }

    // quick 会发现 QuickTime 实现了 Generator，并反复调用 Generate。
    if err := quick.Check(property, nil); err != nil {
        t.Fatal(err)
    }
}

这个实现有三个关键点：方法接收者必须是 QuickTime；返回值必须是 reflect.ValueOf(QuickTime(...))；随机值应满足属性测试的业务边界，而不是为了“看起来随机”覆盖无限范围。size 是提示值，可以使用，也可以在明确不需要时忽略。

外部类型：用 Config.Values 按参数位置注入

如果属性函数直接接收 time.Time，不能给标准库类型添加 Generate 方法。这时把输入生成逻辑放进 quick.Config.Values。它收到与属性函数参数对应的 []reflect.Value，需要为每个位置填入可赋给该参数的值。

func TestExternalTime(t *testing.T) {
    property := func(value time.Time) bool {
        // 业务断言只关心时间不早于 Unix 起点。
        return !value.Before(time.Unix(0, 0).UTC())
    }

    config := &quick.Config{
        Values: func(values []reflect.Value, r *rand.Rand) {
            // 这里的 values[0] 对应 property 的第一个参数。
            seconds := r.Int63n(30 * 24 * 60 * 60)
            values[0] = reflect.ValueOf(time.Unix(seconds, 0).UTC())
        },
    }

    // Values 只对这次 Check 生效，不改变 time.Time 的定义。
    if err := quick.Check(property, config); err != nil {
        t.Fatal(err)
    }
}

Config.Values 适合“参数本身不能扩展”的情况，也适合一次性为多个参数安排有关联的输入。例如属性函数有两个时间参数，可以在同一个回调里先生成起点，再让第二个参数落在起点之后。回调必须覆盖所有参数位置，否则测试可能拿到零值或不符合预期的默认值。

随机范围和失败复现怎么安排

自定义生成器解决的是“能不能生成”，不自动保证“生成的数据有意义”。建议先把范围缩到属性真正关心的边界，再用配置控制样本数量；如果问题需要复现，给 Config.Rand 传入固定种子的随机源。

source := rand.New(rand.NewSource(42)) // 固定种子，方便复现同一组输入。
config := &quick.Config{
    Rand:     source,
    MaxCount: 200, // 明确样本数量，避免默认次数掩盖测试成本。
    Values: func(values []reflect.Value, r *rand.Rand) {
        // 保持注入类型与属性函数参数一致。
        values[0] = reflect.ValueOf(time.Unix(r.Int63n(86400), 0).UTC())
    },
}

排查时可以按下面的清单判断该选哪条路径：

常见问题

Generator 方法必须返回指针吗？

不必须。返回值只要是实现该方法的类型即可；值接收者和指针接收者都要根据测试函数实际使用的类型选择，最稳妥的是让测试参数、方法接收者和 reflect.Value 的具体类型保持一致。

能不能在 Generate 里直接用 reflect 改私有字段？

不建议。Generator 的价值是由类型自己构造合法值，而不是绕过封装。若类型来自外部包，优先使用公开构造函数、定义一个测试适配类型，或改用 Config.Values 注入公开可构造的实例。

Config.Values 和 Generator 能同时使用吗？

可以，但对同一个参数通常只需要一种入口。Config.Values 在本次检查中直接提供参数切片；只有没有被覆盖的位置才需要依赖默认的 quick.Value 生成逻辑。

判断这类问题的核心不是“怎么让反射更强”，而是把生成责任放到正确的位置：本地类型用 Generator 表达自身的不变量，外部类型用 Config.Values 在测试边界注入样本。这样既避开不可导出字段，也让随机输入的范围和复现方式留在测试代码里。