def test_key(self): from gcloud.datastore.key import Key key = Key('PATH', 1234, dataset_id='DATASET') name, value = self._callFUT(key) self.assertEqual(name, 'key_value') self.assertEqual(value, key.to_protobuf())
def test_key(self): from gcloud.datastore.key import Key pb = self._makePB() key = Key('KIND', 1234, dataset_id='DATASET') self._callFUT(pb, key) value = pb.key_value self.assertEqual(value, key.to_protobuf())
def test_entity_w_key(self): from gcloud.datastore.entity import Entity from gcloud.datastore.key import Key pb = self._makePB() key = Key('KIND', 123, dataset_id='DATASET') entity = Entity(key=key) entity['foo'] = u'Foo' self._callFUT(pb, entity) value = pb.entity_value self.assertEqual(value.key, key.to_protobuf()) props = list(value.property) self.assertEqual(len(props), 1) self.assertEqual(props[0].name, 'foo') self.assertEqual(props[0].value.string_value, u'Foo')
def test_key(self): from gcloud.datastore._datastore_v1_pb2 import Value from gcloud.datastore.key import Key pb = Value() expected = Key('KIND', 1234, dataset_id='DATASET').to_protobuf() pb.key_value.CopyFrom(expected) found = self._callFUT(pb) self.assertEqual(found.to_protobuf(), expected)
def _make_key(kind, id, dataset_id): from gcloud.datastore._datastore_v1_pb2 import Key key = Key() key.partition_id.dataset_id = dataset_id elem = key.path_element.add() elem.kind = kind elem.id = id return key
def _makePB(self, dataset_id=None, namespace=None, path=()): from gcloud.datastore._datastore_v1_pb2 import Key pb = Key() if dataset_id is not None: pb.partition_id.dataset_id = dataset_id if namespace is not None: pb.partition_id.namespace = namespace for elem in path: added = pb.path_element.add() added.kind = elem['kind'] if 'id' in elem: added.id = elem['id'] if 'name' in elem: added.name = elem['name'] return pb
def __init__(self): from gcloud.datastore.key import Key self.key = Key('KIND', dataset_id='DATASET')