Let the callback result set the URL of a response

README.rst

@@ -190,7 +190,7 @@ for convenience use *payload* argument to mock out json response. Example below.
  # will throw an exception.
 
 
-**aioresponses allows to use callbacks to provide dynamic responses**
+**aioresponses allows you to use callbacks to provide dynamic responses**
 
 .. code:: python
 

aioresponses/core.py

@@ -36,7 +36,8 @@ def __init__(self, method: str = hdrs.METH_GET,
  payload: Dict = None,
  headers: Dict = None,
  response_class: 'ClientResponse' = None,
- reason: Optional[str] = None):
+ reason: Optional[str] = None,
+ url: Optional[URL] = None):
  self.method = method
  self.status = status
  self.body = body
@@ -45,6 +46,7 @@ def __init__(self, method: str = hdrs.METH_GET,
  self.headers = headers
  self.response_class = response_class
  self.reason = reason
+ self.url = url
 
 
 class RequestMatch(object):
@@ -166,6 +168,8 @@ async def build_response(
  return self.exception
  if callable(self.callback):
  result = self.callback(url, **kwargs)
+ if result.url is not None:
+ url = result.url
  else:
  result = None
  result = self if result is None else result

tests/test_aioresponses.py

@@ -328,3 +328,18 @@ def callback(url, **kwargs):
  response = self.run_async(self.request(self.url))
  data = self.run_async(response.read())
  assert data == body
+
+ @aioresponses()
+ def test_callback_setting_url(self, m):
+ body = b'New body'
+
+ def callback(url, **kwargs):
+ self.assertEqual(str(url), self.url)
+ self.assertEqual(kwargs, {'allow_redirects': True})
+ return CallbackResult(body=body, url=url.with_query(foo='bar'))
+
+ m.get(self.url, callback=callback)
+ response = self.run_async(self.request(self.url))
+ data = self.run_async(response.read())
+ assert response.url == URL(self.url).with_query(foo='bar')
+ assert data == body