From f55fa3d6aed22752250f905d3ca4c52593c6a1a9 Mon Sep 17 00:00:00 2001
From: Corneil du Plessis <pcorneil@vmware.com>
Date: Thu, 17 Nov 2022 16:39:50 +0200
Subject: [PATCH] [SB3] Resolve race conditions.

---
 .../fn/consumer/websocket/WebsocketConsumerTests.java | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java b/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
index 1ba3730b..0f03be26 100644
--- a/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
+++ b/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
@@ -43,6 +43,7 @@ import static org.assertj.core.api.Assertions.assertThat;
  * @author Oliver Moser
  * @author Gary Russell
  * @author Artem Bilan
+ * @author Corneil du Plessis
  */
 @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT,
 		properties = {
@@ -162,10 +163,12 @@ public class WebsocketConsumerTests {
 
 	private List<String> submitMultipleMessages(int messageCount) {
 		List<String> messagesToSend = new ArrayList<>(messageCount);
-		for (int i = 0; i < messageCount; i++) {
-			String message = "message_" + i;
-			messagesToSend.add(message);
-			websocketConsumer.accept(MessageBuilder.withPayload(message).build());
+		synchronized (websocketConsumer) {
+			for (int i = 0; i < messageCount; i++) {
+				String message = "message_" + i;
+				messagesToSend.add(message);
+				websocketConsumer.accept(MessageBuilder.withPayload(message).build());
+			}
 		}
 
 		return messagesToSend;