def test_get_request_states(self): request_state.start_request('1') request_state.start_request('2') request_state.start_request('3') self.assertEqual(3, len(request_state.get_request_states())) self.assertItemsEqual( [request_state.get_request_state(request_id) for request_id in ['1', '2', '3']], request_state.get_request_states())
def _start_new_thread(target, args, kw=None): if kw is None: kw = {} request_id = remote_api_stub.RemoteStub._GetRequestId() request = request_state.get_request_state(request_id) def _run(): try: remote_api_stub.RemoteStub._SetRequestId(request_id) request.start_thread() target(*args, **kw) finally: request_environment.current_request.Clear() request.end_thread() return base_start_new_thread(_run, ())
def _start_new_thread(target, args, kw=None): if kw is None: kw = {} request_id = remote_api_stub.RemoteStub._GetRequestId() if request_id is None: return base_start_new_thread(target, args, kw) request = request_state.get_request_state(request_id) def _run(): try: remote_api_stub.RemoteStub._SetRequestId(request_id) request.start_thread() target(*args, **kw) finally: request_environment.current_request.Clear() request.end_thread() return base_start_new_thread(_run, ())
def test_start_request_function(self): request_state.start_request('id') self.assertEqual(1, len(request_state.get_request_states())) self.assertEqual('id', request_state.get_request_state('id').request_id)