带有 *s3.S3 对象的 Golang 单元测试

Question

我应该如何对以下代码进行单元测试。我试图使用 coutnerfiter 伪造输入“*s3.S3”对象，但它对我不起作用。我是 coutnerfiter 和 Go 的新手，有人可以帮助我吗？func (l *listContentImp) ListS3Content(client *s3.S3) (bool, error) {  listObj := &s3.ListObjectsV2Input{    Bucket: aws.String(l.bucket),  }  var err error  l.lObj, err = client.ListObjectsV2(listObj)  if err != nil {    return false, err  }  return true, nil}

Answer 1

您不应该传递对该s3.S3结构的引用。使用适用于 Go v1 的 AWS 开发工具包时，您通常会传递相应的服务接口。对于 S3，这是s3iface。

您的函数签名如下所示：

func (l *listContentImp) ListS3Content(client s3iface.S3API) (bool, error)

现在，您传递的每个实现至少一种方法的结构s3iface.S3API都将起作用。

在运行时，您将传递适当的服务客户端，但在单元测试中，您可以只传递一个模拟：

type mock struct {

    s3iface.S3API

    output *s3.ListObjectsV2Output

    err     error

}

func (m mock) ListObjectsV2(*s3.ListObjectsV2Input) (*s3.ListObjectsV2Output, error) {

    return m.output, m.err

}

在您的测试中，您创建模拟并将其传递给您的函数：

func Test_ListObject(t *testing.T) {

    l := &listContentImp{...}

    m := mock{

        output: &s3.ListObjectsV2Output{...},

        err: nil

    }

    result, err := l.ListS3Content(m)

    

    [... add checks here...]

}