import unittest
from trustcafeapiwrapper.wrappers.vote.votecast import votecast
class TestVoteCast(unittest.TestCase):
    def test_vote_cast_to_post(self):
        vote = 'up'
        parent_path = '/'
        item_path = '/post/12345'
        result = votecast(vote, parent_path, item_path)

        self.assertIsInstance(result, dict)
        self.assertIn("job_function", result)
        self.assertIn("payload", result)
        self.assertEqual(result["job_function"], "vote.votecast")
        self.assertEqual(result["payload"]["vote"], vote)
        self.assertEqual(result["payload"]["parent"]["pk"], "maintrunk#maintrunk")
        self.assertEqual(result["payload"]["parent"]["sk"], "post#12345")
        self.assertEqual(result["payload"]["parent"]["entity"], "post")
        self.assertEqual(result["payload"]["parent"]["slug"], "12345")

    def test_vote_cast_to_comment(self):
        vote = 'down'
        parent_path = '/post/12345'
        item_path = '/comment/67890'
        result = votecast(vote, parent_path, item_path)

        self.assertIsInstance(result, dict)
        self.assertIn("job_function", result)
        self.assertIn("payload", result)
        self.assertEqual(result["job_function"], "vote.votecast")
        self.assertEqual(result["payload"]["vote"], vote)
        self.assertEqual(result["payload"]["parent"]["pk"], "post#12345")
        self.assertEqual(result["payload"]["parent"]["sk"], "comment#67890")
        self.assertEqual(result["payload"]["parent"]["entity"], "comment")
        self.assertEqual(result["payload"]["parent"]["slug"], "67890")

    def test_vote_cast_with_parent_key(self):
        vote = 'up'
        parent_key = {
            'pk': 'post#12345',
            'sk': 'comment#67890'
        }

        result = votecast(vote, item_key=parent_key)    

        self.assertIsInstance(result, dict)
        self.assertIn("job_function", result)
        self.assertIn("payload", result)
        self.assertEqual(result["job_function"], "vote.votecast")
        self.assertEqual(result["payload"]["vote"], vote)
        self.assertEqual(result["payload"]["parent"]["pk"], "post#12345")
        self.assertEqual(result["payload"]["parent"]["sk"], "comment#67890")
        self.assertEqual(result["payload"]["parent"]["entity"], "comment")
        self.assertEqual(result["payload"]["parent"]["slug"], "67890")