在Golang中测试/模拟第3方软件包

Question

我是Golang的新手，在学习语言时一直采用TDD方法。 我一直相处得很好，但是我发现测试第三方软件包相当笨拙，这使我相信我使用了错误的方法。

我遇到麻烦的特定情况是模拟Redis客户端进行错误处理。 我采用的方法是创建自己的接口，该实现包装了我要使用的客户端方法。

type Redis interface {
    Get(key string) (string, error)
}

type RedisClient struct {
    client *redis.Client
}

func (redisClient *RedisClient) New(client *redis.Client) *RedisClient {
    redisClient.client = client

    return redisClient
}

func (redisClient *RedisClient) Get(key string) (string, error) {
    return redisClient.client.Get(key).Result()
}

然后，我可以创建一个模拟，该模拟实现相同的接口以返回我指定的任何值，特别是对于测试错误处理。

我遇到了一个障碍，即客户端上执行事务的特定方法（MULTI）返回属于该包的另一个接口。 在这种情况下我该怎么办？ 我自己实现该接口似乎是不可能的。

同样，随着该客户端使用率的增长，我自己的实现可能会增长到实现Redis整个接口的地步-这似乎与将其委托给外部依赖项的整个想法背道而驰。

是否有更好的方法来测试诸如错误处理之类的第三方程序包？

Answer 1

一种方法是创建一个类型，将其重点放在您要完成的事情上，而不是您在使用客户端的哪种方法。

假设您想要的只是一个用于保存和获取用户的存储，您可以想象这样的界面：

type UserStore interface {
  SaveUser(*User) error
  GetUserByID(id string) (*User, error)
  SearchUsers(query string) ([]User, error)
}

然后，您可以实现此存储的Redis版本，并在内部调用所需的任何客户端方法，这无关紧要。 您甚至可以在PostgreSQL或其他任何程序中实现一个。 此外，使用此方法可以使模拟更加容易，因为您所需要做的就是实现此接口而不是Redis接口。

这是此接口的模拟版本的示例：

type UserStoreMock struct {
  SaveUserFn func (*User) error
  SaveUserInvoked bool
  ...
}

func (m *UserStoreMock) SaveUser(u *User) error {
  m.SaveUserInvoked = true

  if m.SaveUserFn != nil {
    return m.SaveUserFn(u)
  }

  return nil
}
...

然后，您可以在以下测试中使用此模拟：

var m UserStoreMock

m.SaveUserFn = func(u *User) error {
  if u.ID != "123" {
    t.Fail("Bad ID")
  }

  return ErrDuplicateError
}
...