def test_method_all_with_one_rejected_promise(self):
class MyException(Exception):
pass
p1 = Promise.reject(MyException())
p = Promise.all([p1])
with pytest.raises(MyException):
p.result(0.001)
def on(self, path, response):
"""Set response for the next request to "path"
Order are executed in the same order as declared (fifo).
"""
if isinstance(response, Exception):
p = Promise.reject(response)
else:
p = Promise.resolve(response)
self._expected_requests[path].append(p)
def test_method_all_with_rejected_promise(self):
class MyException(Exception):
pass
promises = [Promise.resolve(1) for _ in range(0, 5)]
promises += [Promise.reject(MyException())]
promises += [Promise.resolve(1) for _ in range(0, 5)]
p = Promise.all(promises)
with pytest.raises(MyException):
p.result()
def execute_request(request):
if context['pass']:
return Promise.resolve(None)
else:
return Promise.reject(NetworkError())
def _token_expired_response(self):
"""HTTP response 401 expired token error."""
return Promise.reject(MockTokenExpiredError())
def execute_request(request):
if request.host == host:
return Promise.resolve(None)
else:
return Promise.reject(NetworkError())
def bad(_request):
"""Request always in error."""
return Promise.reject(NetworkError())
def execute_request(request):
context['count'] += 1
if context['count'] < 3:
return Promise.reject(NetworkError())
else:
return Promise.resolve(None)
def task():
yield Promise.resolve('OK')
yield Promise.reject(Err())
def task():
yield Promise.resolve('OK')
with pytest.raises(Err):
yield Promise.reject(Err())
yield 'OK'