fix testContainers (#365)

Melkeydev · Jan 17, 2025 · 9f2a097 · 9f2a097
1 parent 1d95d78
commit 9f2a097
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/cmd/template/dbdriver/files/tests/mongo.tmpl b/cmd/template/dbdriver/files/tests/mongo.tmpl
@@ -5,10 +5,11 @@ import (
 	"log"
 	"testing"
 
+	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/modules/mongodb"
 )
 
-func mustStartMongoContainer() (func(context.Context) error, error) {
+func mustStartMongoContainer() (func(context.Context, ...testcontainers.TerminateOption) error, error) {
 	dbContainer, err := mongodb.Run(context.Background(), "mongo:latest")
 	if err != nil {
 		return nil, err

diff --git a/cmd/template/dbdriver/files/tests/mysql.tmpl b/cmd/template/dbdriver/files/tests/mysql.tmpl
@@ -11,7 +11,7 @@ import (
 	"github.com/testcontainers/testcontainers-go/wait"
 )
 
-func mustStartMySQLContainer() (func(context.Context) error, error) {
+func mustStartMySQLContainer() (func(context.Context, ...testcontainers.TerminateOption) error, error) {
 	var (
 		dbName = "database"
 		dbPwd  = "password"

diff --git a/cmd/template/dbdriver/files/tests/redis.tmpl b/cmd/template/dbdriver/files/tests/redis.tmpl
@@ -5,10 +5,11 @@ import (
 	"log"
 	"testing"
 
+	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/modules/redis"
 )
 
-func mustStartRedisContainer() (func(context.Context) error, error) {
+func mustStartRedisContainer() (func(context.Context, ...testcontainers.TerminateOption) error, error) {
 	dbContainer, err := redis.Run(
 		context.Background(),
 		"docker.io/redis:7.2.4",