chore: use cleanup function

This commit is contained in:
Manuel de la Peña
2024-12-12 12:12:56 +01:00
parent b074e83e98
commit cac4fd5e8d
7 changed files with 13 additions and 35 deletions

View File

@@ -7,6 +7,7 @@ import (
"time"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/modules/mongodb"
)
@@ -29,14 +30,10 @@ func newTestStore(t testing.TB) (*Storage, error) {
ctx := context.Background()
c, err := mongodb.Run(ctx, img, mongodb.WithUsername(mongoDBUser), mongodb.WithPassword(mongoDBPass))
testcontainers.CleanupContainer(t, c)
if err != nil {
return nil, err
}
t.Cleanup(func() {
if c != nil {
require.NoError(t, c.Terminate(ctx))
}
})
conn, err := c.ConnectionString(ctx)
if err != nil {