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