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"])