Merge pull request #1105 from skivis/fix/catch-response

Fix empty response.content when using catch_response=True with FastHttpLocust
locustio · Oct 18, 2019 · 2a0d18a · 2a0d18a
2 parents 644496f + dabc1b8
commit 2a0d18a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/locust/contrib/fasthttp.py b/locust/contrib/fasthttp.py
@@ -316,6 +316,7 @@ class ResponseContextManager(FastResponse):
     def __init__(self, response):
         # copy data from response to this object
         self.__dict__ = response.__dict__
+        self._cached_content = response.content
 
     def __enter__(self):
         return self

diff --git a/locust/test/test_fasthttp.py b/locust/test/test_fasthttp.py
@@ -314,6 +314,7 @@ def test_catch_response(self):
         with self.locust.client.get("/ultra_fast", catch_response=True) as response: pass
         self.assertEqual(1, self.num_failures)
         self.assertEqual(1, self.num_success)
+        self.assertIn("ultra fast", str(response.content))
 
         with self.locust.client.get("/ultra_fast", catch_response=True) as response:
             raise ResponseError("Not working")