如何模拟 rand.Reader 的故障以进行测试？

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《如何模拟 rand.Reader 的故障以进行测试？》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我正在尝试为我的函数编写单元测试；请注意，它调用 rand.read （来自 crypto/rand 包），它依赖于 rand.reader：

func GenerateBytes(length int) ([]byte, error) {
    bytes := make([]byte, length)
    if _, err := rand.Read(bytes); err != nil {
        return nil, err
    }
    return bytes, nil
}

我能够为快乐路径编写测试，但无法测试 rand.read 返回错误的情况...有没有办法模拟 rand.read 的失败？

正确答案

您到底想检查什么？那么，如果 rand.read 返回非零错误，那么 generatebytes 也会返回非零错误吗？您只是想使覆盖率高于某个任意阈值，还是这样的检查对您来说实际上很重要？对于前一种情况，请重新考虑。

在后一种情况下，请注意直接依赖像 rand.Reader 这样的包级单例会使测试变得困难。提高可测试性需要一些额外的（但不是禁止性的）努力。

一种可能的方法

值得庆幸的是，已经有一个 rand.reader 满足的便捷接口：io.Reader。不过，如果没有这样的接口可用，也没什么大不了的；您可以声明您自己的定制。这就是 go 接口隐式满足感的伟大之处！

选择一种将 io.reader 传递到 generatebytes 的方法。最惯用的方法是声明 generatebytes，不是作为顶级函数，而是作为某些自定义结构类型（下面名为 sourceofrandomness）的方法，该类型具有（可能是匿名的）io.reader 字段：

type sourceofrandomness struct {
    reader io.reader
}

这样做可以让您选择在自定义类型实例化时注入哪个 io.reader 实现器。 generatebytes 然后可以通过其接收器访问它所需的 io.reader：

func (sor *sourceofrandomness) generatebytes(length int) ([]byte, error) {
    bytes := make([]byte, length)
    if _, err := sor.reader.read(bytes); err != nil {
        return nil, err
    }
    return bytes, nil
}

然后，在生产代码中，您可以使用实际的 rand.reader 初始化 sourceofrandomness 值：

sor := sourceofrandomness{reader: rand.reader}
sor.generatebytes(42)

对于您想要的特定测试用例，您可以简单地利用（如 his comment 中的 mh-cbon 中提到的）iotest.ErrReader，它返回一个 io.reader，其 read 方法无条件失败：

import (
  "crypto/rand"
  "testing/iotest"
)

func TestGenerateBytesFails(t *testing.T) {
  // Arrange
  sor := SourceOfRandomness{
    Reader: iotest.ErrReader(errors.New("oops"))
  }
  const dummyLength = 42
  // Act
  _, err := sor.GenerateBytes(dummyLength)
  // Assert
  if err != nil {
    t.Error("want non-nil error; got nil error")
  }
}

本篇关于《如何模拟 rand.Reader 的故障以进行测试？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！