feat: add clean up discord message method for REST test (#72)

EiffelFly · Mar 16, 2023 · 2fe54bb · 2fe54bb
1 parent a9e5698
commit 2fe54bb
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 14 deletions.
diff --git a/.changeset/tender-toys-warn.md b/.changeset/tender-toys-warn.md
@@ -0,0 +1,5 @@
+---
+"curioucity": minor
+---
+
+Add clean up discord message method for REST test
diff --git a/tests/api/rest.js b/tests/api/rest.js
@@ -9,20 +9,21 @@ export const options = {};
 export const API_HOST = "http://localhost:8080";
 
 const main = () => {
-  urlServices.createUrl();
-  urlServices.deleteUrl();
-  urlServices.getUrl();
-  urlServices.listUrl();
-  urlServices.cleanUpUrl();
-  tagServices.createTag();
-  tagServices.deleteTag();
-  tagServices.getTag();
-  tagServices.listTag();
-  tagServices.cleanUpTag();
-  // discordServices.createDiscordMessage();
-  // discordServices.deleteDiscordMessage();
-  // discordServices.getDiscordMessage();
-  // discordServices.listDiscordMessage();
+  // urlServices.createUrl();
+  // urlServices.deleteUrl();
+  // urlServices.getUrl();
+  // urlServices.listUrl();
+  // urlServices.cleanUpUrl();
+  // tagServices.createTag();
+  // tagServices.deleteTag();
+  // tagServices.getTag();
+  // tagServices.listTag();
+  // tagServices.cleanUpTag();
+  discordMessageServices.createDiscordMessage();
+  discordMessageServices.deleteDiscordMessage();
+  discordMessageServices.getDiscordMessage();
+  discordMessageServices.listDiscordMessage();
+  discordMessageServices.cleanUpDiscordMessage();
   // discordServices.createDiscordThread();
   // discordServices.deleteDiscordThread();
   // discordServices.getDiscordThread();

diff --git a/tests/api/rest_discord_message.js b/tests/api/rest_discord_message.js
@@ -329,3 +329,43 @@ export const listDiscordMessage = () => {
     }
   });
 };
+
+export const cleanUpDiscordMessage = () => {
+  const listDiscordMessageResponse = http.request(
+    "GET",
+    `${API_HOST}/discord/messages`,
+    undefined,
+    {
+      headers: {
+        "Content-Type": "application/json",
+      },
+    }
+  );
+
+  if (
+    listDiscordMessageResponse.status === 200 &&
+    listDiscordMessageResponse.json() &&
+    listDiscordMessageResponse.json().discord_messages &&
+    listDiscordMessageResponse.json().discord_messages.length > 0
+  ) {
+    for (const discordMessage of listDiscordMessageResponse.json()
+      .discord_messages) {
+      check(
+        http.request(
+          "DELETE",
+          `${API_HOST}/discord/messages/${discordMessage.message_id}`,
+          undefined,
+          {
+            headers: {
+              "Content-Type": "application/json",
+            },
+          }
+        ),
+        {
+          "cleanUpDiscordMessage - DELETE /discord/messages/:message_id - response status should be 204":
+            (r) => r.status === 204,
+        }
+      );
+    }
+  }
+};