Add wrapper method edit_thread that works exactly the same as edit_post and update unit tests

Author: hnrkcode

loading_api_wrapper/api.py

@@ -273,3 +273,9 @@ 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 edit_thread(self, thread_id, message):
+        return self.edit_post(thread_id, message)

tests/test_loading_api_wrapper.py

@@ -1574,6 +1574,7 @@ def test_edit_post_success(self, mock_requests, mock_authenticate):
         mock_requests.patch.return_value = mock_response
         mock_authenticate.return_value = {"code": 200, "cookies": self.cookie_jar}
 
+        # Edit post.
         api = LoadingApiWrapper("test@email.com", "password")
         response = api.edit_post(
             post_id="000000000000000000000000",
@@ -1585,6 +1586,18 @@ def test_edit_post_success(self, mock_requests, mock_authenticate):
         self.assertEqual(response.get("code"), 200)
         self.assertDictEqual(response.get("data"), expected_response)
 
+        # Edit thread.
+        api = LoadingApiWrapper("test@email.com", "password")
+        response = api.edit_thread(
+            thread_id="000000000000000000000000",
+            message="updated message",
+        )
+
+        self.assertIsNotNone(api._cookies)
+        self.assertEqual(api._cookies, self.cookie_jar)
+        self.assertEqual(response.get("code"), 200)
+        self.assertDictEqual(response.get("data"), expected_response)
+
     @patch("loading_api_wrapper.api.requests")
     def test_edit_post_failure_no_auth_token(self, mock_requests):
         status_code = 401
@@ -1595,11 +1608,21 @@ def test_edit_post_failure_no_auth_token(self, mock_requests):
         mock_response.json.return_value = expected_response
         mock_requests.patch.return_value = mock_response
 
+        # Edit post.
         api = LoadingApiWrapper()
         response = api.edit_post(post_id="post_id_to_edit", message="updated message")
 
         self.assertEqual(response, expected_response)
 
+        # Edit thread.
+        api = LoadingApiWrapper()
+        response = api.edit_thread(
+            thread_id="thread_id_to_edit",
+            message="updated message",
+        )
+
+        self.assertEqual(response, expected_response)
+
     @patch("loading_api_wrapper.api.requests")
     def test_edit_post_failure_post_does_not_exist(self, mock_requests):
         status_code = 404
@@ -1610,9 +1633,20 @@ def test_edit_post_failure_post_does_not_exist(self, mock_requests):
         mock_response.json.return_value = expected_response
         mock_requests.patch.return_value = mock_response
 
+        # Edit post.
         api = LoadingApiWrapper()
         response = api.edit_post(
-            post_id="non_existing_post_id", message="new updated message"
+            post_id="non_existing_post_id",
+            message="new updated message",
+        )
+
+        self.assertEqual(response, expected_response)
+
+        # Edit thread.
+        api = LoadingApiWrapper()
+        response = api.edit_thread(
+            thread_id="non_existing_thread_id",
+            message="new updated message",
         )
 
         self.assertEqual(response, expected_response)
@@ -1626,13 +1660,22 @@ def test_edit_post_failure_empty_thread_id(self, mock_authenticate):
 
         mock_authenticate.return_value = {"code": 200, "cookies": self.cookie_jar}
 
+        # Edit post.
         api = LoadingApiWrapper("test@email.com", "password")
         response = api.edit_post(post_id="existing_post_id", message="")
 
         self.assertIsNotNone(api._cookies)
         self.assertEqual(api._cookies, self.cookie_jar)
         self.assertEqual(response, expected_response)
 
+        # Edit thread.
+        api = LoadingApiWrapper("test@email.com", "password")
+        response = api.edit_thread(thread_id="existing_thread_id", message="")
+
+        self.assertIsNotNone(api._cookies)
+        self.assertEqual(api._cookies, self.cookie_jar)
+        self.assertEqual(response, expected_response)
+
     @patch("loading_api_wrapper.api.LoadingApiWrapper._authenticate")
     def test_create_post_failure_empty_thread_id(self, mock_authenticate):
         expected_response = {