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()