import unittest from trustcafeapiwrapper.wrappers.reaction.react import react class TestReact(unittest.TestCase): def test_react_to_post(self): reaction_type = 'like' parent_path = '/' item_path = '/post/12345-abcv' result = react(reaction_type, parent_path, item_path) self.assertIsInstance(result, dict) self.assertIn("job_function", result) self.assertIn("payload", result) self.assertEqual(result["job_function"], "reaction.reacttosomething") self.assertEqual(result["payload"]["reaction"], reaction_type) self.assertEqual(result["payload"]["parent"]["pk"], "maintrunk#maintrunk") self.assertEqual(result["payload"]["parent"]["sk"], "post#12345-abcv") self.assertEqual(result["payload"]["parent"]["entity"], "post") self.assertEqual(result["payload"]["parent"]["slug"], "12345-abcv") def test_react_to_comment(self): reaction_type = 'like' parent_path = '/post/12345-abcv' item_path = '/comment/67890' result = react(reaction_type, parent_path, item_path) self.assertIsInstance(result, dict) self.assertIn("job_function", result) self.assertIn("payload", result) self.assertEqual(result["job_function"], "reaction.reacttosomething") self.assertEqual(result["payload"]["reaction"], reaction_type) self.assertEqual(result["payload"]["parent"]["pk"], "post#12345-abcv") self.assertEqual(result["payload"]["parent"]["sk"], "comment#67890") self.assertEqual(result["payload"]["parent"]["entity"], "comment") self.assertEqual(result["payload"]["parent"]["slug"], "67890")