本文整理汇总了Python中appscale.datastore.datastore_distributed.DatastoreDistributed.fetch_keys方法的典型用法代码示例。如果您正苦于以下问题：Python DatastoreDistributed.fetch_keys方法的具体用法？Python DatastoreDistributed.fetch_keys怎么用？Python DatastoreDistributed.fetch_keys使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类appscale.datastore.datastore_distributed.DatastoreDistributed的用法示例。
在下文中一共展示了DatastoreDistributed.fetch_keys方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: testFetchKeys
# 需要导入模块: from appscale.datastore.datastore_distributed import DatastoreDistributed [as 别名]
# 或者: from appscale.datastore.datastore_distributed.DatastoreDistributed import fetch_keys [as 别名]
def testFetchKeys(self):
entity_proto1 = self.get_new_entity_proto("test", "test_kind", "bob", "prop1name",
"prop1val", ns="blah")
db_batch = flexmock()
db_batch.should_receive('valid_data_version').and_return(True)
db_batch.should_receive("batch_delete").and_return(None)
db_batch.should_receive("batch_put_entity").and_return(None)
db_batch.should_receive("batch_get_entity").and_return({'test\x00blah\x00test_kind:bob\x01':
{APP_ENTITY_SCHEMA[0]: entity_proto1.Encode(),
APP_ENTITY_SCHEMA[1]: 1}}).and_return({'test\x00blah\x00test_kind:bob\x01\x000000000002':
{JOURNAL_SCHEMA[0]: entity_proto1.Encode()}})
zk_client = flexmock()
zk_client.should_receive('add_listener')
zookeeper = flexmock(handle=zk_client)
zookeeper.should_receive("acquire_lock").and_return(True)
zookeeper.should_receive("release_lock").and_return(True)
transaction_manager = flexmock()
dd = DatastoreDistributed(db_batch, transaction_manager, zookeeper)
self.assertEquals(({'test\x00blah\x00test_kind:bob\x01':
{'txnID': 1, 'entity': entity_proto1.Encode()}
},
['test\x00blah\x00test_kind:bob\x01']),
dd.fetch_keys([entity_proto1.key()]))示例2: test_fetch_keys
# 需要导入模块: from appscale.datastore.datastore_distributed import DatastoreDistributed [as 别名]
# 或者: from appscale.datastore.datastore_distributed.DatastoreDistributed import fetch_keys [as 别名]
def test_fetch_keys(self):
entity_proto1 = self.get_new_entity_proto(
"test", "test_kind", "bob", "prop1name", "prop1val", ns="blah")
async_get_1 = gen.Future()
async_get_1.set_result({
'test\x00blah\x00test_kind:bob\x01': {
APP_ENTITY_SCHEMA[0]: entity_proto1.Encode(),
APP_ENTITY_SCHEMA[1]: 1
}
})
async_get_2 = gen.Future()
async_get_2.set_result({
'test\x00blah\x00test_kind:bob\x01\x000000000002': {
JOURNAL_SCHEMA[0]: entity_proto1.Encode()
}})
db_batch = flexmock()
db_batch.should_receive('valid_data_version_sync').and_return(True)
db_batch.should_receive("batch_delete").and_return(ASYNC_NONE)
db_batch.should_receive("batch_put_entity").and_return(ASYNC_NONE)
db_batch.should_receive("batch_get_entity").\
and_return(async_get_1).\
and_return(async_get_2)
zk_client = flexmock()
zk_client.should_receive('add_listener')
zookeeper = flexmock(handle=zk_client)
zookeeper.should_receive("acquire_lock").and_return(True)
zookeeper.should_receive("release_lock").and_return(True)
transaction_manager = flexmock()
dd = DatastoreDistributed(db_batch, transaction_manager, zookeeper)
fetched = yield dd.fetch_keys([entity_proto1.key()])
self.assertEquals(fetched[0], {
'test\x00blah\x00test_kind:bob\x01': {
'txnID': 1, 'entity': entity_proto1.Encode()
}
})
self.assertEqual(fetched[1], ['test\x00blah\x00test_kind:bob\x01'])