Skip to content

Commit

Permalink
refactor(e2e): handle errors consistently while setting up dbs
Browse files Browse the repository at this point in the history
  • Loading branch information
olexsmir committed Oct 24, 2024
1 parent a5d882b commit 4f17e84
Showing 1 changed file with 10 additions and 21 deletions.
31 changes: 10 additions & 21 deletions e2e/e2e_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -78,15 +78,11 @@ func (e *AppTestSuite) SetupSuite() {
e.ctx = context.Background()
e.require = e.Require()

db, stop, err := e.prepPostgres()
e.require.NoError(err)

db, stop := e.prepPostgres()
e.postgresDB = db
e.stopPostgres = stop

rdb, stop, err := e.prepRedis()
e.require.NoError(err)

rdb, stop := e.prepRedis()
e.redis = rdb
e.stopRedis = stop

Expand Down Expand Up @@ -143,7 +139,7 @@ func (e *AppTestSuite) initDeps() {
e.router = handler.Handler()
}

func (e *AppTestSuite) prepPostgres() (*psqlutil.DB, stopDBFunc, error) {
func (e *AppTestSuite) prepPostgres() (*psqlutil.DB, stopDBFunc) {
dbCredential := "testing"
postgresContainer, err := postgres.Run(e.ctx,
"postgres:16-alpine",
Expand All @@ -153,10 +149,7 @@ func (e *AppTestSuite) prepPostgres() (*psqlutil.DB, stopDBFunc, error) {
testcontainers.WithWaitStrategy(wait.ForListeningPort("5432/tcp")))
e.require.NoError(err)

stop := func() {
err = postgresContainer.Terminate(e.ctx)
e.require.NoError(err)
}
stop := func() { e.require.NoError(postgresContainer.Terminate(e.ctx)) }

// connect to the db
host, err := postgresContainer.Host(e.ctx)
Expand Down Expand Up @@ -189,20 +182,17 @@ func (e *AppTestSuite) prepPostgres() (*psqlutil.DB, stopDBFunc, error) {
)
e.require.NoError(err)

err = m.Up()
e.require.NoError(err)
e.require.NoError(m.Up())
e.require.NoError(driver.Close())

return db, stop, driver.Close()
return db, stop
}

func (e *AppTestSuite) prepRedis() (*redis.Client, stopDBFunc, error) {
func (e *AppTestSuite) prepRedis() (*redis.Client, stopDBFunc) {
redisContainer, err := tsredis.Run(e.ctx, "redis:7.4-alpine")
e.require.NoError(err)

stop := func() {
err := redisContainer.Terminate(e.ctx)
e.require.NoError(err)
}
stop := func() { e.require.NoError(redisContainer.Terminate(e.ctx)) }

uri, err := redisContainer.ConnectionString(e.ctx)
e.require.NoError(err)
Expand All @@ -211,8 +201,7 @@ func (e *AppTestSuite) prepRedis() (*redis.Client, stopDBFunc, error) {
e.require.NoError(err)

rdb := redis.NewClient(connOpts)

return rdb, stop, nil
return rdb, stop
}

func (e *AppTestSuite) getConfig() *config.Config {
Expand Down

0 comments on commit 4f17e84

Please sign in to comment.