def test_root_key_from_entity_key(self):
zookeeper = flexmock()
db_batch = flexmock()
db_batch.should_receive('valid_data_version').and_return(True)
db_batch.should_receive("batch_put_entity").and_return(None)
dd = DatastoreDistributed(db_batch, zookeeper)
self.assertEquals("test\x00blah\x00test_kind:bob\x01",
dd.get_root_key_from_entity_key("test\x00blah\x00test_kind:bob\x01nancy\x01"))
entity_proto1 = self.get_new_entity_proto("test", "test_kind", "nancy", "prop1name",
"prop2val", ns="blah")
self.assertEquals("test\x00blah\x00test_kind:nancy\x01",
dd.get_root_key_from_entity_key(entity_proto1.key()))
def test_root_key_from_entity_key(self):
zookeeper = flexmock()
db_batch = flexmock()
db_batch.should_receive('valid_data_version').and_return(True)
db_batch.should_receive("batch_put_entity").and_return(None)
dd = DatastoreDistributed(db_batch, zookeeper)
self.assertEquals("test\x00blah\x00test_kind:bob\x01",
dd.get_root_key_from_entity_key("test\x00blah\x00test_kind:bob\x01nancy\x01"))
entity_proto1 = self.get_new_entity_proto("test", "test_kind", "nancy", "prop1name",
"prop2val", ns="blah")
self.assertEquals("test\x00blah\x00test_kind:nancy\x01",
dd.get_root_key_from_entity_key(entity_proto1.key()))
def test_root_key_from_entity_key(self):
zk_client = flexmock()
zk_client.should_receive('add_listener')
zookeeper = flexmock(handle=zk_client)
db_batch = flexmock()
db_batch.should_receive('valid_data_version_sync').and_return(True)
transaction_manager = flexmock()
dd = DatastoreDistributed(db_batch, transaction_manager, zookeeper)
self.assertEquals(
"test\x00blah\x00test_kind:bob\x01",
dd.get_root_key_from_entity_key("test\x00blah\x00test_kind:bob\x01nancy\x01")
)
entity_proto1 = self.get_new_entity_proto(
"test", "test_kind", "nancy", "prop1name", "prop2val", ns="blah")
self.assertEquals(
"test\x00blah\x00test_kind:nancy\x01",
dd.get_root_key_from_entity_key(entity_proto1.key())
)
