class TestSunspearClient(object):
def setUp(self):
backend = RiakBackend(**riak_connection_options)
self._backend = backend
self._client = SunspearClient(backend)
def test_create_obj(self):
self._backend._objects.get('1234').delete()
obj = {"objectType": "Hello", "id": "1234", "published": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"}
actstream_obj = self._client.create_object(obj)
ok_(self._backend._objects.get(key=obj['id']).exists)
eq_(actstream_obj, obj)
def test_create_activity(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"
actor = {"objectType": "something", "id": actor_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
act_obj = self._client.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
act_obj_dict = act_obj
ok_(self._backend._activities.get(key=act_obj['id']).exists)
eq_(act_obj_dict['actor'], actor)
eq_(act_obj_dict['object'], obj)
def test_create_reply_with_extra_data(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
actor2_id = '1234'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow()
actor = {"objectType": "something", "id": actor_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
extra = {'published': datetime.datetime(year=2012, month=1, day=1), 'foo': 'bar'}
#create the activity
self._backend.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
#now create a reply for the activity
reply_activity_dict, activity_obj_dict = self._client.create_reply(5, actor2_id, "This is a reply.", extra=extra)
eq_(extra['published'].strftime('%Y-%m-%dT%H:%M:%S') + "Z", reply_activity_dict['published'])
eq_(extra['foo'], reply_activity_dict['foo'])
eq_(reply_activity_dict['actor']['id'], actor2_id)
eq_(reply_activity_dict['verb'], 'reply')
eq_(activity_obj_dict['replies']['totalItems'], 1)
eq_(activity_obj_dict['replies']['items'][0]['object']['id'], reply_activity_dict['id'])
eq_(activity_obj_dict['replies']['items'][0]['verb'], 'reply')
eq_(activity_obj_dict['replies']['items'][0]['actor']['id'], actor2_id)
def test_create_like(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
actor2_id = '4321'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(actor2_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow()
actor = {"objectType": "something", "id": actor_id, "published": published_time}
actor2 = {"objectType": "something", "id": actor2_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
self._backend.create_obj(actor2)
#create the activity
self._backend.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
#now create a reply for the activity
like_activity_dict, activity_obj_dict = self._client.create_like(5, actor2_id)
eq_(like_activity_dict['actor']['id'], actor2_id)
eq_(like_activity_dict['verb'], 'like')
eq_(activity_obj_dict['likes']['totalItems'], 1)
eq_(activity_obj_dict['likes']['items'][0]['object']['id'], like_activity_dict['id'])
eq_(activity_obj_dict['likes']['items'][0]['verb'], 'like')
eq_(activity_obj_dict['likes']['items'][0]['actor']['id'], actor2_id)
def test_delete_activity(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"
actor = {"objectType": "something", "id": actor_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
act_obj = self._backend.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
act_obj_dict = act_obj
eq_(act_obj_dict['actor'], actor)
eq_(act_obj_dict['object'], obj)
self._client.delete_activity(act_obj_dict)
ok_(not self._backend._activities.get(key=act_obj_dict['id']).exists)
def test_delete_activity_with_like(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
actor2_id = '4321'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow()
actor = {"objectType": "something", "id": actor_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
#create the activity
self._backend.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
#now create a reply for the activity
like_activity_dict, activity_obj_dict = self._backend.create_sub_activity(
5, actor2_id, "", sub_activity_verb='like')
eq_(like_activity_dict['actor']['id'], actor2_id)
eq_(like_activity_dict['verb'], 'like')
eq_(activity_obj_dict['likes']['totalItems'], 1)
eq_(activity_obj_dict['likes']['items'][0]['object']['id'], like_activity_dict['id'])
eq_(activity_obj_dict['likes']['items'][0]['verb'], 'like')
eq_(activity_obj_dict['likes']['items'][0]['actor']['id'], actor2_id)
#now delete the activity and make sure the like gets deleted too:
self._client.delete_like(like_activity_dict)
ok_(not self._backend._activities.get(like_activity_dict['id']).exists)
ok_(not self._backend._activities.get(activity_obj_dict['likes']['items'][0]['id']).exists)
def test_delete_like(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
actor2_id = '4321'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow()
actor = {"objectType": "something", "id": actor_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
#create the activity
self._backend.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
#now create a reply for the activity
like_activity_dict, activity_obj_dict = self._backend.create_sub_activity(
5, actor2_id, "", sub_activity_verb='like')
eq_(like_activity_dict['actor']['id'], actor2_id)
eq_(like_activity_dict['verb'], 'like')
eq_(activity_obj_dict['likes']['totalItems'], 1)
eq_(activity_obj_dict['likes']['items'][0]['object']['id'], like_activity_dict['id'])
eq_(activity_obj_dict['likes']['items'][0]['verb'], 'like')
eq_(activity_obj_dict['likes']['items'][0]['actor']['id'], actor2_id)
#now delete the like and make sure everything is ok:
returned_updated_activity = self._client.delete_like(like_activity_dict['id'])
activity_obj_dict = self._backend._activities.get('5').data
ok_('likes' not in returned_updated_activity)
ok_('likes' not in activity_obj_dict)
def test_reply_delete(self):
self._backend._activities.get('5').delete()
actor_id = '1234'
actor2_id = '1234'
object_id = '4353'
#make sure these 2 keys don't exist anymore
self._backend._objects.get(actor_id).delete()
self._backend._objects.get(object_id).delete()
published_time = datetime.datetime.utcnow()
actor = {"objectType": "something", "id": actor_id, "published": published_time}
obj = {"objectType": "something", "id": object_id, "published": published_time}
#create the activity
self._backend.create_activity({"id": 5, "title": "Stream Item", "verb": "post", "actor": actor, "object": obj})
#now create a reply for the activity
reply_activity_dict, activity_obj_dict = self._backend.sub_activity_create(
5, actor2_id, "This is a reply.", sub_activity_verb='reply')
eq_(reply_activity_dict['actor']['id'], actor['id'])
eq_(reply_activity_dict['verb'], 'reply')
eq_(activity_obj_dict['replies']['totalItems'], 1)
eq_(activity_obj_dict['replies']['items'][0]['object']['id'], reply_activity_dict['id'])
eq_(activity_obj_dict['replies']['items'][0]['verb'], 'reply')
eq_(activity_obj_dict['replies']['items'][0]['actor']['id'], actor2_id)
#now delete the reply and make sure everything is ok:
returned_updated_activity = self._client.delete_reply(reply_activity_dict['id'])
activity_obj_dict = self._backend._activities.get('5').data
ok_('replies' not in returned_updated_activity)
ok_('replies' not in activity_obj_dict)
def test_get_objects(self):
obj1_id = '1111'
obj2_id = '1112'
obj3_id = '1113'
obj4_id = '1114'
self._backend._objects.get(obj1_id).delete()
self._backend._objects.get(obj2_id).delete()
self._backend._objects.get(obj3_id).delete()
self._backend._objects.get(obj4_id).delete()
obj1 = {
"objectType": "Hello",
"id": obj1_id,
"published": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"
}
obj2 = {
"objectType": "Hello",
"id": obj2_id,
"published": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"
}
obj3 = {
"objectType": "Hello",
"id": obj3_id,
"published": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"
}
obj4 = {
"objectType": "Hello",
"id": obj4_id,
"published": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S') + "Z"
}
self._backend.create_obj(obj1)
self._backend.create_obj(obj2)
self._backend.create_obj(obj3)
self._backend.create_obj(obj4)
objects = self._client.get_objects(object_ids=[obj1_id, obj2_id, obj3_id, obj4_id, 'xxx'])
eq_(len(objects), 4)
for obj in objects:
ok_(obj['id'] in [obj1_id, obj2_id, obj3_id, obj4_id])
