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

This commit is contained in:
Mathias Fredriksson
2025-01-20 19:25:31 +02:00
committed by GitHub
parent c0db364f3f
commit 4d9dd0aa94

View File

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