From d7f52c1ed4ba42a0af645ceb99b861f35916dd79 Mon Sep 17 00:00:00 2001 From: "Ryan T. Dean" Date: Wed, 9 Jul 2014 14:27:34 -0700 Subject: [PATCH] tests: imap test for exception handler returns value --- tests.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests.py b/tests.py index a5747eb..b3789e3 100644 --- a/tests.py +++ b/tests.py @@ -202,6 +202,18 @@ def exception_handler(request, exception): out.append(r) self.assertEquals(out, []) + def test_imap_timeout_exception_handler_returns_value(self): + """ + ensure behaviour for a handler that returns a value + """ + def exception_handler(request, exception): + return request + reqs = [grequests.get(httpbin('delay/1'), timeout=0.001)] + out = [] + for r in grequests.imap(reqs, exception_handler=exception_handler): + out.append(r) + self.assertEquals(out, []) + def test_map_timeout_exception(self): class ExceptionHandler: def __init__(self):