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)
