def __call__(self, crosstown_task=None):
self.crosstown_task = crosstown_task
try:
self.response = get_response_for_thread()
self.log.info("Adding '%s' to crosstown_traffic for %s" % (str(crosstown_task), self.response))
if not self.no_go:
self.response.crosstown_tasks.append(self)
except ThreadHasNoResponse:
self.responseless_fallback(crosstown_task)
return self.run
def __call__(self, crosstown_task=None):
self.crosstown_task = crosstown_task
try:
self.response = get_response_for_thread()
self.log.info("Adding '%s' to crosstown_traffic for %s" % (str(crosstown_task), self.response))
if not self.no_go:
self.response.crosstown_tasks.append(self)
except ThreadHasNoResponse:
self.responseless_fallback(crosstown_task)
return self.run
def application(environ, start_response):
start_response('200 OK', [('Content-type', 'text/plain')])
if 'test_crosstown_traffic' in environ['QUERY_STRING'] or environ['PATH_INFO'] == '/r/1':
log.debug('Starting first cycle...')
@crosstown_traffic()
def delayed_callable():
TestNameSpace.async_task_was_run = True
# This is the end of the test logic.
TestNameSpace.async_task_was_done.put(None)
# Save this response as the "first response" so that we can check later
nameSpace.first_response = get_response_for_thread()
tasks = nameSpace.first_response.crosstown_tasks
# We have one task to run after the response.
nameSpace.test_case.assertEqual(len(tasks), 1)
# The async task flag is still false (ie, the task hasn't run.)
nameSpace.test_case.assertFalse(nameSpace.async_task_was_run)
# Clear second cycle to begin.
nameSpace.ready_to_proceed_with_second_cycle.put(True)
# OK! Hold here until the second cycle completes.
nameSpace.second_cycle_complete.get(True, 3)
# Second cycle has completed, yet it didn't steal our crosstown_traffic
nameSpace.test_case.assertEqual(len(tasks), 1)
# ...and again, the async task still hasn't been run yet.
nameSpace.test_case.assertFalse(nameSpace.async_task_was_run)
return ['The first sync response']
if environ['PATH_INFO'] == '/r/2':
nameSpace.ready_to_proceed_with_second_cycle.get(True, 3)
second_response_tasks = get_tasks_to_follow_current_response()
# We didn't set any tasks during the second response.
nameSpace.test_case.assertFalse(second_response_tasks)
# However, the single task assigned during the first response is still
# milling about.
first_response_tasks = nameSpace.first_response.crosstown_tasks
# And The async task flag is *still* false (ie, the task still has not
# run, but it will once the first request finishes.)
nameSpace.test_case.assertFalse(nameSpace.async_task_was_run)
nameSpace.test_case.assertEqual(len(first_response_tasks), 1)
# ...and it's the one defined above.
nameSpace.test_case.assertEqual(
first_response_tasks[0].crosstown_task.__name__,
'delayed_callable'
)
nameSpace.second_cycle_complete.put(True)
return['The second sync request.']
other_inane_thing = False
return [str(threading.current_thread()),
str(crosstown_traffic.tasks_to_follow_response)]
def application(environ, start_response):
start_response('200 OK', [('Content-type', 'text/plain')])
if 'test_crosstown_traffic' in environ['QUERY_STRING'] or environ[
'PATH_INFO'] == '/r/1':
log.debug('Starting first cycle...')
@crosstown_traffic()
def delayed_callable():
TestNameSpace.async_task_was_run = True
# This is the end of the test logic.
TestNameSpace.async_task_was_done.put(None)
# Save this response as the "first response" so that we can check later
nameSpace.first_response = get_response_for_thread()
tasks = nameSpace.first_response.crosstown_tasks
# We have one task to run after the response.
nameSpace.test_case.assertEqual(len(tasks), 1)
# The async task flag is still false (ie, the task hasn't run.)
nameSpace.test_case.assertFalse(nameSpace.async_task_was_run)
# Clear second cycle to begin.
nameSpace.ready_to_proceed_with_second_cycle.put(True)
# OK! Hold here until the second cycle completes.
nameSpace.second_cycle_complete.get(True, 3)
# Second cycle has completed, yet it didn't steal our crosstown_traffic
nameSpace.test_case.assertEqual(len(tasks), 1)
# ...and again, the async task still hasn't been run yet.
nameSpace.test_case.assertFalse(nameSpace.async_task_was_run)
return b'The first sync response'
if environ['PATH_INFO'] == '/r/2':
nameSpace.ready_to_proceed_with_second_cycle.get(True, 3)
second_response_tasks = get_tasks_to_follow_current_response()
# We didn't set any tasks during the second response.
nameSpace.test_case.assertFalse(second_response_tasks)
# However, the single task assigned during the first response is still
# milling about.
first_response_tasks = nameSpace.first_response.crosstown_tasks
# And The async task flag is *still* false (ie, the task still has not
# run, but it will once the first request finishes.)
nameSpace.test_case.assertFalse(nameSpace.async_task_was_run)
nameSpace.test_case.assertEqual(len(first_response_tasks), 1)
# ...and it's the one defined above.
nameSpace.test_case.assertEqual(
first_response_tasks[0].crosstown_task.__name__,
'delayed_callable')
nameSpace.second_cycle_complete.put(True)
return b'The second sync request.'
other_inane_thing = False
return [
str(threading.current_thread()),
str(crosstown_traffic.tasks_to_follow_response)
]