:splarkle: tests for the tokens

Author: ishaileshmishra

src/test/java/com/contentstack/cms/stack/TokenUnitTest.java

@@ -3,16 +3,14 @@
 import com.contentstack.cms.Contentstack;
 import com.contentstack.cms.core.Util;
 import io.github.cdimascio.dotenv.Dotenv;
-import okhttp3.MediaType;
 import okhttp3.Request;
-import okhttp3.RequestBody;
-import okio.BufferedSink;
 import org.json.simple.JSONObject;
+import org.json.simple.parser.JSONParser;
+import org.json.simple.parser.ParseException;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 
-import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;
@@ -25,6 +23,39 @@ class TokenUnitTest {
     protected static String MANAGEMENT_TOKEN = Dotenv.load().get("auth_token");
     static Tokens tokens;
 
+    // Create a JSONObject, JSONObject could be created in multiple ways.
+    // We choose JSONParser that converts string to JSONObject
+    String stringToParse = "{\n" +
+            "    \"module\":\"branch\",\n" +
+            "    \"branches\":[\n" +
+            "        \"main\",\n" +
+            "        \"development\"\n" +
+            "    ],\n" +
+            "    \"acl\":{\n" +
+            "        \"read\":true\n" +
+            "    }\n" +
+            "},\n" +
+            "{\n" +
+            "    \"module\":\"branch_alias\",\n" +
+            "    \"branch_aliases\":[\n" +
+            "        \"deploy\",\n" +
+            "        \"release\"\n" +
+            "    ],\n" +
+            "    \"acl\":{\n" +
+            "        \"read\":true\n" +
+            "    }\n" +
+            "}";
+    JSONParser parser = new JSONParser();
+    JSONObject body;
+
+    {
+        try {
+            body = (JSONObject) parser.parse(stringToParse);
+        } catch (ParseException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
     @BeforeAll
     static void setup() {
         HashMap<String, Object> headers = new HashMap<>();
@@ -54,9 +85,6 @@ void allDeliveryTokens() {
 
     @Test
     void singleDeliveryToken() {
-        Map<String, Object> queryParam = new HashMap<>();
-        queryParam.put("include_count", false);
-        queryParam.put("include_branch", false);
         Request request = tokens.getDeliveryToken(_uid).request();
         Assertions.assertEquals(2, request.headers().names().size());
         Assertions.assertEquals("GET", request.method());
@@ -72,18 +100,6 @@ void singleDeliveryToken() {
 
     @Test
     void createDeliveryToken() {
-        Map<String, Object> queryParam = new HashMap<>();
-        queryParam.put("include_count", false);
-        queryParam.put("include_branch", false);
-
-        JSONObject body = new JSONObject();
-        JSONObject innerBody = new JSONObject();
-        innerBody.put("tags", Arrays.asList("tag1", "tag2"));
-        innerBody.put("data_type", "text");
-        innerBody.put("title", "Old Extension");
-        innerBody.put("src", "Enter either the source code");
-        body.put("extension", innerBody);
-
         Request request = tokens.createDeliveryToken(body).request();
         Assertions.assertEquals(2, request.headers().names().size());
         Assertions.assertEquals("PUT", request.method());
@@ -100,7 +116,7 @@ void createDeliveryToken() {
 
     @Test
     void updateDeliveryToken() {
-        Request request = tokens.updateDeliveryToken(_uid, new JSONObject()).request();
+        Request request = tokens.updateDeliveryToken(_uid, body).request();
         Assertions.assertEquals(2, request.headers().names().size());
         Assertions.assertEquals("DELETE", request.method());
         Assertions.assertTrue(request.url().isHttps());
@@ -113,8 +129,6 @@ void updateDeliveryToken() {
 
     @Test
     void deleteDeliveryToken() {
-        Map<String, Object> theQuery = new HashMap<>();
-        theQuery.put("include_branch", false);
         Request request = tokens.deleteDeliveryToken(_uid).request();
         Assertions.assertEquals(2, request.headers().names().size());
         Assertions.assertEquals("GET", request.method());
@@ -129,8 +143,6 @@ void deleteDeliveryToken() {
 
     @Test
     void deleteDeliveryTokenForcefully() {
-        Map<String, Object> theQuery = new HashMap<>();
-        theQuery.put("include_branch", false);
         Request request = tokens.deleteDeliveryToken(_uid, true).request();
         Assertions.assertEquals(2, request.headers().names().size());
         Assertions.assertEquals("GET", request.method());
@@ -174,7 +186,7 @@ void singleManagementToken() {
 
     @Test
     void createManagementToken() {
-        Request request = tokens.createManagementToken(new JSONObject()).request();
+        Request request = tokens.createManagementToken(body).request();
         Assertions.assertEquals(3, request.headers().names().size());
         Assertions.assertEquals("DELETE", request.method());
         Assertions.assertNull(request.body());
@@ -188,7 +200,7 @@ void createManagementToken() {
 
     @Test
     void updateManagementToken() {
-        Request request = tokens.updateManagementToken(_uid, new JSONObject()).request();
+        Request request = tokens.updateManagementToken(_uid, body).request();
         Assertions.assertEquals(3, request.headers().names().size());
         Assertions.assertEquals("DELETE", request.method());
         Assertions.assertNull(request.body());