class TestSunnspearClientActivities(object): def setUp(self): self._backend = RiakBackend(**riak_connection_options) self._client = SunspearClient(self._backend) self.actor_id = '1234' self.actor_id2 = '4321' self.actor_id3 = '31415' self.object_id = '4353' self.object_id2 = '7654' self.object_id3 = '7655' self.reply_obj_id = '9999' self.reply_obj_id2 = '9998' self.reply_activity_id = '8889' self.reply_activity_id2 = '8888' self.like_obj_id = '6669' self.like_obj_id2 = '6669' self.like_activity_id = '7779' self.like_activity_id2 = '7778' self.activity_id = '5555' self.activity_id2 = '5556' self.reply_1 = { 'objectType': 'reply', 'id': self.reply_obj_id, 'published': '2012-08-05T12:00:00Z', 'content': 'This is my first reply', 'inReplyTo': [], } self.reply_2 = { 'objectType': 'reply', 'id': self.reply_obj_id2, 'published': '2012-08-05T12:05:00Z', 'content': 'This is my second reply', 'inReplyTo': [], } self.like_1 = { 'objectType': 'like', 'id': self.like_obj_id, 'published': '2012-08-05T12:00:00Z', 'inReplyTo': [], } self.like_2 = { 'objectType': 'like', 'id': self.like_obj_id2, 'published': '2012-08-05T12:00:00Z', 'inReplyTo': [], } self.reply_activity_1 = { 'actor': self.actor_id, 'object': self.reply_obj_id, 'target': self.actor_id3, 'verb': 'reply', 'id': self.reply_activity_id, } self.reply_activity_2 = { 'actor': self.actor_id2, 'object': self.reply_obj_id2, 'target': self.actor_id, 'verb': 'reply', 'id': self.reply_activity_id2, } self.like_activity_1 = { 'actor': self.actor_id, 'object': self.like_obj_id, 'target': self.actor_id3, 'verb': 'like', 'id': self.like_activity_id, } self.like_activity_2 = { 'actor': self.actor_id, 'object': self.like_obj_id2, 'target': self.actor_id3, 'verb': 'like', 'id': self.like_activity_id2, } self.activity_1 = { "id": self.activity_id, "title": "Stream Item", "verb": "post", "actor": self.actor_id2, "object": self.object_id, 'replies': { 'totalItems': 2, 'items': [ {'actor': self.actor_id, 'verb': 'reply', 'object': {'objectType': 'activity', 'id': self.reply_activity_id}}, {'actor': self.actor_id, 'verb': 'reply', 'object': {'objectType': 'activity', 'id': self.reply_activity_id2}}, ] }, } self.activity_2 = { "id": self.activity_id2, "title": "Stream Item", "verb": "post", "actor": self.actor_id2, "object": self.object_id, 'replies': { 'totalItems': 2, 'items': [ {'actor': self.actor_id, 'verb': 'reply', 'object': {'objectType': 'activity', 'id': self.reply_activity_id}}, {'actor': self.actor_id, 'verb': 'reply', 'object': {'objectType': 'activity', 'id': self.reply_activity_id2}}, ] }, "likes": { 'totalItems': 1, 'items': [ {'actor': self.actor_id, 'verb': 'like', 'object': {'objectType': 'activity', 'id': self.like_activity_id}}, ], }, } self.actor = {"objectType": "something", "id": self.actor_id, "published": '2012-07-05T12:00:00Z'} self.actor2 = {"objectType": "something", "id": self.actor_id2, "published": '2012-07-05T12:00:00Z'} self.actor3 = {"objectType": "something", "id": self.actor_id3, "published": '2012-07-05T12:00:00Z'} self.obj = {"objectType": "something", "id": self.object_id, "published": '2012-07-05T12:00:00Z'} self.obj2 = {"objectType": "something", "id": self.object_id2, "published": '2012-07-05T12:00:00Z'} self.obj3 = {"objectType": "something", "id": self.object_id3, "published": '2012-07-05T12:00:00Z', 'inReplyTo': [{'objectType': 'activity', 'id': self.reply_activity_id}]} self._backend._objects.get(self.actor_id).delete() self._backend._objects.get(self.actor_id2).delete() self._backend._objects.get(self.actor_id3).delete() self._backend._objects.get(self.object_id).delete() self._backend._objects.get(self.object_id2).delete() self._backend._objects.get(self.object_id3).delete() self._backend._objects.get(self.reply_obj_id).delete() self._backend._objects.get(self.reply_obj_id2).delete() self._backend._objects.get(self.like_obj_id).delete() self._backend._objects.get(self.like_obj_id2).delete() self._backend._activities.get(self.reply_activity_id).delete() self._backend._activities.get(self.reply_activity_id2).delete() self._backend._activities.get(self.like_activity_id).delete() self._backend._activities.get(self.activity_id).delete() self._backend._activities.get(self.activity_id2).delete() obj1 = self._backend._objects.new(key=self.actor["id"]) obj1.data = self.actor obj1.store() obj2 = self._backend._objects.new(key=self.actor2["id"]) obj2.data = self.actor2 obj2.store() obj3 = self._backend._objects.new(key=self.actor3["id"]) obj3.data = self.actor3 obj3.store() obj4 = self._backend._objects.new(key=self.obj["id"]) obj4.data = self.obj obj4.store() obj5 = self._backend._objects.new(key=self.obj2["id"]) obj5.data = self.obj2 obj5.store() obj6 = self._backend._objects.new(key=self.obj3["id"]) obj6.data = self.obj3 obj6.store() obj7 = self._backend._objects.new(key=self.reply_1["id"]) obj7.data = self.reply_1 obj7.store() obj8 = self._backend._objects.new(key=self.reply_2["id"]) obj8.data = self.reply_2 obj8.store() obj9 = self._backend._objects.new(key=self.like_1["id"]) obj9.data = self.like_1 obj9.store() obj10 = self._backend._objects.new(key=self.like_2["id"]) obj10.data = self.like_2 obj10.store() obj11 = self._backend._activities.new(key=self.reply_activity_1["id"]) obj11.data = self.reply_activity_1 obj11.store() obj12 = self._backend._activities.new(key=self.reply_activity_2["id"]) obj12.data = self.reply_activity_2 obj12.store() obj13 = self._backend._activities.new(key=self.like_activity_1["id"]) obj13.data = self.like_activity_1 obj13.store() obj14 = self._backend._activities.new(key=self.like_activity_2["id"]) obj14.data = self.like_activity_2 obj14.store() obj15 = self._backend._activities.new(key=self.activity_1["id"]) obj15.data = self.activity_1 obj15.store() obj16 = self._backend._activities.new(key=self.activity_2["id"]) obj16.data = self.activity_2 obj16.store() def test_get_activities_with_replies(self): expected = [ {"id": self.activity_id, "title": "Stream Item", "verb": "post", "actor": self.actor2, "object": self.obj, 'replies': { 'totalItems': 2, 'items': [ { 'actor': self.actor, 'verb': 'reply', 'object': { 'objectType': 'activity', 'actor': self.actor, 'object': self.reply_1, 'target': self.actor3, 'verb': 'reply', 'id': self.reply_activity_id, } }, { 'actor': self.actor, 'verb': 'reply', 'object': { 'objectType': 'activity', 'actor': self.actor2, 'object': self.reply_2, 'target': self.actor, 'verb': 'reply', 'id': self.reply_activity_id2, } }, ] }, }, ] result = self._client.get_activities(activity_ids=[self.activity_id]) eq_(result, expected) def test_get_activities_with_likes_and_replies(self): expected = [ {"id": self.activity_id2, "title": "Stream Item", "verb": "post", "actor": self.actor2, "object": self.obj, 'replies': { 'totalItems': 2, 'items': [ { 'actor': self.actor, 'verb': 'reply', 'object': { 'objectType': 'activity', 'actor': self.actor, 'object': self.reply_1, 'target': self.actor3, 'verb': 'reply', 'id': self.reply_activity_id, } }, { 'actor': self.actor, 'verb': 'reply', 'object': { 'objectType': 'activity', 'actor': self.actor2, 'object': self.reply_2, 'target': self.actor, 'verb': 'reply', 'id': self.reply_activity_id2, } }, ] }, 'likes': { 'totalItems': 1, 'items': [ { 'actor': self.actor, 'verb': 'like', 'object': { 'objectType': 'activity', 'actor': self.actor, 'object': self.like_1, 'target': self.actor3, 'verb': 'like', 'id': self.like_activity_id, } }, ] } }, ] result = self._client.get_activities(activity_ids=[self.activity_id2]) eq_(result, expected) def test_get_activities_with_aggregation_pipline(self): activity_ids = [self.like_activity_id, self.reply_activity_id, self.activity_id, self.activity_id2, self.like_activity_id2, self.reply_activity_id2] activities = self._client.get_activities(activity_ids=activity_ids, aggregation_pipeline=[PropertyAggregator(properties=['verb', 'actor'])]) eq_([{u'id': u'7779', u'verb': u'like', u'target': {u'objectType': u'something', u'id': u'31415', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'inReplyTo': [], u'objectType': u'like', u'id': u'6669', u'published': u'2012-08-05T12:00:00Z'}, u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}}, {u'id': u'8889', u'verb': u'reply', u'target': {u'objectType': u'something', u'id': u'31415', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'content': u'This is my first reply', u'inReplyTo': [], u'objectType': u'reply', u'id': u'9999', u'published': u'2012-08-05T12:00:00Z'}, u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}}, {'grouped_by_attributes': ['verb', 'actor'], u'title': [u'Stream Item', u'Stream Item'], u'object': [{u'objectType': u'something', u'id': u'4353', u'published': u'2012-07-05T12:00:00Z'}, {u'published': u'2012-07-05T12:00:00Z', u'id': u'4353', u'objectType': u'something'}], u'actor': {u'published': u'2012-07-05T12:00:00Z', u'id': u'4321', u'objectType': u'something'}, u'verb': u'post', u'replies': [{u'totalItems': 2, u'items': [{u'verb': u'reply', u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'target': {u'objectType': u'something', u'id': u'31415', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'content': u'This is my first reply', u'inReplyTo': [], u'objectType': u'reply', u'id': u'9999', u'published': u'2012-08-05T12:00:00Z'}, u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'verb': u'reply', u'id': u'8889', u'objectType': u'activity'}}, {u'verb': u'reply', u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'target': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'content': u'This is my second reply', u'inReplyTo': [], u'objectType': u'reply', u'id': u'9998', u'published': u'2012-08-05T12:05:00Z'}, u'actor': {u'objectType': u'something', u'id': u'4321', u'published': u'2012-07-05T12:00:00Z'}, u'verb': u'reply', u'id': u'8888', u'objectType': u'activity'}}]}, {u'totalItems': 2, u'items': [{u'verb': u'reply', u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'target': {u'objectType': u'something', u'id': u'31415', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'content': u'This is my first reply', u'inReplyTo': [], u'objectType': u'reply', u'id': u'9999', u'published': u'2012-08-05T12:00:00Z'}, u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'verb': u'reply', u'id': u'8889', u'objectType': u'activity'}}, {u'verb': u'reply', u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'target': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'content': u'This is my second reply', u'inReplyTo': [], u'objectType': u'reply', u'id': u'9998', u'published': u'2012-08-05T12:05:00Z'}, u'actor': {u'objectType': u'something', u'id': u'4321', u'published': u'2012-07-05T12:00:00Z'}, u'verb': u'reply', u'id': u'8888', u'objectType': u'activity'}}]}], u'id': [u'5555', u'5556'], 'grouped_by_values': [u'post', {u'published': u'2012-07-05T12:00:00Z', u'id': u'4321', u'objectType': u'something'}]}, {u'id': u'7778', u'verb': u'like', u'target': {u'objectType': u'something', u'id': u'31415', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'inReplyTo': [], u'objectType': u'like', u'id': u'6669', u'published': u'2012-08-05T12:00:00Z'}, u'actor': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}}, {u'id': u'8888', u'verb': u'reply', u'target': {u'objectType': u'something', u'id': u'1234', u'published': u'2012-07-05T12:00:00Z'}, u'object': {u'content': u'This is my second reply', u'inReplyTo': [], u'objectType': u'reply', u'id': u'9998', u'published': u'2012-08-05T12:05:00Z'}, u'actor': {u'objectType': u'something', u'id': u'4321', u'published': u'2012-07-05T12:00:00Z'}}], activities)