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

@@ -11,6 +11,7 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/modules/clickhouse"
)
@@ -39,14 +40,10 @@ func getTestConnection(t testing.TB, cfg Config) (*Storage, error) {
clickhouse.WithPassword(clickhousePass),
clickhouse.WithDatabase(clickhouseDB),
)
testcontainers.CleanupContainer(t, c)
if err != nil {
return nil, err
}
t.Cleanup(func() {
if c != nil {
require.NoError(t, c.Terminate(ctx))
}
})
hostPort, err := c.ConnectionHost(ctx)
if err != nil {