def test__datastore_lookup(datastore_pb2, runstate):
runstate.client = mock.Mock(project="theproject", spec=("project", ))
runstate.stub = mock.Mock(spec=("Lookup", ))
runstate.stub.return_value = mock.Mock(spec=("future", ))
_api._datastore_lookup(["foo", "bar"]) is runstate.stub.return_value
datastore_pb2.LookupRequest.assert_called_once_with(
project_id="theproject", keys=["foo", "bar"])
runstate.stub.Lookup.future.assert_called_once_with(
datastore_pb2.LookupRequest.return_value)
def test__datastore_lookup(datastore_pb2, context):
client = mock.Mock(project="theproject", spec=("project",))
stub = mock.Mock(spec=("Lookup",))
with context.new(client=client, stub=stub) as context:
context.stub.Lookup = Lookup = mock.Mock(spec=("future",))
future = Lookup.future.return_value
assert _api._datastore_lookup(["foo", "bar"], None) is future
datastore_pb2.LookupRequest.assert_called_once_with(
project_id="theproject", keys=["foo", "bar"], read_options=None
)
context.stub.Lookup.future.assert_called_once_with(
datastore_pb2.LookupRequest.return_value
)
def test__datastore_lookup(datastore_pb2, context):
client = mock.Mock(project="theproject", spec=("project", ))
stub = mock.Mock(spec=("Lookup", ))
with context.new(client=client, stub=stub).use() as context:
context.stub.Lookup = Lookup = mock.Mock(spec=("future", ))
future = tasklets.Future()
future.set_result("response")
Lookup.future.return_value = future
assert (_api._datastore_lookup(["foo", "bar"],
None).result() == "response")
datastore_pb2.LookupRequest.assert_called_once_with(
project_id="theproject", keys=["foo", "bar"], read_options=None)
context.stub.Lookup.future.assert_called_once_with(
datastore_pb2.LookupRequest.return_value)
