package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: use cleanup function

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

7
clickhouse/clickhouse_test.go
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 {

7
couchbase/couchbase_test.go
View File

@@ -7,6 +7,7 @@ import (
"time"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/modules/couchbase"
)
@@ -36,14 +37,10 @@ func newTestStore(t testing.TB) (*Storage, error) {
couchbase.WithAdminCredentials(couchbaseUser, couchbasePass),
couchbase.WithBuckets(bucket),
)
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 {

7
minio/minio_test.go
View File

@@ -8,6 +8,7 @@ import (
"time"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/modules/minio"
)
@@ -34,14 +35,10 @@ func newTestStore(t testing.TB) (*Storage, error) {
minio.WithUsername(minioUser),
minio.WithPassword(minioPass),
)
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 {

7
mongodb/mongodb_test.go
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 {

7
mssql/mssql_test.go
View File

@@ -8,6 +8,7 @@ import (
"time"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/modules/mssql"
)
@@ -35,14 +36,10 @@ func newTestStore(t testing.TB) (*Storage, error) {
mssql.WithPassword(mssqlPass),
mssql.WithAcceptEULA(),
)
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 {

7
mysql/mysql_test.go
View File

@@ -8,6 +8,7 @@ import (
"time"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/modules/mysql"
)
@@ -51,12 +52,8 @@ func mustStartMySQL(t testing.TB) *mysql.MySQLContainer {
mysql.WithUsername(mysqlUser),
mysql.WithDatabase(mysqlDatabase),
)
testcontainers.CleanupContainer(t, c)
require.NoError(t, err)
t.Cleanup(func() {
if c != nil {
require.NoError(t, c.Terminate(ctx))
}
})
return c
}

6
postgres/postgres_test.go
View File

@@ -42,12 +42,8 @@ func newTestStore(t testing.TB) (*Storage, error) {
wait.ForLog("database system is ready to accept connections").WithOccurrence(2),
),
)
testcontainers.CleanupContainer(t, c)
require.NoError(t, err)
t.Cleanup(func() {
if c != nil {
require.NoError(t, c.Terminate(ctx))
}
})
conn, err := c.ConnectionString(ctx, "sslmode=disable")
if err != nil {