locust 2.29.2.dev34__py3-none-any.whl → 2.29.2.dev45__py3-none-any.whl

locust/test/test_fasthttp.py

@@ -1,785 +0,0 @@
-from locust import FastHttpUser
-from locust.argument_parser import parse_options
-from locust.contrib.fasthttp import FastHttpSession
-from locust.exception import CatchResponseError, InterruptTaskSet, LocustError, ResponseError
-from locust.user import TaskSet, task
-from locust.util.load_locustfile import is_user_class
-
-import socket
-import time
-from tempfile import NamedTemporaryFile
-
-import gevent
-from geventhttpclient.client import HTTPClientPool
-from pyquery import PyQuery as pq
-
-from .testcases import LocustTestCase, WebserverTestCase
-from .util import create_tls_cert
-
-
-class TestFastHttpSession(WebserverTestCase):
-    def get_client(self):
-        return FastHttpSession(self.environment, base_url="http://127.0.0.1:%i" % self.port, user=None)
-
-    def test_get(self):
-        s = self.get_client()
-        r = s.get("/ultra_fast")
-        self.assertEqual(200, r.status_code)
-
-    def test_connection_error(self):
-        s = FastHttpSession(self.environment, "http://localhost:1", user=None)
-        r = s.get("/", headers={"X-Test-Headers": "hello"})
-        self.assertEqual(r.status_code, 0)
-        self.assertEqual(None, r.content)
-        self.assertEqual(1, len(self.runner.stats.errors))
-        self.assertTrue(isinstance(r.error, ConnectionRefusedError))
-        self.assertTrue(isinstance(next(iter(self.runner.stats.errors.values())).error, ConnectionRefusedError))
-        self.assertEqual(r.url, "http://localhost:1/")
-        self.assertEqual(r.request.url, r.url)
-        self.assertEqual(r.request.headers.get("X-Test-Headers", ""), "hello")
-
-    def test_404(self):
-        s = self.get_client()
-        r = s.get("/does_not_exist")
-        self.assertEqual(404, r.status_code)
-        self.assertEqual(1, self.runner.stats.get("/does_not_exist", "GET").num_failures)
-
-    def test_204(self):
-        s = self.get_client()
-        r = s.get("/status/204")
-        self.assertEqual(204, r.status_code)
-        self.assertEqual(1, self.runner.stats.get("/status/204", "GET").num_requests)
-        self.assertEqual(0, self.runner.stats.get("/status/204", "GET").num_failures)
-        self.assertEqual(r.url, "http://127.0.0.1:%i/status/204" % self.port)
-        self.assertEqual(r.request.url, r.url)
-
-    def test_streaming_response(self):
-        """
-        Test a request to an endpoint that returns a streaming response
-        """
-        s = self.get_client()
-        r = s.get("/streaming/30")
-
-        # verify that the time reported includes the download time of the whole streamed response
-        self.assertGreater(self.runner.stats.get("/streaming/30", method="GET").avg_response_time, 250)
-        self.runner.stats.clear_all()
-
-        # verify that response time does NOT include whole download time, when using stream=True
-        r = s.get("/streaming/30", stream=True)
-        self.assertGreaterEqual(self.runner.stats.get("/streaming/30", method="GET").avg_response_time, 0)
-        self.assertLess(self.runner.stats.get("/streaming/30", method="GET").avg_response_time, 250)
-
-        # download the content of the streaming response (so we don't get an ugly exception in the log)
-        _ = r.content
-
-    def test_streaming_response_catch_response(self):
-        """
-        Test a request to an endpoint that returns a streaming response, and uses catch_response
-        """
-        s = self.get_client()
-
-        with s.get("/streaming/30", stream=True, catch_response=True) as r:
-            # typical usage of r when stream=True is to read the stream as desired,
-            # with the possibility to "fail fast" when some things are read early on
-            response_content = str(r.stream.read())
-            r.failure("some error")
-
-        self.assertRegex(response_content, "streaming response")
-
-        stats = self.runner.stats.get("/streaming/30", "GET")
-        self.assertEqual(1, stats.num_requests)
-        self.assertEqual(1, stats.num_failures)
-
-        # verify that response time does NOT include whole download time, when using stream=True
-        self.assertGreaterEqual(stats.avg_response_time, 0)
-        self.assertLess(stats.avg_response_time, 250)
-
-    def test_slow_redirect(self):
-        s = self.get_client()
-        url = "/redirect?url=/redirect&delay=0.5"
-        s.get(url)
-        stats = self.runner.stats.get(url, method="GET")
-        self.assertEqual(1, stats.num_requests)
-        self.assertGreater(stats.avg_response_time, 500)
-
-    def test_post_redirect(self):
-        s = self.get_client()
-        url = "/redirect"
-        r = s.post(url)
-        self.assertEqual(200, r.status_code)
-        post_stats = self.runner.stats.get(url, method="POST")
-        get_stats = self.runner.stats.get(url, method="GET")
-        self.assertEqual(1, post_stats.num_requests)
-        self.assertEqual(0, get_stats.num_requests)
-
-    def test_cookie(self):
-        s = self.get_client()
-        r = s.post("/set_cookie?name=testcookie&value=1337")
-        self.assertEqual(200, r.status_code)
-        r = s.get("/get_cookie?name=testcookie")
-        self.assertEqual("1337", r.content.decode())
-        self.assertEqual("1337", r.text)
-
-    def test_head(self):
-        s = self.get_client()
-        r = s.head("/request_method")
-        self.assertEqual(200, r.status_code)
-        self.assertEqual("", r.content.decode())
-
-    def test_delete(self):
-        s = self.get_client()
-        r = s.delete("/request_method")
-        self.assertEqual(200, r.status_code)
-        self.assertEqual("DELETE", r.content.decode())
-
-    def test_patch(self):
-        s = self.get_client()
-        r = s.patch("/request_method")
-        self.assertEqual(200, r.status_code)
-        self.assertEqual("PATCH", r.content.decode())
-
-    def test_options(self):
-        s = self.get_client()
-        r = s.options("/request_method")
-        self.assertEqual(200, r.status_code)
-        self.assertEqual("", r.content.decode())
-        self.assertEqual(
-            {"OPTIONS", "DELETE", "PUT", "GET", "POST", "HEAD", "PATCH"},
-            set(r.headers["allow"].split(", ")),
-        )
-
-    def test_json_payload(self):
-        s = self.get_client()
-        r = s.post("/request_method", json={"foo": "bar"})
-        self.assertEqual(200, r.status_code)
-        self.assertEqual(r.request.body, '{"foo": "bar"}')
-        self.assertEqual(r.request.headers.get("Content-Type"), "application/json")
-
-    def test_catch_response_fail_successful_request(self):
-        s = self.get_client()
-        with s.get("/ultra_fast", catch_response=True) as r:
-            r.failure("nope")
-        self.assertEqual(1, self.environment.stats.get("/ultra_fast", "GET").num_requests)
-        self.assertEqual(1, self.environment.stats.get("/ultra_fast", "GET").num_failures)
-
-    def test_catch_response_pass_failed_request(self):
-        s = self.get_client()
-        with s.get("/fail", catch_response=True) as r:
-            r.success()
-        self.assertEqual(1, self.environment.stats.total.num_requests)
-        self.assertEqual(0, self.environment.stats.total.num_failures)
-
-    def test_catch_response_multiple_failure_and_success(self):
-        s = self.get_client()
-        with s.get("/ultra_fast", catch_response=True) as r:
-            r.failure("nope")
-            r.success()
-            r.failure("nooo")
-            r.success()
-        self.assertEqual(1, self.environment.stats.total.num_requests)
-        self.assertEqual(0, self.environment.stats.total.num_failures)
-
-    def test_catch_response_pass_failed_request_with_other_exception_within_block(self):
-        class OtherException(Exception):
-            pass
-
-        s = self.get_client()
-        try:
-            with s.get("/fail", catch_response=True) as r:
-                r.success()
-                raise OtherException("wtf")
-        except OtherException:
-            pass
-        else:
-            self.fail("OtherException should have been raised")
-
-        self.assertEqual(1, self.environment.stats.total.num_requests)
-        self.assertEqual(0, self.environment.stats.total.num_failures)
-
-    def test_catch_response_default_success(self):
-        s = self.get_client()
-        with s.get("/ultra_fast", catch_response=True):
-            pass
-        self.assertEqual(1, self.environment.stats.get("/ultra_fast", "GET").num_requests)
-        self.assertEqual(0, self.environment.stats.get("/ultra_fast", "GET").num_failures)
-
-    def test_catch_response_default_fail(self):
-        s = self.get_client()
-        with s.get("/fail", catch_response=True):
-            pass
-        self.assertEqual(1, self.environment.stats.total.num_requests)
-        self.assertEqual(1, self.environment.stats.total.num_failures)
-
-    def test_error_message_with_name_replacement(self):
-        s = self.get_client()
-        kwargs = {}
-
-        def on_request(**kw):
-            self.assertIsNotNone(kw["exception"])
-            kwargs.update(kw)
-
-        self.environment.events.request.add_listener(on_request)
-        before_request = time.time()
-        s.request("get", "/wrong_url/01", name="replaced_url_name", context={"foo": "bar"})
-        after_request = time.time()
-        # self.assertIn("for url: replaced_url_name", str(kwargs["exception"])) # this is actually broken for FastHttpUser right now...
-        self.assertAlmostEqual(before_request, kwargs["start_time"], delta=0.01)
-        self.assertAlmostEqual(after_request, kwargs["start_time"] + kwargs["response_time"] / 1000, delta=0.01)
-        self.assertEqual(s.base_url + "/wrong_url/01", kwargs["url"])  # url is unaffected by name
-        self.assertDictEqual({"foo": "bar"}, kwargs["context"])
-
-    def test_custom_ssl_context_fail_with_bad_context(self):
-        """
-        Test FastHttpSession with a custom SSLContext factory that will fail as
-        we can not set verify_mode to CERT_NONE when check_hostname is enabled
-        """
-
-        def create_custom_context():
-            context = gevent.ssl.create_default_context()
-            context.check_hostname = True
-            context.verify_mode = gevent.ssl.CERT_NONE
-            return context
-
-        s = FastHttpSession(
-            self.environment,
-            "https://127.0.0.1:%i" % self.port,
-            ssl_context_factory=create_custom_context,
-            user=None,
-        )
-        with self.assertRaises(ValueError) as e:
-            s.get("/")
-        self.assertEqual(e.exception.args, ("Cannot set verify_mode to CERT_NONE when check_hostname is enabled.",))
-
-    def test_custom_ssl_context_passed_correct_to_client_pool(self):
-        """
-        Test FastHttpSession with a custom SSLContext factory with a options.name
-        that will be passed correctly to the ClientPool. It will also test a 2nd
-        factory which is not the correct one.
-        """
-
-        def custom_ssl_context():
-            context = gevent.ssl.create_default_context()
-            context.check_hostname = False
-            context.verify_mode = gevent.ssl.CERT_NONE
-            context.options.name = "FAKEOPTION"
-            return context
-
-        def custom_context_with_wrong_option():
-            context = gevent.ssl.create_default_context()
-            context.check_hostname = False
-            context.verify_mode = gevent.ssl.CERT_NONE
-            context.options.name = "OPTIONFAKED"
-            return context
-
-        s = FastHttpSession(
-            self.environment,
-            "https://127.0.0.1:%i" % self.port,
-            ssl_context_factory=custom_ssl_context,
-            user=None,
-        )
-        self.assertEqual(s.client.clientpool.client_args["ssl_context_factory"], custom_ssl_context)
-        self.assertNotEqual(s.client.clientpool.client_args["ssl_context_factory"], custom_context_with_wrong_option)
-
-
-class TestRequestStatsWithWebserver(WebserverTestCase):
-    def test_request_stats_content_length(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        locust.client.get("/ultra_fast")
-        self.assertEqual(
-            self.runner.stats.get("/ultra_fast", "GET").avg_content_length, len("This is an ultra fast response")
-        )
-        locust.client.get("/ultra_fast")
-        self.assertEqual(
-            self.runner.stats.get("/ultra_fast", "GET").avg_content_length, len("This is an ultra fast response")
-        )
-
-    def test_request_stats_no_content_length(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        l = MyUser(self.environment)
-        path = "/no_content_length"
-        l.client.get(path)
-        self.assertEqual(
-            self.runner.stats.get(path, "GET").avg_content_length,
-            len("This response does not have content-length in the header"),
-        )
-
-    def test_request_stats_no_content_length_streaming(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        l = MyUser(self.environment)
-        path = "/no_content_length"
-        l.client.get(path, stream=True)
-        self.assertEqual(0, self.runner.stats.get(path, "GET").avg_content_length)
-
-    def test_request_stats_named_endpoint(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        locust.client.get("/ultra_fast", name="my_custom_name")
-        self.assertEqual(1, self.runner.stats.get("my_custom_name", "GET").num_requests)
-
-    def test_request_stats_query_variables(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        locust.client.get("/ultra_fast?query=1")
-        self.assertEqual(1, self.runner.stats.get("/ultra_fast?query=1", "GET").num_requests)
-
-    def test_request_stats_put(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        locust.client.put("/put")
-        self.assertEqual(1, self.runner.stats.get("/put", "PUT").num_requests)
-
-    def test_request_connection_error(self):
-        class MyUser(FastHttpUser):
-            host = "http://localhost:1"
-
-        locust = MyUser(self.environment)
-        response = locust.client.get("/")
-        self.assertEqual(response.status_code, 0)
-        self.assertEqual(1, self.runner.stats.get("/", "GET").num_failures)
-        self.assertEqual(1, self.runner.stats.get("/", "GET").num_requests)
-
-
-class TestFastHttpUserClass(WebserverTestCase):
-    def test_is_abstract(self):
-        self.assertTrue(FastHttpUser.abstract)
-        self.assertFalse(is_user_class(FastHttpUser))
-
-    def test_class_context(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-            def context(self):
-                return {"user": self.username}
-
-        kwargs = {}
-
-        def on_request(**kw):
-            kwargs.update(kw)
-
-        self.environment.events.request.add_listener(on_request)
-        user = MyUser(self.environment)
-        user.username = "foo"
-        user.client.request("get", "/request_method")
-        self.assertDictEqual({"user": "foo"}, kwargs["context"])
-        self.assertEqual("GET", kwargs["response"].text)
-        user.client.request("get", "/request_method", context={"user": "bar"})
-        self.assertDictEqual({"user": "bar"}, kwargs["context"])
-
-    def test_get_request(self):
-        self.response = ""
-
-        def t1(l):
-            self.response = l.client.get("/ultra_fast")
-
-        class MyUser(FastHttpUser):
-            tasks = [t1]
-            host = "http://127.0.0.1:%i" % self.port
-
-        my_locust = MyUser(self.environment)
-        t1(my_locust)
-        self.assertEqual(self.response.text, "This is an ultra fast response")
-
-    def test_client_request_headers(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        r = locust.client.get("/request_header_test", headers={"X-Header-Test": "hello"})
-        self.assertEqual("hello", r.text)
-        self.assertEqual("hello", r.headers.get("X-Header-Test"))
-        self.assertEqual("hello", r.request.headers.get("X-Header-Test"))
-
-    def test_client_get(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        self.assertEqual("GET", locust.client.get("/request_method").text)
-
-    def test_client_get_absolute_url(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        self.assertEqual("GET", locust.client.get("http://127.0.0.1:%i/request_method" % self.port).text)
-
-    def test_client_post(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        self.assertEqual("POST", locust.client.post("/request_method", {"arg": "hello world"}).text)
-        self.assertEqual("hello world", locust.client.post("/post", {"arg": "hello world"}).text)
-
-    def test_client_put(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        self.assertEqual("PUT", locust.client.put("/request_method", {"arg": "hello world"}).text)
-        self.assertEqual("hello world", locust.client.put("/put", {"arg": "hello world"}).text)
-
-    def test_client_delete(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        self.assertEqual("DELETE", locust.client.delete("/request_method").text)
-        self.assertEqual(200, locust.client.delete("/request_method").status_code)
-
-    def test_client_head(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        self.assertEqual(200, locust.client.head("/request_method").status_code)
-
-    def test_complex_content_type(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-
-        self.assertEqual("stuff", locust.client.get("/content_type_missing_charset").text)
-        self.assertEqual("stuff", locust.client.get("/content_type_regular").text)
-        self.assertEqual("stuff", locust.client.get("/content_type_with_extra_stuff").text)
-
-    def test_log_request_name_argument(self):
-        self.response = ""
-
-        class MyUser(FastHttpUser):
-            tasks = []
-            host = "http://127.0.0.1:%i" % self.port
-
-            @task()
-            def t1(l):
-                self.response = l.client.get("/ultra_fast", name="new name!")
-
-        my_locust = MyUser(self.environment)
-        my_locust.t1()
-
-        self.assertEqual(1, self.runner.stats.get("new name!", "GET").num_requests)
-        self.assertEqual(0, self.runner.stats.get("/ultra_fast", "GET").num_requests)
-
-    def test_redirect_url_original_path_as_name(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        l = MyUser(self.environment)
-        l.client.get("/redirect")
-
-        self.assertEqual(1, len(self.runner.stats.entries))
-        self.assertEqual(1, self.runner.stats.get("/redirect", "GET").num_requests)
-        self.assertEqual(0, self.runner.stats.get("/ultra_fast", "GET").num_requests)
-
-    def test_network_timeout_setting(self):
-        class MyUser(FastHttpUser):
-            network_timeout = 0.5
-            host = "http://127.0.0.1:%i" % self.port
-
-        l = MyUser(self.environment)
-
-        timeout = gevent.Timeout(
-            seconds=0.6,
-            exception=AssertionError(
-                "Request took longer than 0.6 even though FastHttpUser.network_timeout was set to 0.5"
-            ),
-        )
-        timeout.start()
-        r = l.client.get("/redirect?url=/redirect&delay=5.0")
-        timeout.cancel()
-
-        self.assertTrue(isinstance(r.error.original, socket.timeout))
-        self.assertEqual(1, self.runner.stats.get("/redirect?url=/redirect&delay=5.0", "GET").num_failures)
-
-    def test_max_redirect_setting(self):
-        class MyUser(FastHttpUser):
-            max_redirects = 0
-            host = "http://127.0.0.1:%i" % self.port
-
-        l = MyUser(self.environment)
-        l.client.get("/redirect")
-        self.assertEqual(1, self.runner.stats.get("/redirect", "GET").num_failures)
-
-    def test_allow_redirects_override(self):
-        class MyLocust(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        l = MyLocust(self.environment)
-        resp = l.client.get("/redirect", allow_redirects=False)
-        self.assertTrue(resp.headers["location"].endswith("/ultra_fast"))
-        resp = l.client.get("/redirect")  # ensure redirect still works
-        self.assertFalse("location" in resp.headers)
-
-    def test_slow_redirect(self):
-        s = FastHttpSession(self.environment, "http://127.0.0.1:%i" % self.port, user=None)
-        url = "/redirect?url=/redirect&delay=0.5"
-        s.get(url)
-        stats = self.runner.stats.get(url, method="GET")
-        self.assertEqual(1, stats.num_requests)
-        self.assertGreater(stats.avg_response_time, 500)
-
-    def test_client_basic_auth(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        class MyAuthorizedUser(FastHttpUser):
-            host = "http://locust:menace@127.0.0.1:%i" % self.port
-
-        class MyUnauthorizedUser(FastHttpUser):
-            host = "http://locust:wrong@127.0.0.1:%i" % self.port
-
-        locust = MyUser(self.environment)
-        unauthorized = MyUnauthorizedUser(self.environment)
-        authorized = MyAuthorizedUser(self.environment)
-        response = authorized.client.get("/basic_auth")
-        self.assertEqual(200, response.status_code)
-        self.assertEqual("Authorized", response.text)
-        self.assertEqual(401, locust.client.get("/basic_auth").status_code)
-        self.assertEqual(401, unauthorized.client.get("/basic_auth").status_code)
-
-    def test_shared_client_pool(self):
-        shared_client_pool = HTTPClientPool(concurrency=1)
-
-        class MyUserA(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-            client_pool = shared_client_pool
-
-        class MyUserB(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-            client_pool = shared_client_pool
-
-        user_a = MyUserA(self.environment)
-        user_b = MyUserB(self.environment)
-
-        user_a.client.get("/ultra_fast")
-        user_b.client.get("/ultra_fast")
-        user_b.client.get("/ultra_fast")
-        user_a.client.get("/ultra_fast")
-
-        self.assertEqual(1, self.connections_count)
-        self.assertEqual(4, self.requests_count)
-
-    def test_client_pool_per_user_instance(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        user_a = MyUser(self.environment)
-        user_b = MyUser(self.environment)
-
-        user_a.client.get("/ultra_fast")
-        user_b.client.get("/ultra_fast")
-        user_b.client.get("/ultra_fast")
-        user_a.client.get("/ultra_fast")
-
-        self.assertEqual(2, self.connections_count)
-        self.assertEqual(4, self.requests_count)
-
-    def test_client_pool_concurrency(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-            @task
-            def t(self):
-                def concurrent_request(url):
-                    response = self.client.get(url)
-                    assert response.status_code == 200
-
-                pool = gevent.pool.Pool()
-                urls = ["/slow?delay=0.2"] * 20  # these urls are all the same, but they could be different
-                for url in urls:
-                    pool.spawn(concurrent_request, url)
-                pool.join()
-
-        user = MyUser(self.environment)
-        before_requests = time.time()
-        user.t()
-        after_requests = time.time()
-        expected_delta = 0.4  # 20 requests with concurrency 10 and response time 0.2
-        self.assertAlmostEqual(before_requests + expected_delta, after_requests, delta=0.1)
-
-
-class TestFastHttpCatchResponse(WebserverTestCase):
-    def setUp(self):
-        super().setUp()
-
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-
-        self.user = MyUser(self.environment)
-
-        self.num_failures = 0
-        self.num_success = 0
-
-        def on_request(exception, **kwargs):
-            if exception:
-                self.num_failures += 1
-                self.last_failure_exception = exception
-            else:
-                self.num_success += 1
-
-        self.environment.events.request.add_listener(on_request)
-
-    def test_catch_response(self):
-        self.assertEqual(500, self.user.client.get("/fail").status_code)
-        self.assertEqual(1, self.num_failures)
-        self.assertEqual(0, self.num_success)
-
-        with self.user.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.user.client.get("/ultra_fast", catch_response=True) as response:
-            raise ResponseError("Not working")
-
-        self.assertEqual(2, self.num_failures)
-        self.assertEqual(1, self.num_success)
-
-    def test_catch_response_http_fail(self):
-        with self.user.client.get("/fail", catch_response=True):
-            pass
-        self.assertEqual(1, self.num_failures)
-        self.assertEqual(0, self.num_success)
-
-    def test_catch_response_http_manual_fail(self):
-        with self.user.client.get("/ultra_fast", catch_response=True) as response:
-            response.failure("Haha!")
-        self.assertEqual(1, self.num_failures)
-        self.assertEqual(0, self.num_success)
-        self.assertTrue(
-            isinstance(self.last_failure_exception, CatchResponseError),
-            "Failure event handler should have been passed a CatchResponseError instance",
-        )
-
-    def test_catch_response_http_manual_success(self):
-        with self.user.client.get("/fail", catch_response=True) as response:
-            response.success()
-        self.assertEqual(0, self.num_failures)
-        self.assertEqual(1, self.num_success)
-
-    def test_catch_response_allow_404(self):
-        with self.user.client.get("/does/not/exist", catch_response=True) as response:
-            self.assertEqual(404, response.status_code)
-            if response.status_code == 404:
-                response.success()
-        self.assertEqual(0, self.num_failures)
-        self.assertEqual(1, self.num_success)
-
-    def test_interrupt_taskset_with_catch_response(self):
-        class MyTaskSet(TaskSet):
-            @task
-            def interrupted_task(self):
-                with self.client.get("/ultra_fast", catch_response=True):
-                    raise InterruptTaskSet()
-
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:%i" % self.port
-            tasks = [MyTaskSet]
-
-        l = MyUser(self.environment)
-        ts = MyTaskSet(l)
-        self.assertRaises(InterruptTaskSet, lambda: ts.interrupted_task())
-        self.assertEqual(0, self.num_failures)
-        self.assertEqual(0, self.num_success)
-
-    def test_catch_response_connection_error_success(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:1"
-
-        l = MyUser(self.environment)
-        with l.client.get("/", catch_response=True) as r:
-            self.assertEqual(r.status_code, 0)
-            self.assertEqual(None, r.content)
-            r.success()
-        self.assertEqual(1, self.num_success)
-        self.assertEqual(0, self.num_failures)
-
-    def test_catch_response_connection_error_fail(self):
-        class MyUser(FastHttpUser):
-            host = "http://127.0.0.1:1"
-
-        l = MyUser(self.environment)
-        with l.client.get("/", catch_response=True) as r:
-            self.assertEqual(r.status_code, 0)
-            self.assertEqual(None, r.content)
-            r.failure("Manual fail")
-        self.assertEqual(0, self.num_success)
-        self.assertEqual(1, self.num_failures)
-
-    def test_catch_response_missing_with_block(self):
-        # incorrect usage, missing with-block
-        r = self.user.client.get("/fail", catch_response=True)
-        self.assertRaises(LocustError, r.success)
-        self.assertRaises(LocustError, r.failure, "")
-
-    def test_missing_catch_response_true(self):
-        # incorrect usage, missing catch_response=True
-        with self.user.client.get("/fail") as resp:
-            self.assertRaises(LocustError, resp.success)
-
-    def test_rest_success(self):
-        self.last_failure_exception = None
-        with self.user.rest("POST", "/rest", json={"foo": "bar"}) as response:
-            assert response.js["foo"] == "bar"
-
-        self.assertEqual(0, self.num_failures)
-        self.assertEqual(1, self.num_success)
-
-    def test_rest_fail(self):
-        with self.user.rest("POST", "/rest", json={"foo": "bar"}) as response:
-            assert response.js["foo"] == "NOPE"
-
-        self.assertTrue(
-            isinstance(self.last_failure_exception, CatchResponseError),
-            "Failure event handler should have been passed a CatchResponseError instance",
-        )
-        self.assertEqual(1, self.num_failures)
-        self.assertEqual(0, self.num_success)
-
-
-class TestFastHttpSsl(LocustTestCase):
-    def setUp(self):
-        super().setUp()
-        tls_cert, tls_key = create_tls_cert("127.0.0.1")
-        self.tls_cert_file = NamedTemporaryFile()
-        self.tls_key_file = NamedTemporaryFile()
-        with open(self.tls_cert_file.name, "w") as f:
-            f.write(tls_cert.decode())
-        with open(self.tls_key_file.name, "w") as f:
-            f.write(tls_key.decode())
-
-        self.web_ui = self.environment.create_web_ui(
-            "127.0.0.1",
-            0,
-            tls_cert=self.tls_cert_file.name,
-            tls_key=self.tls_key_file.name,
-        )
-        gevent.sleep(0.01)
-        self.web_port = self.web_ui.server.server_port
-
-    def tearDown(self):
-        super().tearDown()
-        self.web_ui.stop()
-
-    def test_ssl_request_insecure(self):
-        s = FastHttpSession(self.environment, "https://127.0.0.1:%i" % self.web_port, insecure=True, user=None)
-        response = s.get("/")
-        d = pq(response.content.decode("utf-8"))
-
-        self.assertEqual(200, response.status_code)
-        self.assertIn('"users": null', str(d))