Tweak tests

2 files changed, 6 insertions, 3 deletions

diff --git a/test/automated/api/chatmoderation.test.js b/test/automated/api/chatmoderation.test.js
index 84fc83a71..48585408c 100644
--- a/test/automated/api/chatmoderation.test.js
+++ b/test/automated/api/chatmoderation.test.js
@@ -63,6 +63,8 @@ test('verify we can make API call to mark message as hidden', async (done) => {
 });
 
 test('verify message has become hidden', async (done) => {
+  await new Promise((r) => setTimeout(r, 2000));
+
   const res = await request
     .get('/api/admin/chat/messages')
     .expect(200)
diff --git a/test/automated/api/lib/chat.js b/test/automated/api/lib/chat.js
index e5c47e121..8b97f5336 100644
--- a/test/automated/api/lib/chat.js
+++ b/test/automated/api/lib/chat.js
@@ -11,7 +11,7 @@ async function registerChat() {
   }
 }
 
-function sendChatMessage(message, accessToken, done) {
+async function sendChatMessage(message, accessToken, done) {
   const ws = new WebSocket(
     `ws://localhost:8080/ws?accessToken=${accessToken}`,
     {
@@ -19,9 +19,10 @@ function sendChatMessage(message, accessToken, done) {
     }
   );
 
-  function onOpen() {
-    ws.send(JSON.stringify(message), function () {
+  async function onOpen() {
+    ws.send(JSON.stringify(message), async function () {
       ws.close();
+      await new Promise((r) => setTimeout(r, 2000));
       done();
     });
   }