def setUpTestData(cls): super(TestProcessEntities, cls).setUpTestData() cls.profile = Mock() cls.post = entities.PostFactory() cls.retraction = base.Retraction(handle=cls.post.handle, target_guid=cls.post.guid, entity_type="Post")
def make_federable_retraction(obj: Union[Content, Profile], author: Optional[Profile] = None): """Make object retraction federable by converting it to a federation entity.""" logger.info("make_federable_retraction - Object: %s", obj) try: if isinstance(obj, Content): entity_type = { ContentType.REPLY: "Comment", ContentType.SHARE: "Share", ContentType.CONTENT: "Post", }.get(obj.content_type) actor_id = author.fid handle = author.handle elif isinstance(obj, Profile): entity_type = "Profile" actor_id = obj.fid handle = obj.handle else: logger.warning( "make_federable_retraction - Unknown object type %s", obj) return return base.Retraction( entity_type=entity_type, actor_id=actor_id, target_id=obj.fid, handle=handle, target_guid=obj.guid, ) except Exception as ex: logger.exception( "make_federable_retraction - Failed to convert %s: %s", obj.fid, ex)
def setUpTestData(cls): super().setUpTestData() cls.post = entities.PostFactory() cls.comment = base.Comment() cls.retraction = base.Retraction() cls.relationship = base.Relationship() cls.follow = base.Follow()
def test_removes_follower(self): self.remote_profile.following.add(self.profile) process_entity_retraction( base.Retraction(entity_type="Profile", _receiving_guid=self.profile.guid), self.remote_profile, ) self.assertEqual(self.remote_profile.following.count(), 0)
def setUpTestData(cls): super().setUpTestData() cls.post = entities.PostFactory() cls.comment = base.Comment() cls.retraction = base.Retraction() cls.follow = base.Follow() cls.profile = BaseProfileFactory() cls.share = BaseShareFactory() cls.receiving_user = UserFactory() cls.receiving_profile = cls.receiving_user.profile
def make_federable_retraction(content, author): """Make Content retraction federable by converting it to a federation entity.""" logging.info("make_federable_retraction - Content: %s" % content) try: return base.Retraction( entity_type="Comment" if content.parent else "Post", handle=author.handle, target_guid=content.guid, ) except Exception as ex: logger.exception( "make_federable_retraction - Failed to convert %s: %s", content.guid, ex) return None
def make_federable_retraction(content, author): """Make Content retraction federable by converting it to a federation entity.""" logging.info("make_federable_retraction - Content: %s" % content) try: return base.Retraction( entity_type= "Post", # Well, currently, at some point this could be something else handle=author.handle, target_guid=content.guid, ) except Exception as ex: logger.exception( "make_federable_retraction - Failed to convert %s: %s", content.guid, ex) return None
def make_federable_retraction(content, author): """Make Content retraction federable by converting it to a federation entity.""" logger.info("make_federable_retraction - Content: %s", content) try: entity_type = "Comment" if content.content_type == ContentType.REPLY else \ "Share" if content.content_type == ContentType.SHARE else "Post" return base.Retraction( # TODO check share federation entity_type=entity_type, handle=author.handle, target_guid=content.guid, ) except Exception as ex: logger.exception( "make_federable_retraction - Failed to convert %s: %s", content.guid, ex) return None