def test_it(self):
from google.cloud.datastore_v1.proto import datastore_pb2
from google.cloud.datastore_v1.proto 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._call_fut(response)
self.assertEqual(result, (index_updates, keys))
def _make_mutation(id_):
from google.cloud.datastore_v1.proto import datastore_pb2
from google.cloud.datastore_v1.proto import entity_pb2
key = entity_pb2.Key()
key.partition_id.project_id = 'PROJECT'
elem = key.path.add()
elem.kind = 'Kind'
elem.id = id_
return datastore_pb2.MutationResult(key=key)
def _make_commit_response(*keys):
from google.cloud.datastore_v1.proto import datastore_pb2
mutation_results = [
datastore_pb2.MutationResult(key=key) for key in keys]
return datastore_pb2.CommitResponse(mutation_results=mutation_results)