def test_created_related_actions_multiple_subscribers(self): collection_name = "sharing_collection" subscribers = [self.__subscriber_id, self.__second_subscriber] sharing_secret = self.__create_sharing_record(subscribers, collection_name) subscribers.append(self.__account_id) file = open("../test_resources/XooML.xml") manifest_action = UpdateSharedManifestAction(self.__account_id, collection_name, file) SharingActionFactory.create_related_sharing_actions(sharing_secret, manifest_action, callback=self.stop) sharing_action_list = self.wait() self.assertEqual(3, len(sharing_action_list)) for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() self.assertTrue(user_id in subscribers) subscribers.remove(user_id) self.assertEqual(SharingEvent.UPDATE_MANIFEST, sharing_action.get_action_type()) self.assertEqual(0, len(subscribers)) # cleanup SharingController.remove_sharing_record_by_secret(sharing_secret, callback=self.stop) self.wait() for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() collection_name = sharing_action.get_collection_name() StorageServer.remove_collection(user_id, collection_name, callback=self.stop) self.wait()
def test_create_related_actions_update_thumbnail(self): collection_name = "sharing_collection" subscribers = [self.__subscriber_id] sharing_secret = self.__create_sharing_record(subscribers, collection_name) subscribers.append(self.__account_id) file = open("../test_resources/XooML.xml") thumbnail_action = UpdateSharedThumbnailAction(self.__account_id, collection_name, file) SharingActionFactory.create_related_sharing_actions(sharing_secret, thumbnail_action, callback=self.stop) sharing_action_list = self.wait() self.assertEqual(2, len(sharing_action_list)) for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() thumbnail_file = sharing_action.get_associated_file() self.assertTrue(thumbnail_file is not None) self.assertTrue(user_id in subscribers) subscribers.remove(user_id) self.assertEqual(SharingEvent.UPDATE_THUMBNAIL, sharing_action.get_action_type()) self.assertEqual(0, len(subscribers)) # cleanup SharingController.remove_sharing_record_by_secret(sharing_secret, callback=self.stop) self.wait() for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() collection_name = sharing_action.get_collection_name() StorageServer.remove_collection(user_id, collection_name, callback=self.stop) self.wait()
def test_get_sharing_space_info_existing(self): servers = ['a', 'b', 'c'] Properties.sharing_space_servers = servers collection_name = str(uuid.uuid4()) subscriber_list = [self.__subscriber_id, self.__second_subscriber_id] sharing_secret, subscribers_collections =\ self.__create_sharing_record(subscriber_list, collection_name) load_balancer = SharingLoadBalancer.get_instance() load_balancer.get_sharing_space_info(sharing_secret, callback=self.stop) server_info = self.wait() server_adrs = server_info['server'] self.assertTrue(server_adrs in servers) load_balancer.get_sharing_space_info(sharing_secret, callback=self.stop) second_server_info = self.wait() second_server_adrs = second_server_info['server'] self.assertEqual(server_adrs, second_server_adrs) #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()
def test_create_related_actions_update_note_img(self): collection_name = "sharing_collection" subscribers = [self.__subscriber_id] sharing_secret = self.__create_sharing_record(subscribers, collection_name) subscribers.append(self.__account_id) note_name = "lala_note" file = open("../test_resources/XooML.xml") note_img_action = UpdateSharedNoteImageAction(self.__account_id, collection_name, note_name, file) SharingActionFactory.create_related_sharing_actions(sharing_secret, note_img_action, callback=self.stop) sharing_action_list = self.wait() self.assertEqual(2, len(sharing_action_list)) for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() actual_note_name = sharing_action.get_note_name() self.assertEqual(note_name, actual_note_name) self.assertTrue(user_id in subscribers) subscribers.remove(user_id) self.assertEqual(SharingEvent.UPDATE_NOTE_IMG, sharing_action.get_action_type()) self.assertEqual(0, len(subscribers)) # cleanup SharingController.remove_sharing_record_by_secret(sharing_secret, callback=self.stop) self.wait() for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() collection_name = sharing_action.get_collection_name() StorageServer.remove_collection(user_id, collection_name, callback=self.stop) self.wait()
def test_update_sharing_thumbnail(self): first_collection_name = 'shareable_collection_thumbnail' file = open('../test_resources/XooML.xml') StorageServer.add_collection(user_id=self.__account_id, collection_name=first_collection_name, callback=self.stop, file= file) response = self.wait() self.assertEqual(StorageResponse.OK, response) #add thumbnail file = open('../test_resources/note_img.jpg') StorageServer.add_thumbnail(self.__account_id, first_collection_name, file, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #update thumnail action file = open('../test_resources/note_img2.jpg') update_thumbnail_action = UpdateSharedThumbnailAction(self.__account_id, first_collection_name, file) update_thumbnail_action.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #cleanup StorageServer.remove_collection(self.__account_id, first_collection_name, callback=self.stop) self.wait()
def test_create_related_actions_delete_note(self): collection_name = "sharing_collection" subscribers = [self.__subscriber_id] sharing_secret = self.__create_sharing_record(subscribers, collection_name) subscribers.append(self.__account_id) note_name = "lala note" delete_action = DeleteSharedNoteAction(self.__account_id, collection_name, note_name) SharingActionFactory.create_related_sharing_actions(sharing_secret, delete_action, callback=self.stop) sharing_action_list = self.wait() self.assertEqual(2, len(sharing_action_list)) for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() actual_note_name = sharing_action.get_note_name() note_file = sharing_action.get_associated_file() self.assertTrue(note_file is None) self.assertEqual(note_name, actual_note_name) self.assertTrue(user_id in subscribers) subscribers.remove(user_id) self.assertEqual(SharingEvent.DELETE_NOTE, sharing_action.get_action_type()) self.assertEqual(0, len(subscribers)) # cleanup SharingController.remove_sharing_record_by_secret(sharing_secret, callback=self.stop) self.wait() for sharing_action in sharing_action_list: user_id = sharing_action.get_user_id() collection_name = sharing_action.get_collection_name() StorageServer.remove_collection(user_id, collection_name, callback=self.stop) self.wait()
def test_get_sharing_space_server_existing(self): collection_name = str(uuid.uuid4()) subscriber_list = [self.__subscriber_id, self.__second_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_adrs1 = self.wait(timeout=10000) self.assertIn(server_adrs1, Properties.sharing_space_servers) print server_adrs1 joker.get_sharing_space_server(sharing_secret, callback=self.stop) server_adrs2 = self.wait(timeout=10000) self.assertIn(server_adrs2, Properties.sharing_space_servers) self.assertEqual(server_adrs1, server_adrs2) print server_adrs2 #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()
def test_update_sharing_note(self): #create collection collection_name = 'shareable_collection' file = open('../test_resources/XooML.xml') StorageServer.add_collection(user_id=self.__account_id, collection_name=collection_name, callback=self.stop, file= file) response = self.wait() self.assertEqual(StorageResponse.OK, response) #add note note_name = "noteName" note_file = open('../test_resources/note.xml') StorageServer.add_note_to_collection(self.__account_id, collection_name, note_name, note_file, callback = self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #update note note_file = open('../test_resources/note2.xml') update_note_action = UpdateSharedNoteAction(self.__account_id, collection_name, note_name, note_file) update_note_action.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_copy_collections_between_accounts_different_name(self): first_collection_name = str(uuid.uuid1()) file = open("../test_resources/XooML.xml") StorageServer.add_collection( user_id=self.__account_id, collection_name=first_collection_name, callback=self.stop, file=file ) response = self.wait() self.assertEqual(StorageResponse.OK, response) second_collection_name = str(uuid.uuid1()) StorageServer.copy_collection_between_accounts( self.__account_id, self.__second_account_id, first_collection_name, second_collection_name, callback=self.stop, ) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageServer.list_collections(self.__second_account_id, callback=self.stop) col_list = self.wait() self.assertTrue(second_collection_name in col_list) # cleanup StorageServer.remove_collection(self.__account_id, first_collection_name, callback=self.stop) self.wait() StorageServer.remove_collection(self.__second_account_id, second_collection_name, callback=self.stop) self.wait()
def test_reshare_already_shared_collection(self): # create collection first_collection_name = "shareable_collection" file = open("../test_resources/XooML.xml") StorageServer.add_collection( user_id=self.__account_id, collection_name=first_collection_name, callback=self.stop, file=file ) response = self.wait() self.assertEqual(StorageResponse.OK, response) # create sharing record SharingController.create_sharing_record(self.__account_id, first_collection_name, callback=self.stop) sharing_secret = self.wait() self.assertTrue(sharing_secret is not None) # duplicate create sharing record SharingController.create_sharing_record(self.__account_id, first_collection_name, callback=self.stop) new_sharing_secret = self.wait() self.assertEqual(sharing_secret, new_sharing_secret) # verify that only one sharing record was added SharingController.get_sharing_record_by_owner_info(self.__account_id, first_collection_name, callback=self.stop) sharing_record = self.wait() subscribers_list = sharing_record.get_subscribers() self.assertEqual(1, len(subscribers_list)) # cleanup SharingController.remove_sharing_record_by_secret(sharing_secret, callback=self.stop) self.wait() StorageServer.remove_collection(self.__account_id, first_collection_name, callback=self.stop) self.wait()
def test_double_deleting_shared_note(self): note_name = 'note_name' collection_name = 'collection_name' #update note note_file = open('../test_resources/note2.xml') update_note_action = UpdateSharedNoteAction(self.__account_id, collection_name, note_name, note_file) update_note_action.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) delete_action = DeleteSharedNoteAction(self.__account_id, collection_name, note_name) delete_action.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) delete_action2 = DeleteSharedNoteAction(self.__account_id, collection_name, note_name) delete_action2.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.NOT_FOUND, response) #cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_add_thumbnail_invalid_collection(self): thumbnail = open("../test_resources/thumbnail.jpg") StorageServer.add_thumbnail(self.__account_id, "dummy", thumbnail, callback=self.stop) response = self.wait() # Accepted self.assertEqual(StorageResponse.OK, response) StorageServer.remove_collection(self.__account_id, "dummy", callback=self.stop) self.wait()
def test_remove_collection_with_no_file(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) StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response)
def test_rename_shared_collection_owner(self): # create collection first_collection_name = "shareable_collection" + str(random.randint) file = open("../test_resources/XooML.xml") StorageServer.add_collection( user_id=self.__account_id, collection_name=first_collection_name, callback=self.stop, file=file ) response = self.wait() self.assertEqual(StorageResponse.OK, response) # create sharing record SharingController.create_sharing_record(self.__account_id, first_collection_name, callback=self.stop) sharing_secret = self.wait() self.assertTrue(sharing_secret is not None) # subscribe SharingController.subscribe_to_sharing_space(self.__subscriber_id, sharing_secret, callback=self.stop) subscribers_collection_name = self.wait() self.assertTrue(subscribers_collection_name is not None) # rename new_collection_name = "new_name" + str(random.randint) SharingController.rename_shared_collection( self.__account_id, first_collection_name, new_collection_name, callback=self.stop ) response_code = self.wait() self.assertTrue(StorageResponse.OK, response_code) # verify SharingController.get_sharing_record_by_secret(sharing_secret, callback=self.stop) sharing_record = self.wait() renamed_collection_name = sharing_record.get_owner_collection_name() self.assertEqual(new_collection_name, renamed_collection_name) # verify owner collection SharingController.get_sharing_secret_from_subscriber_info( self.__account_id, new_collection_name, callback=self.stop ) renamed_sharing_secret = self.wait() self.assertEqual(sharing_secret, renamed_sharing_secret) # verify subscriber collection SharingController.get_sharing_record_from_subscriber_info( self.__subscriber_id, subscribers_collection_name, callback=self.stop ) renamed_sharing_record = self.wait() owner_collection_name = renamed_sharing_record.get_collection_name_for_subscriber(self.__account_id) self.assertEqual(new_collection_name, renamed_sharing_record.get_owner_collection_name()) self.assertEqual(new_collection_name, owner_collection_name) # cleanup SharingController.remove_sharing_record_by_secret(sharing_secret, callback=self.stop) self.wait() StorageServer.remove_collection(self.__account_id, first_collection_name, callback=self.stop) self.wait() StorageServer.remove_collection(self.__subscriber_id, subscribers_collection_name, callback=self.stop) self.wait()
def test_add_note_img_to_invalid_collection(self): # add image img_file = open("../test_resources/note_img.jpg") StorageServer.add_image_to_note(self.__account_id, "dummy", "dummy", img_file, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) # clean up StorageServer.remove_collection(self.__account_id, "dummy", callback=self.stop) self.wait()
def test_remove_collection_with_file(self): collection_name = str(uuid.uuid1()) file = open("../test_resources/XooML.xml") StorageServer.add_collection( user_id=self.__account_id, collection_name=collection_name, callback=self.stop, file=file ) response = self.wait(timeout=10) self.assertEqual(StorageResponse.OK, response) StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response)
def test_save_collection_manifest_invalid_collection(self): collection_name = "non_existing" collection_file = open("../test_resources/collection.xml") StorageServer.save_collection_manifest(self.__account_id, collection_name, collection_file, callback=self.stop) response = self.wait() # Accepted self.assertEqual(StorageResponse.OK, response) # clean up StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_list_collections_valid_account(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) StorageServer.list_collections(self.__account_id, callback=self.stop) collections = self.wait() self.assertTrue(len(collections) > 0) # cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_add_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) # cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_add_note_to_collection_no_file(self): collection_name = "col_name" note_name = "noteName" StorageServer.add_note_to_collection( self.__account_id, collection_name, note_name, note_file=None, callback=self.stop ) response = self.wait() self.assertEqual(StorageResponse.OK, response) # clean up StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_does_collection_exist(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) StorageServer.does_collection_exist(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertTrue(response) # cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_find_best_collection_name_with_one_existing_collection(self): collection_name = "name" StorageServer.add_collection(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageUtils.find_best_collection_name_for_user(collection_name, self.__account_id, callback=self.stop) new_name = self.wait() expected_new_name = collection_name + StorageUtils.SHARING_POSTFIX self.assertEqual(expected_new_name, new_name) # cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_list_all_notes_empty_collection(self): collection_name = "dummy" StorageServer.add_collection(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageServer.list_all_notes(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(0, len(response)) # clean up StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_add_note_to_non_existing_collection(self): collection_name = "col_name" note_name = "noteName" note_file = open("../test_resources/note.xml") StorageServer.add_note_to_collection( self.__account_id, collection_name, note_name, note_file, callback=self.stop ) response = self.wait() # Accepted self.assertEqual(StorageResponse.OK, response) # clean up StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_update_sharing_manifest_non_existing_manifest(self): collection_name = 'collection_name' #update manifest file = open('../test_resources/XooML2.xml') update_manifest_action = UpdateSharedManifestAction(self.__account_id, collection_name, file) update_manifest_action.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_get_note_img_invalid_note(self): # create note collection_name = "col_name" StorageServer.add_collection(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageServer.get_note_image(self.__account_id, collection_name, "dummy", callback=self.stop) response = self.wait() self.assertTrue(response is None) # clean up StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_delete_note(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) note_file = open('../test_resources/sharing_template1.xml') note_name = str(uuid.uuid4()) joker.update_note(server_adrs, sharing_secret, self.__account_id, collection_name, note_name, note_file, callback=self.stop) result = self.wait(timeout=1000) self.assertEqual(StorageResponse.OK,result) #wait for a while self.__wait(10) #now delete joker.delete_note(server_adrs, sharing_secret, self.__account_id, collection_name, note_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #wait for a while self.__wait(10) #Now Try to retreive the note StorageServer.get_note_from_collection(self.__account_id, collection_name, note_name, callback=self.stop) response = self.wait() self.assertTrue(response is 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()
def test_update_note_img_non_existing_img(self): collection_name = 'collection_name' note_name = 'note_name' #update note image img_file2 = open('../test_resources/note_img2.jpg') update_not_img_action = UpdateSharedNoteImageAction(self.__account_id, collection_name, note_name, img_file2) update_not_img_action.execute(callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) #cleanup StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()
def test_retrieve_renamed_collection(self): collection_name = str(uuid.uuid1()) new_collection_name = str(uuid.uuid1()) StorageServer.add_collection(self.__account_id, collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageServer.rename_collection(self.__account_id, collection_name, new_collection_name, callback=self.stop) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageServer.list_collections(self.__account_id, callback=self.stop) collections = self.wait() self.assertTrue(collection_name not in collections) self.assertTrue(new_collection_name in collections) # cleanup StorageServer.remove_collection(self.__account_id, new_collection_name, callback=self.stop) self.wait()
def test_get_note(self): collection_name = "col_name" note_name = "noteName" note_file = open("../test_resources/note.xml") StorageServer.add_note_to_collection( self.__account_id, collection_name, note_name, note_file, callback=self.stop ) response = self.wait() self.assertEqual(StorageResponse.OK, response) StorageServer.get_note_from_collection(self.__account_id, collection_name, note_name, callback=self.stop) response = self.wait() self.assertTrue(response is not None) # clean up StorageServer.remove_collection(self.__account_id, collection_name, callback=self.stop) self.wait()