def test_delete_entities(self): app_id = 'test' entity_proto1 = self.get_new_entity_proto( app_id, "test_kind", "bob", "prop1name", "prop1val", ns="blah") row_key = "test\x00blah\x00test_kind:bob\x01" row_values = {row_key: {APP_ENTITY_SCHEMA[0]: entity_proto1.Encode(), APP_ENTITY_SCHEMA[1]: '1'}} zk_client = flexmock() zk_client.should_receive('add_listener') zookeeper = flexmock(handle=zk_client) zookeeper.should_receive("get_valid_transaction_id").and_return(1) db_batch = flexmock() db_batch.should_receive('valid_data_version').and_return(True) db_batch.should_receive("batch_get_entity").and_return(row_values) db_batch.should_receive('batch_mutate') db_batch.should_receive('_normal_batch') transaction_manager = flexmock() dd = DatastoreDistributed(db_batch, transaction_manager, zookeeper) row_keys = [entity_proto1.key()] dd.delete_entities(entity_proto1.key(), 1, row_keys)
def test_delete_entities(self): app_id = 'test' entity_proto1 = self.get_new_entity_proto( app_id, "test_kind", "bob", "prop1name", "prop1val", ns="blah") row_key = "test\x00blah\x00test_kind:bob\x01" row_values = {row_key: {APP_ENTITY_SCHEMA[0]: entity_proto1.Encode(), APP_ENTITY_SCHEMA[1]: '1'}} async_result = gen.Future() async_result.set_result(row_values) zk_client = flexmock() zk_client.should_receive('add_listener') zookeeper = flexmock(handle=zk_client) zookeeper.should_receive("get_valid_transaction_id").and_return(1) db_batch = flexmock() db_batch.should_receive('valid_data_version_sync').and_return(True) db_batch.should_receive("batch_get_entity").and_return(async_result) db_batch.should_receive('normal_batch').and_return(ASYNC_NONE) transaction_manager = flexmock() dd = DatastoreDistributed(db_batch, transaction_manager, zookeeper) row_keys = [entity_proto1.key()] yield dd.delete_entities(entity_proto1.key(), 1, row_keys)
def test_delete_entities(self): app_id = 'test' entity_proto1 = self.get_new_entity_proto( app_id, "test_kind", "bob", "prop1name", "prop1val", ns="blah") row_key = "test\x00blah\x00test_kind:bob\x01" row_values = {row_key: {APP_ENTITY_SCHEMA[0]: entity_proto1.Encode(), APP_ENTITY_SCHEMA[1]: '1'}} zookeeper = flexmock() zookeeper.should_receive("get_valid_transaction_id").and_return(1) zookeeper.should_receive("register_updated_key").and_return(1) db_batch = flexmock() db_batch.should_receive('valid_data_version').and_return(True) db_batch.should_receive("batch_get_entity").and_return(row_values) db_batch.should_receive('batch_mutate') db_batch.should_receive('_normal_batch') dd = DatastoreDistributed(db_batch, zookeeper) row_keys = [entity_proto1.key()] dd.delete_entities(entity_proto1.key(), 1, row_keys)
def test_delete_entities(self): app_id = 'test' entity_proto1 = self.get_new_entity_proto( app_id, "test_kind", "bob", "prop1name", "prop1val", ns="blah") row_key = "test\x00blah\x00test_kind:bob\x01" row_values = {row_key: {APP_ENTITY_SCHEMA[0]: entity_proto1.Encode(), APP_ENTITY_SCHEMA[1]: '1'}} zookeeper = flexmock() zookeeper.should_receive("get_valid_transaction_id").and_return(1) zookeeper.should_receive("register_updated_key").and_return(1) db_batch = flexmock() db_batch.should_receive('valid_data_version').and_return(True) db_batch.should_receive("batch_get_entity").and_return(row_values) db_batch.should_receive('batch_mutate') dd = DatastoreDistributed(db_batch, zookeeper) row_keys = [entity_proto1.key()] txn_hash = {row_key: 1} dd.delete_entities(app_id, row_keys, txn_hash)