test(coderd/database/awsiamrds): fix unclosed pubsub (#16202)

mafredri · web-flow · commit 4d9dd0aa9451 · 2025-01-20T17:25:31.000Z
diff --git a/coderd/database/awsiamrds/awsiamrds_test.go b/coderd/database/awsiamrds/awsiamrds_test.go
@@ -52,13 +52,14 @@ func TestDriver(t *testing.T) {
 
 	ps, err := pubsub.New(ctx, logger, db, url)
 	require.NoError(t, err)
+	defer ps.Close()
 
 	gotChan := make(chan struct{})
 	subCancel, err := ps.Subscribe("test", func(_ context.Context, _ []byte) {
 		close(gotChan)
 	})
-	defer subCancel()
 	require.NoError(t, err)
+	defer subCancel()
 
 	err = ps.Publish("test", []byte("hello"))
 	require.NoError(t, err)
