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()