def test_it(self):
from google.cloud.datastore._generated import datastore_pb2
from google.cloud.datastore._generated import entity_pb2
index_updates = 1337
keys = [
entity_pb2.Key(path=[
entity_pb2.Key.PathElement(
kind='Foo',
id=1234,
),
], ),
entity_pb2.Key(path=[
entity_pb2.Key.PathElement(
kind='Bar',
name='baz',
),
], ),
]
response = datastore_pb2.CommitResponse(
mutation_results=[
datastore_pb2.MutationResult(key=key) for key in keys
],
index_updates=index_updates,
)
result = self._callFUT(response)
self.assertEqual(result, (index_updates, keys))
def _make_key(kind, id_, project):
from google.cloud.datastore._generated import entity_pb2
key = entity_pb2.Key()
key.partition_id.project_id = project
elem = key.path.add()
elem.kind = kind
elem.id = id_
return key
def to_protobuf(self):
from google.cloud.datastore._generated import entity_pb2
key = self._key = entity_pb2.Key()
# Don't assign it, because it will just get ripped out
# key.partition_id.project_id = self.project
element = key.path.add()
element.kind = self._kind
if self._id is not None:
element.id = self._id
return key
def _makePB(self, project=None, namespace=None, path=()):
from google.cloud.datastore._generated import entity_pb2
pb = entity_pb2.Key()
if project is not None:
pb.partition_id.project_id = project
if namespace is not None:
pb.partition_id.namespace_id = namespace
for elem in path:
added = pb.path.add()
added.kind = elem['kind']
if 'id' in elem:
added.id = elem['id']
if 'name' in elem:
added.name = elem['name']
return pb
def to_protobuf(self):
"""Return a protobuf corresponding to the key.
:rtype: :class:`google.cloud.datastore._generated.entity_pb2.Key`
:returns: The protobuf representing the key.
"""
key = _entity_pb2.Key()
key.partition_id.project_id = self.project
if self.namespace:
key.partition_id.namespace_id = self.namespace
for item in self.path:
element = key.path.add()
if 'kind' in item:
element.kind = item['kind']
if 'id' in item:
element.id = item['id']
if 'name' in item:
element.name = item['name']
return key
