mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
test(coderd/database/pubsub): ensure db closure on unhappy paths (#16327)
This commit is contained in:
@@ -305,6 +305,9 @@ func TestMeasureLatency(t *testing.T) {
|
||||
require.NoError(t, err)
|
||||
db, err := sql.Open("postgres", connectionURL)
|
||||
require.NoError(t, err)
|
||||
t.Cleanup(func() {
|
||||
_ = db.Close()
|
||||
})
|
||||
ps, err := pubsub.New(ctx, logger, db, connectionURL)
|
||||
require.NoError(t, err)
|
||||
|
||||
|
||||
@@ -137,6 +137,7 @@ func TestPGPubsubDriver(t *testing.T) {
|
||||
// use a separate subber and pubber so we can keep track of listener connections
|
||||
db, err := sql.Open("postgres", connectionURL)
|
||||
require.NoError(t, err)
|
||||
defer db.Close()
|
||||
pubber, err := pubsub.New(ctx, logger, db, connectionURL)
|
||||
require.NoError(t, err)
|
||||
defer pubber.Close()
|
||||
@@ -147,6 +148,7 @@ func TestPGPubsubDriver(t *testing.T) {
|
||||
tconn, err := subDriver.Connector(connectionURL)
|
||||
require.NoError(t, err)
|
||||
tcdb := sql.OpenDB(tconn)
|
||||
defer tcdb.Close()
|
||||
subber, err := pubsub.New(ctx, logger, tcdb, connectionURL)
|
||||
require.NoError(t, err)
|
||||
defer subber.Close()
|
||||
|
||||
Reference in New Issue
Block a user