93 lines
No EOL
4.2 KiB
Python
93 lines
No EOL
4.2 KiB
Python
import unittest
|
|
from trustcafeapiwrapper.wrappers.post.update_post import update_post
|
|
class TestUpdatePost(unittest.TestCase):
|
|
def setUp(self):
|
|
self.post_text = "This is an updated test post created via the update_post wrapper function."
|
|
self.blur_label = None
|
|
self.card_url = None
|
|
self.collaborative = False
|
|
|
|
def test_update_post(self):
|
|
result = update_post(
|
|
parent_path='/',
|
|
post_path='/post/1235-abcv',
|
|
post_text=self.post_text,
|
|
blur_label=self.blur_label,
|
|
card_url=self.card_url,
|
|
collaborative=self.collaborative
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertIn("job_function", result)
|
|
self.assertIn("payload", result)
|
|
self.assertEqual(result["job_function"], "post.update")
|
|
self.assertEqual(result["payload"]["postText"], self.post_text)
|
|
self.assertEqual(result["payload"]["postSlug"], "1235-abcv")
|
|
self.assertEqual(result["payload"]["key"]["pk"], "maintrunk#maintrunk")
|
|
self.assertEqual(result["payload"]["key"]["sk"], "post#1235-abcv")
|
|
self.assertNotIn("slug", result["payload"]["key"])
|
|
|
|
def test_update_post_with_post_key(self):
|
|
post_key = {
|
|
"pk": "maintrunk#maintrunk",
|
|
"sk": "post#1235-abcv"
|
|
}
|
|
result = update_post(
|
|
post_key=post_key,
|
|
post_text=self.post_text,
|
|
blur_label=self.blur_label,
|
|
card_url=self.card_url,
|
|
collaborative=self.collaborative,
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertIn("job_function", result)
|
|
self.assertIn("payload", result)
|
|
self.assertEqual(result["job_function"], "post.update")
|
|
self.assertEqual(result["payload"]["postText"], self.post_text)
|
|
self.assertEqual(result["payload"]["postSlug"], "1235-abcv")
|
|
self.assertEqual(result["payload"]["key"]["pk"], post_key["pk"])
|
|
self.assertEqual(result["payload"]["key"]["sk"], post_key["sk"])
|
|
self.assertNotIn("slug", result["payload"]["key"])
|
|
|
|
def test_update_collaborative_post(self):
|
|
collaborative = True
|
|
result = update_post(
|
|
parent_path='/',
|
|
post_path='/post/1235-abcv',
|
|
post_text=self.post_text,
|
|
blur_label=self.blur_label,
|
|
card_url=self.card_url,
|
|
collaborative=collaborative
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertIn("job_function", result)
|
|
self.assertIn("payload", result)
|
|
self.assertEqual(result["job_function"], "post.update")
|
|
self.assertEqual(result["payload"]["postText"], self.post_text)
|
|
self.assertEqual(result["payload"]["postSlug"], "1235-abcv")
|
|
self.assertEqual(result["payload"]["key"]["pk"], "maintrunk#maintrunk")
|
|
self.assertEqual(result["payload"]["key"]["sk"], "post#1235-abcv")
|
|
self.assertNotIn("slug", result["payload"]["key"])
|
|
self.assertTrue(result["payload"]["collaborative"])
|
|
|
|
def test_update_blur_label_and_card_url(self):
|
|
blur_label = "Sensitive Content"
|
|
card_url = "https://example.com/card"
|
|
result = update_post(
|
|
parent_path='/',
|
|
post_path='/post/1235-abcv',
|
|
post_text=self.post_text,
|
|
blur_label=blur_label,
|
|
card_url=card_url,
|
|
collaborative=self.collaborative
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertIn("job_function", result)
|
|
self.assertIn("payload", result)
|
|
self.assertEqual(result["job_function"], "post.update")
|
|
self.assertEqual(result["payload"]["postText"], self.post_text)
|
|
self.assertEqual(result["payload"]["postSlug"], "1235-abcv")
|
|
self.assertEqual(result["payload"]["key"]["pk"], "maintrunk#maintrunk")
|
|
self.assertEqual(result["payload"]["key"]["sk"], "post#1235-abcv")
|
|
self.assertNotIn("slug", result["payload"]["key"])
|
|
self.assertEqual(result["payload"]["blurLabel"], blur_label)
|
|
self.assertEqual(result["payload"]["cardUrl"], card_url) |