41 lines
No EOL
1.9 KiB
Python
41 lines
No EOL
1.9 KiB
Python
import unittest
|
|
from trustcafeapiwrapper.wrappers.comment.create_comment import create_comment
|
|
class TestCreateComment(unittest.TestCase):
|
|
def setUp(self):
|
|
self.comment_text = "This is a test comment created via the create_comment wrapper function."
|
|
self.blur_label = None
|
|
|
|
def test_create_comment(self):
|
|
result = create_comment(
|
|
post_slug='1774951384-98fe38df',
|
|
comment_text=self.comment_text,
|
|
parent_path='/',
|
|
blur_label=self.blur_label
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertIn("job_function", result)
|
|
self.assertIn("payload", result)
|
|
self.assertEqual(result["job_function"], "comment.create")
|
|
self.assertEqual(result["payload"]["commentText"], self.comment_text)
|
|
self.assertEqual(result["payload"]["blurLabel"], self.blur_label)
|
|
self.assertEqual(result["payload"]["parent"]["pk"], "maintrunk#maintrunk")
|
|
self.assertEqual(result["payload"]["parent"]["sk"], "post#1774951384-98fe38df")
|
|
|
|
def test_create_comment_with_post_key(self):
|
|
post_key = {
|
|
"pk": "maintrunk#maintrunk",
|
|
"sk": "post#1774951384-98fe38df"
|
|
}
|
|
result = create_comment(
|
|
comment_text=self.comment_text,
|
|
post_key=post_key,
|
|
blur_label=self.blur_label
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertIn("job_function", result)
|
|
self.assertIn("payload", result)
|
|
self.assertEqual(result["job_function"], "comment.create")
|
|
self.assertEqual(result["payload"]["commentText"], self.comment_text)
|
|
self.assertEqual(result["payload"]["blurLabel"], self.blur_label)
|
|
self.assertEqual(result["payload"]["parent"]["pk"], post_key["pk"])
|
|
self.assertEqual(result["payload"]["parent"]["sk"], post_key["sk"]) |