Purge opened connections on retries during tests (#4452)

This commit is contained in:
Calvin Leung Huang
2018-04-26 11:28:58 -04:00
committed by GitHub
parent 783a5c3422
commit 6fc57a91d8
7 changed files with 7 additions and 0 deletions

View File

@@ -152,6 +152,7 @@ func testPostgresDB(t testing.TB) (string, func()) {
if err != nil {
return err
}
defer db.Close()
return db.Ping()
}); err != nil {
t.Fatalf("postgresdb: could not connect: %s", err)

View File

@@ -53,6 +53,7 @@ func prepareMSSQLTestContainer(t *testing.T) (func(), string) {
if err != nil {
return err
}
defer db.Close()
return db.Ping()
}); err != nil {
cleanup()

View File

@@ -47,6 +47,7 @@ func prepareTestContainer(t *testing.T) (func(), string) {
if err != nil {
return err
}
defer db.Close()
return db.Ping()
}); err != nil {
cleanup()

View File

@@ -58,6 +58,7 @@ func prepareCockroachDBTestContainer(t *testing.T) (cleanup func(), retURL, tabl
if err != nil {
return err
}
defer db.Close()
_, err = db.Exec("CREATE DATABASE database")
return err
}); err != nil {

View File

@@ -55,6 +55,7 @@ func prepareMongoDBTestContainer(t *testing.T) (cleanup func(), retURL string) {
if err != nil {
return err
}
defer session.Close()
session.SetSyncTimeout(1 * time.Minute)
session.SetSocketTimeout(1 * time.Minute)
return session.Ping()

View File

@@ -52,6 +52,7 @@ func prepareMySQLTestContainer(t *testing.T, legacy bool) (cleanup func(), retUR
if err != nil {
return err
}
defer db.Close()
return db.Ping()
}); err != nil {
cleanup()

View File

@@ -50,6 +50,7 @@ func preparePostgresTestContainer(t *testing.T) (cleanup func(), retURL string)
if err != nil {
return err
}
defer db.Close()
return db.Ping()
}); err != nil {
t.Fatalf("Could not connect to PostgreSQL docker container: %s", err)