test(coderd/database/pubsub): ensure db closure on unhappy paths (#16327)

This commit is contained in:
Mathias Fredriksson
2025-01-29 16:47:38 +02:00
committed by GitHub
parent c069563af1
commit b77b5432c6
2 changed files with 5 additions and 0 deletions

View File

@ -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()