Allow asserts on add_callback() matches (#774)

Allow asserts on add_callback() matches

Fixes #727

Co-authored-by: Martijn van Oosterhout <[email protected]>
Co-authored-by: Mark Story <[email protected]>

Author: 3 people

CHANGES

@@ -2,6 +2,7 @@
 ------
 
 * Fix bug where the content type is always recorded as either text/plain or application/json. See #770
+* Allow asserts on add_callback() matches. See #727
 
 0.25.7
 ------

responses/__init__.py

@@ -967,8 +967,8 @@ def add_callback(
         match_querystring: Union[bool, FalseBool] = FalseBool(),
         content_type: Optional[str] = "text/plain",
         match: "_MatcherIterable" = (),
-    ) -> None:
-        self._registry.add(
+    ) -> BaseResponse:
+        return self._registry.add(
             CallbackResponse(
                 url=url,
                 method=method,

responses/tests/test_responses.py

@@ -574,14 +574,15 @@ def request_callback(_request):
 
     @responses.activate
     def run():
-        responses.add_callback(responses.GET, url, request_callback)
+        rsp = responses.add_callback(responses.GET, url, request_callback)
         resp = requests.get(url)
         assert resp.text == "test callback"
         assert resp.status_code == status
         assert resp.reason == reason
         assert "bar" == resp.headers.get("foo")
         assert "application/json" == resp.headers.get("Content-Type")
         assert "13" == resp.headers.get("Content-Length")
+        assert len(rsp.calls) == 1
 
     run()
     assert_reset()