def toplevel_wrapper(*args, **kwargs):
context = context_module.get_context()
try:
with context.new().use():
return synctasklet_wrapped(*args, **kwargs)
finally:
_eventloop.run()
def test_it(self):
options = {"foo": "bar"}
batch = _batch.get_batch(MockBatch, options)
assert batch.options is options
assert not batch.idle_called
different_options = {"food": "barn"}
assert _batch.get_batch(MockBatch, different_options) is not batch
assert _batch.get_batch(MockBatch) is not batch
assert _batch.get_batch(MockBatch, options) is batch
_eventloop.run()
assert batch.idle_called
def test__advance_tasklet_yields_rpc():
def generator_function(dependency):
value = yield dependency
return value + 3
dependency = mock.Mock(spec=grpc.Future)
dependency.exception.return_value = None
dependency.result.return_value = 8
generator = generator_function(dependency)
future = tasklets.TaskletFuture(generator)
future._advance_tasklet()
callback = dependency.add_done_callback.call_args[0][0]
callback(dependency)
_eventloop.run()
assert future.result() == 11
def test__advance_tasklet_yields_rpc(in_context):
def generator_function(dependency):
value = yield dependency
raise tasklets.Return(value + 3)
dependency = mock.Mock(spec=_remote.RemoteCall)
dependency.exception.return_value = None
dependency.result.return_value = 8
generator = generator_function(dependency)
future = tasklets._TaskletFuture(generator, in_context)
future._advance_tasklet()
callback = dependency.add_done_callback.call_args[0][0]
callback(dependency)
_eventloop.run()
assert future.result() == 11
def test_run(context):
loop = mock.Mock(spec=("run", ))
with context.new(eventloop=loop).use():
_eventloop.run()
loop.run.assert_called_once_with()
def test_run(EventLoop):
EventLoop.return_value = loop = unittest.mock.Mock(spec=("run", ))
with _runstate.state_context(None):
_eventloop.run()
loop.run.assert_called_once_with()
def test_run():
with pytest.raises(NotImplementedError):
eventloop.run()
def test_run(EventLoop):
EventLoop.return_value = loop = unittest.mock.Mock(spec=("run",))
with eventloop.async_context():
eventloop.run()
loop.run.assert_called_once_with()
def test_run(EventLoop):
EventLoop.return_value = loop = unittest.mock.Mock(spec=("run",))
with _runstate.ndb_context():
_eventloop.run()
loop.run.assert_called_once_with()
