import unittest from trustcafeapiwrapper.utils.get_post_pksk import get_post_pksk class TestGetPostPksk(unittest.TestCase): def test_get_post_pksk(self): parent_pksk = 'userprofile#johndoe' post_url = '/post/12345' expected_result = { "pk": 'userprofile#johndoe', "sk": 'post#12345' } self.assertEqual(get_post_pksk(parent_pksk, post_url), expected_result)