def test_get_thumbnail(self):
collection_name = str(uuid.uuid1())
StorageServer.add_collection(self.__account_id, collection_name, callback=self.stop)
response = self.wait()
self.assertEqual(StorageResponse.OK, response)
thumbnail = open("../test_resources/thumbnail.jpg")
StorageServer.add_thumbnail(self.__account_id, collection_name, thumbnail, callback=self.stop)
response = self.wait()
self.assertEqual(StorageResponse.OK, response)
StorageServer.get_thumbnail(self.__account_id, collection_name, callback=self.stop)
response = self.wait()
result_file = open("../test_resources/thumbnail2.jpg", "w")
result_file.write(response.read())
response.close()
# cleanup
StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop)
self.wait()
def test_update_thumbnail(self):
collection_name = str(uuid.uuid4())
subscriber_list = [self.__subscriber_id]
sharing_secret, subscribers_collections =\
self.__create_sharing_record(subscriber_list, collection_name)
joker = JokerHelper.get_instance()
joker.get_sharing_space_server(sharing_secret, callback=self.stop)
server_adrs = self.wait(timeout=10000)
self.assertIn(server_adrs, Properties.sharing_space_servers)
thumbnail_file = open('../test_resources/note_img.jpg')
joker.update_thumbnail(server_adrs, sharing_secret, self.__account_id,
collection_name, thumbnail_file, callback=self.stop)
response = self.wait()
self.assertEqual(StorageResponse.OK, response)
#wait for a while
self.__wait(10)
#try to retreive the thumbnail
StorageServer.get_thumbnail(self.__account_id, collection_name,
callback=self.stop)
response = self.wait()
self.assertTrue(response is not None)
#cleanup
SharingController.remove_sharing_record_by_secret(sharing_secret, callback =self.stop)
self.wait()
StorageServer.remove_collection(self.__account_id, collection_name,
callback=self.stop)
for subscriber_id in subscribers_collections:
subscriber_collection = subscribers_collections[subscriber_id]
StorageServer.remove_collection(subscriber_id, subscriber_collection,
callback=self.stop)
self.wait()
