Add create thread with unit tests and fix edit_thread and update tests

Author: hnrkcode

loading_api_wrapper/api.py

@@ -13,7 +13,6 @@
     "podcast",
     "conversation",
 ]
-
 EDITORIAL_SORT = ["title"]
 
 
@@ -274,8 +273,51 @@ def edit_post(self, post_id, message):
         # Handle any other unknown status code.
         return response.json()
 
-    def create_thread(self, title, text, category, post_type):
-        pass
+    def create_thread(self, title, message, category_name, post_type=None):
+        if category_name not in ["games", "other"]:
+            return {"code": 400, "message": "Invalid forum category"}
+
+        if post_type and post_type not in EDITORIAL_POST_TYPES:
+            return {"code": 400, "message": "Invalid post_type"}
+
+        if not post_type:
+            post_type = "regular"
+
+        url = f"{API_URL}/{API_VERSION}/posts/"
+        headers = {
+            "User-Agent": USER_AGENT,
+            "content-type": "application/x-www-form-urlencoded",
+        }
+        data = {
+            "category": category_name,
+            "postType": post_type,
+            "title": title,
+            "body": message,
+        }
+        response = requests.post(url, headers=headers, data=data, cookies=self._cookies)
+
+        # Validation errors. Happens when title or message is empty. Possibly in other cases too.
+        if response.status_code == 400:
+            return response.json()
+
+        # No auth token.
+        if response.status_code == 401:
+            return response.json()
+
+        if response.status_code == 201:
+            return {
+                "code": response.status_code,
+                "message": "Thread created",
+                "data": response.json(),
+            }
+
+        # Handle any other unknown status code.
+        return response.json()
 
     def edit_thread(self, thread_id, message):
-        return self.edit_post(thread_id, message)
+        thread_data = self.edit_post(thread_id, message)
+
+        if thread_data["code"] == 200:
+            thread_data["message"] = "Thread updated"
+
+        return thread_data

tests/test_loading_api_wrapper.py

@@ -1652,7 +1652,7 @@ def test_edit_post_failure_post_does_not_exist(self, mock_requests):
         self.assertEqual(response, expected_response)
 
     @patch("loading_api_wrapper.api.LoadingApiWrapper._authenticate")
-    def test_edit_post_failure_empty_thread_id(self, mock_authenticate):
+    def test_edit_post_failure_empty_message(self, mock_authenticate):
         expected_response = {
             "code": 400,
             "message": '"message" is not allowed to be empty',
@@ -1764,3 +1764,120 @@ def test_create_post_success(self, mock_requests, mock_authenticate):
         self.assertEqual(api._cookies, self.cookie_jar)
         self.assertEqual(response.get("code"), 201)
         self.assertEqual(response.get("data"), expected_response)
+
+    @patch("loading_api_wrapper.api.LoadingApiWrapper._authenticate")
+    @patch("loading_api_wrapper.api.requests")
+    def test_create_thread_success(self, mock_requests, mock_authenticate):
+        status_code = 201
+        expected_response = {
+            "id": "000000000000000000000000",
+            "body": "updated message",
+            "postType": "regular",
+            "createdAt": "2022-01-01T00:00:00.000Z",
+            "updatedAt": "2022-01-02T00:00:00.000Z",
+            "parentId": "222222222222222222222222",
+            "userId": "111111111111111111111111",
+            "replies": 0,
+        }
+
+        mock_response = MagicMock()
+        mock_response.status_code = status_code
+        mock_response.json.return_value = expected_response
+        mock_requests.post.return_value = mock_response
+        mock_authenticate.return_value = {"code": 200, "cookies": self.cookie_jar}
+
+        api = LoadingApiWrapper("test@email.com", "password")
+        response = api.create_thread(
+            title="Hello",
+            message="My message",
+            category_name="other",
+        )
+
+        self.assertIsNotNone(api._cookies)
+        self.assertEqual(api._cookies, self.cookie_jar)
+        self.assertEqual(response.get("code"), 201)
+        self.assertDictEqual(response.get("data"), expected_response)
+
+    def test_create_thread_failure_invalid_category(self):
+        expected_response = {"code": 400, "message": "Invalid forum category"}
+
+        api = LoadingApiWrapper()
+        response = api.create_thread(
+            title="Hello",
+            message="My message",
+            category_name="invalid_category",
+        )
+
+        self.assertEqual(response, expected_response)
+
+    def test_create_thread_failure_invalid_post_type(self):
+        expected_response = {"code": 400, "message": "Invalid post_type"}
+
+        api = LoadingApiWrapper()
+        response = api.create_thread(
+            title="Hello",
+            message="My message",
+            category_name="other",
+            post_type="invalid_post_type",
+        )
+
+        self.assertEqual(response, expected_response)
+
+    @patch("loading_api_wrapper.api.LoadingApiWrapper._authenticate")
+    @patch("loading_api_wrapper.api.requests")
+    def test_create_thread_failure_empty_title_or_message(
+        self, mock_requests, mock_authenticate
+    ):
+        status_code = 400
+        expected_response = {
+            "code": status_code,
+            "message": "Validation error",
+            "errors": [
+                {
+                    "field": "title",
+                    "location": "body",
+                    "messages": ['"title" is not allowed to be empty'],
+                    "types": ["any.empty"],
+                },
+                {
+                    "field": "body",
+                    "location": "body",
+                    "messages": ['"body" is not allowed to be empty'],
+                    "types": ["any.empty"],
+                },
+            ],
+        }
+
+        mock_response = MagicMock()
+        mock_response.status_code = status_code
+        mock_response.json.return_value = expected_response
+        mock_requests.post.return_value = mock_response
+        mock_authenticate.return_value = {"code": 200, "cookies": self.cookie_jar}
+
+        api = LoadingApiWrapper("test@email.com", "password")
+        response = api.create_thread(
+            title="",
+            message="",
+            category_name="other",
+        )
+
+        self.assertEqual(response, expected_response)
+
+    @patch("loading_api_wrapper.api.requests")
+    def test_create_thread_failure_no_auth_token(self, mock_requests):
+        status_code = 401
+        expected_response = {"code": status_code, "message": "No auth token"}
+
+        mock_response = MagicMock()
+        mock_response.status_code = status_code
+        mock_response.json.return_value = expected_response
+        mock_requests.post.return_value = mock_response
+
+        api = LoadingApiWrapper()
+        response = api.create_thread(
+            title="Hello",
+            message="My message",
+            category_name="other",
+        )
+
+        self.assertEqual(response, expected_response)