diff --git a/src/trustcafeapiwrapper/wrappers/follow/follow.py b/src/trustcafeapiwrapper/wrappers/follow/follow.py index c3c513a..65c2235 100644 --- a/src/trustcafeapiwrapper/wrappers/follow/follow.py +++ b/src/trustcafeapiwrapper/wrappers/follow/follow.py @@ -30,6 +30,7 @@ def follow( "sk": itemPKSK }, "followType": entity, + "parentSlug": parent_slug, "preferences": { "notification": True, "emailNew": False, diff --git a/testing.py b/testing.py index a3c6dde..0a0faf9 100644 --- a/testing.py +++ b/testing.py @@ -43,8 +43,8 @@ def save_response(response): # print("-----------Get a user profile----------------") -profile = API.run_job('userprofile.get', "simon-little") -print(profile) +# profile = API.run_job('userprofile.get', "simon-little") +# print(profile) # print("-------------- Get a branch -----------------") # branch = API.run_job('branch.get', "music") # print(branch) @@ -199,3 +199,11 @@ print(profile) # post_path="/post/1775143460-ef45186a", # parent_path="/", # ))) + + +from trustcafeapiwrapper.wrappers.follow.follow import follow +save_response(API.wrapped(follow( + entity='userprofile', + is_following=True, + parent_slug='alphaemail-test2' +))) \ No newline at end of file diff --git a/tests/wrappers/follow.py b/tests/wrappers/follow.py index 1994d71..af1f280 100644 --- a/tests/wrappers/follow.py +++ b/tests/wrappers/follow.py @@ -20,4 +20,24 @@ class TestFollow(unittest.TestCase): self.assertIn("followType", result["payload"]) self.assertIn("preferences", result["payload"]) self.assertEqual(result["payload"]["isFollowing"], True) - self.assertEqual(result["payload"]["parent"]["pk"], "userprofile#janedoe") \ No newline at end of file + self.assertEqual(result["payload"]["parentSlug"], "janedoe") + self.assertEqual(result["payload"]["parent"]["pk"], "userprofile#janedoe") + + def test_unfollow(self): + + result = follow( + entity='user', + is_following=False, + parent_slug='janedoe' + ) + + self.assertIsInstance(result, dict) + self.assertIn("job_function", result) + self.assertIn("payload", result) + self.assertEqual(result["job_function"], "follow.follow") + self.assertIsInstance(result["payload"], dict) + self.assertIn("isFollowing", result["payload"]) + self.assertIn("parent", result["payload"]) + self.assertIn("followType", result["payload"]) + self.assertIn("preferences", result["payload"]) + self.assertEqual(result["payload"]["isFollowing"], False) \ No newline at end of file