Go测试中无需传递RNG如何实现math/rand的确定性结果？

核心思路 #

Go 1.20之后，全局math/rand包的默认生成器不再支持直接设置种子，但可以通过替换全局生成器的底层源或封装自定义全局rand包的方式，实现无需传递RNG的确定性测试，不用修改原有函数参数。

具体实现方案 #

方案1：用反射修改全局生成器的私有源 #

全局rand包的默认生成器是私有变量globalRand，内部包含src字段。通过反射可以直接替换这个字段为固定种子的源，让所有全局rand函数调用都使用确定性源。

示例代码：

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

func TestDeterministicRand(t *testing.T) {
    // 保存原有源，测试结束后恢复，避免影响其他测试用例
    randReaderVal := reflect.ValueOf(rand.Reader).Elem()
    originalSrc := randReaderVal.FieldByName("src")
    defer func() {
        randReaderVal.FieldByName("src").Set(originalSrc)
    }()

    // 设置固定种子的源
    fixedSrc := rand.NewSource(42)
    randReaderVal.FieldByName("src").Set(reflect.ValueOf(fixedSrc))

    // 此时全局rand函数会输出确定性结果
    result := rand.Intn(100)
    if result != 51 { // 42种子下Intn(100)的第一个结果为51
        t.Errorf("Expected 51, got %d", result)
    }
}

注意：该方法依赖Go标准库内部实现，若未来globalRand结构变动可能失效，但核心结构变动概率极低，适合测试场景快速解决问题。

方案2：封装自定义全局rand包（长期维护更稳定） #

如果担心反射依赖内部结构，可以封装自己的rand包，内部维护可设置种子的全局生成器，原有代码仅需替换导入路径，无需修改函数参数。

示例封装包（如myrand）：

// myrand/rand.go
package myrand

import "math/rand"

var globalRand = rand.New(rand.NewSource(0))

// 设置种子
func Seed(seed int64) {
    globalRand = rand.New(rand.NewSource(seed))
}

// 封装原有rand的函数，按需添加
func Intn(n int) int {
    return globalRand.Intn(n)
}

func Float64() float64 {
    return globalRand.Float64()
}

func Shuffle(n int, swap func(i, j int)) {
    globalRand.Shuffle(n, swap)
}

测试时直接调用种子设置：

import (
    "testing"
    "your-project-path/myrand"
)

func TestMyFunc(t *testing.T) {
    myrand.Seed(42)
    result := myrand.Intn(100)
    if result != 51 {
        t.Errorf("Expected 51, got %d", result)
    }
}

这种方式工作量可控，且不依赖标准库内部细节，长期维护更稳定。

为什么直接rand.New没用？ #

你之前尝试的rand.New(rand.NewSource(42))只是创建了一个本地生成器，但全局rand包的默认生成器仍使用未固定种子的源，全局函数调用不会关联这个新生成器，因此无法得到确定性结果。

内容的提问来源于stack exchange，提问作者Gal Somekh