test pubsub

f0ssel · f0ssel · commit 9d8681b668c2 · 2024-08-14T17:45:39.000Z
diff --git a/coderd/database/awsiamrds/awsiamrds_test.go b/coderd/database/awsiamrds/awsiamrds_test.go
@@ -63,5 +63,9 @@ func TestDriver(t *testing.T) {
 	err = ps.Publish("test", []byte("hello"))
 	require.NoError(t, err)
 
-	<-gotChan
+	select {
+	case <-gotChan:
+	case <-ctx.Done():
+		require.Fail(t, "timed out waiting for message")
+	}
 }
