Golang使用Docker进行集成测试的示例详解

更新时间：2023年07月27日 08:58:14 作者：江湖十年

集成测试需要解决外部依赖问题，如 MySQL、Redis、网络等依赖，本文就来聊聊 Go 程序如何使用 Docker 来解决集成测试中外部依赖问题吧

登录程序示例

在 Web 开发中，登录需求是一个较为常见的功能。所以，本文就以登录程序为例，讲解使用 Docker 启动 Redis 进行集成测试。

登录程序如下：

func Login(mobile, smsCode string, rdb *redis.Client) (string, error) {
	ctx := context.Background()
	// 查找验证码
	captcha, err := GetSmsCaptchaFromRedis(ctx, rdb, mobile)
	if err != nil {
		if err == redis.Nil {
			return "", fmt.Errorf("invalid sms code or expired")
		}
		return "", err
	}
	if captcha != smsCode {
		return "", fmt.Errorf("invalid sms code")
	}
	token, _ := GenerateToken(32)
	err = SetAuthTokenToRedis(ctx, rdb, token, mobile)
	if err != nil {
		return "", err
	}
	return token, nil
}

可以通过如下方式获取 Redis 客户端对象 rdb：

import "github.com/redis/go-redis/v9"
func NewRedisClient() *redis.Client {
	return redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
	})
}

生成随机 token 的函数定义如下：

var GenerateToken = func(length int) (string, error) {
	token := make([]byte, length)
	_, err := rand.Read(token)
	if err != nil {
		return "", err
	}
	return base64.URLEncoding.EncodeToString(token)[:length], nil
}

本程序提供了如下几个操作 Reids 的函数：

var (
	smsCaptchaExpire    = 5 * time.Minute
	smsCaptchaKeyPrefix = "sms:captcha:%s"
	authTokenExpire    = 24 * time.Hour
	authTokenKeyPrefix = "auth:token:%s"
)
func SetSmsCaptchaToRedis(ctx context.Context, redis *redis.Client, mobile, captcha string) error {
	key := fmt.Sprintf(smsCaptchaKeyPrefix, mobile)
	return redis.Set(ctx, key, captcha, smsCaptchaExpire).Err()
}
func GetSmsCaptchaFromRedis(ctx context.Context, redis *redis.Client, mobile string) (string, error) {
	key := fmt.Sprintf(smsCaptchaKeyPrefix, mobile)
	return redis.Get(ctx, key).Result()
}
func DeleteSmsCaptchaFromRedis(ctx context.Context, redis *redis.Client, mobile string) error {
	key := fmt.Sprintf(smsCaptchaKeyPrefix, mobile)
	return redis.Del(ctx, key).Err()
}
func SetAuthTokenToRedis(ctx context.Context, redis *redis.Client, token, mobile string) error {
	key := fmt.Sprintf(authTokenKeyPrefix, token)
	return redis.Set(ctx, key, mobile, authTokenExpire).Err()
}
func GetAuthTokenFromRedis(ctx context.Context, redis *redis.Client, token string) (string, error) {
	key := fmt.Sprintf(authTokenKeyPrefix, token)
	return redis.Get(ctx, key).Result()
}
func DeleteAuthTokenFromRedis(ctx context.Context, redis *redis.Client, token string) error {
	key := fmt.Sprintf(authTokenKeyPrefix, token)
	return redis.Del(ctx, key).Err()
}

Login 函数用法如下：

func main() {
	rdb := NewRedisClient()
	token, err := Login("13800001111", "123456", rdb)
	if err != nil {
		fmt.Println(err)
		return
	}
	fmt.Println(token)
}

使用 Docker 进行集成测试

要想对 Login 函数进行集成测试，就需要解决 Reids 外部依赖问题。

在 Go 程序中，我们可以使用 testcontainers-go 这个包来解决，它可以让我们很方便的在 Docker 中启动 Reids 服务。

安装 testcontainers-go：

$ go get github.com/testcontainers/testcontainers-go

我们可以在测试代码开始执行之前启动 Docker 容器来运行 Redis 服务，然后执行测试代码，最后测试代码执行完成后再停止并删除 Docker 容器。

可以定义一个 setup 函数用来准备 Docker 容器：

var rdbClient *redis.Client
func setup() func() {
	ctx := context.Background()
	req := testcontainers.ContainerRequest{
		Image:        "redis:6.0.20-alpine",
		ExposedPorts: []string{"6379/tcp"},
		WaitingFor:   wait.ForLog("Ready to accept connections"),
	}
	redisC, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: req,
		Started:          true,
	})
	if err != nil {
		panic(fmt.Sprintf("failed to start container: %s", err.Error()))
	}
	endpoint, err := redisC.Endpoint(ctx, "")
	if err != nil {
		panic(fmt.Sprintf("failed to get endpoint: %s", err.Error()))
	}
	rdbClient = redis.NewClient(&redis.Options{
		Addr: endpoint,
	})
	// 清理 Redis 容器
	return func() {
		if err := redisC.Terminate(ctx); err != nil {
			panic(fmt.Sprintf("failed to terminate container: %s", err.Error()))
		}
	}
}

可以发现使用 testcontainers-go 启动一个 Redis 容器非常简单，我们指定了 Docker 容器镜像为 redis:6.0.20-alpine，映射端口为 6379/tcp。

Redis 容器启动后将实例化的 Redis 客户端保存到全局变量 rdbClient 中，方便在测试函数中使用，setup 函数最终返回一个 teardown 函数可以清理容器。

定义 TestMain 函数如下，作为测试程序的入口：

func TestMain(m *testing.M) {
	teardown := setup()
	code := m.Run()
	teardown()
	os.Exit(code)
}

为了测试 Login 函数，我们需要在 Reids 中准备一些测试数据，因为 Login 函数内部需要查询 Reids 中的验证码，所以可以定义一个 setupLogin 函数来实现：

func setupLogin(tb testing.TB) func(tb testing.TB) {
	// 准备测试数据
	err := SetSmsCaptchaToRedis(context.Background(), rdbClient, "18900001111", "123456")
	assert.NoError(tb, err)
	// 清理测试数据
	return func(tb testing.TB) {
		err := DeleteSmsCaptchaFromRedis(context.Background(), rdbClient, "18900001111")
		assert.NoError(tb, err)
		err = DeleteAuthTokenFromRedis(context.Background(), rdbClient, "token")
		assert.NoError(tb, err)
	}
}

setupLogin 函数返回 teardownLogin 函数用来清理 Redis 中的测试数据，防止当有多个测试函数时互相影响。

现在可以编写 Login 函数的测试代码了：

func TestLogin(t *testing.T) {
	teardownLogin := setupLogin(t)
	defer teardownLogin(t)
	// 测试登录成功情况
	token, err := Login("18900001111", "123456", rdbClient)
	assert.NoError(t, err)
	assert.Equal(t, "token", token)
	// 检查 Redis 中是否存在 token
	mobile, err := GetAuthTokenFromRedis(context.Background(), rdbClient, "token")
	assert.NoError(t, err)
	assert.Equal(t, "18900001111", mobile)
}

TestLogin 函数非常简单，这得益于前期的准备工作做的非常全面。

使用 go test 来执行测试函数：

$ go test -v
2023/07/26 20:48:12 github.com/testcontainers/testcontainers-go - Connected to docker:
Server Version: 20.10.21
API Version: 1.41
Operating System: Docker Desktop
Total Memory: 7851 MB
2023/07/26 20:48:12 🐳 Creating container for image docker.io/testcontainers/ryuk:0.5.1
2023/07/26 20:48:12 ✅ Container created: a261dc723001
2023/07/26 20:48:12 🐳 Starting container: a261dc723001
2023/07/26 20:48:12 ✅ Container started: a261dc723001
2023/07/26 20:48:12 🚧 Waiting for container id a261dc723001 image: docker.io/testcontainers/ryuk:0.5.1. Waiting for: &{Port:8080/tcp timeout:<nil> PollInterval:100ms}
2023/07/26 20:48:13 🐳 Creating container for image redis:6.0.20-alpine
2023/07/26 20:48:13 ✅ Container created: 6420ead815a0
2023/07/26 20:48:13 🐳 Starting container: 6420ead815a0
2023/07/26 20:48:13 ✅ Container started: 6420ead815a0
2023/07/26 20:48:13 🚧 Waiting for container id 6420ead815a0 image: redis:6.0.20-alpine. Waiting for: &{timeout:<nil> Log:Ready to accept connections Occurrence:1 PollInterval:100ms}
=== RUN TestLogin
--- PASS: TestLogin (0.01s)
PASS
2023/07/26 20:48:13 🐳 Terminating container: 6420ead815a0
2023/07/26 20:48:13 🚫 Container terminated: 6420ead815a0
ok github.com/jianghushinian/test/db/redis 1.630s

测试通过。

总结

我们使用 testcontainers-go 包实现了在 Go 程序中启动一个 Docker 容器，以此解决了集成测试中依赖外部 Redis 问题。

可以发现，Docker 非常适合集成测试，使用 Dokcer 来辅助集成测试是 Go 应用程序集成测试的最佳实践。而完善的集成测试，可以确保应用程序的可靠性、可扩展性和可维护性。

以上就是Golang使用Docker进行集成测试的示例详解的详细内容，更多关于Go Docker集成测试的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

Go
Docker

golang简单读写文件示例
这篇文章主要介绍了golang简单读写文件的方法,实例分析了Go简单文件读取与写入操作的相关技巧,需要的朋友可以参考下
2016-07-07
go语言编程实现递归函数示例详解
这篇文章主要为大家介绍了go语言编程实现递归函数示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-09-09
一文教你学会Go中singleflight的使用
缓存在项目中使用应该是非常频繁的,提到缓存只要了解过 singleflight ,基本都会用于缓存实现的一部分吧,下面就跟随小编一起来学习一下singleflight的使用吧
2024-02-02
Go log库的使用示例详解
Go语言内置的log库提供了基本的日志记录功能,支持日志的格式化输出、设置日志前缀、配置输出位置等,可以通过标准logger或创建新的Logger对象来使用,log库简单易用,但功能有限,可能需要配合第三方日志库如logrus、zap等来满足复杂需求
2024-09-09
go语言搬砖之go jmespath实现查询json数据
这篇文章主要为大家介绍了go语言搬砖之go jmespath实现查询json数据，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-06-06
Go语言循环遍历含有中文的字符串的方法小结
这篇文章主要介绍了Go语言循环遍历含有中文的字符串的几种方法，文章通过代码示例讲解的非常详细，具有一定的参考价值，感兴趣的小伙伴跟着小编一起来看看吧
2023-07-07
Golang 1.18新特性模糊测试用法详解
模糊测试是一种软件测试技术。其核心思想是將自动或半自动生成的随机数据输入到一个程序中，并监视程序异常，如崩溃，断言失败，以发现可能的程序错误，比如内存泄漏，本文给大家介绍了Golang 1.18 新特性模糊测试，感兴趣的同学可以参考阅读下
2023-05-05
Go语言之自定义集合Set
本文主要介绍的是Go语言的自定义集合Set，文中介绍的很详细，有需要的可以参考学习。
2016-08-08
Golang实现快速求幂的方法详解
这篇文章主要为大家详细介绍了如何利用Golang实现快速求幂，文中的示例代码讲解详细，对我们学习或工作有一定参考价值，需要的可以参考一下
2022-06-06
go-zero服务部署配置及源码解读
这篇文章主要为大家介绍了go-zero服务部署配置及源码解读,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2023-08-08