def setUp(self):
super(TestSocketIO, self).setUp()
self.client = None
self.server = SimpleSocketIOServer()
self.server.run()
gevent.sleep(0.5)
global_stats.clear_all()
def setUp(self):
super(WebserverTestCase, self).setUp()
self._web_server = gevent.pywsgi.WSGIServer(("127.0.0.1", 0), app, log=None)
gevent.spawn(lambda: self._web_server.serve_forever())
gevent.sleep(0.01)
self.port = self._web_server.server_port
global_stats.clear_all()
def setUp(self):
super(WebserverTestCase, self).setUp()
self._web_server = gevent.pywsgi.WSGIServer(("127.0.0.1", 0), app, log=None)
gevent.spawn(lambda: self._web_server.serve_forever())
gevent.sleep(0.01)
self.port = self._web_server.server_port
global_stats.clear_all()
def test_404(self):
global_stats.clear_all()
s = FastHttpSession("http://127.0.0.1:%i" % self.port)
r = s.get("/does_not_exist")
self.assertEqual(404, r.status_code)
self.assertEqual(
1,
global_stats.get("/does_not_exist", "GET").num_failures)
def test_204(self):
global_stats.clear_all()
s = FastHttpSession("http://127.0.0.1:%i" % self.port)
r = s.get("/status/204")
self.assertEqual(204, r.status_code)
self.assertEqual(1,
global_stats.get("/status/204", "GET").num_requests)
self.assertEqual(0,
global_stats.get("/status/204", "GET").num_failures)
def setUp(self):
super(TestLocustTaskStatistics, self).setUp()
options = config.LocustConfig()
options.update_config({'host': '127.0.0.1'})
class User(WebLocust):
pass
self.locust = User(options)
global_stats.clear_all()
def test_connection_error(self):
global_stats.clear_all()
s = FastHttpSession("http://localhost:1")
r = s.get("/", timeout=0.1)
self.assertEqual(r.status_code, 0)
self.assertEqual(None, r.content)
self.assertEqual(1, len(global_stats.errors))
self.assertTrue(isinstance(r.error, ConnectionRefusedError))
self.assertTrue(
isinstance(
next(iter(global_stats.errors.values())).error,
ConnectionRefusedError))
def test_streaming_response(self):
"""
Test a request to an endpoint that returns a streaming response
"""
s = HttpSession("http://127.0.0.1:%i" % self.port)
r = s.get("/streaming/30")
# verify that the time reported includes the download time of the whole streamed response
self.assertGreater(global_stats.get("/streaming/30", method="GET").avg_response_time, 250)
global_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.assertGreater(global_stats.get("/streaming/30", method="GET").avg_response_time, 0)
self.assertLess(global_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(self):
"""
Test a request to an endpoint that returns a streaming response
"""
s = HttpSession("http://127.0.0.1:%i" % self.port)
r = s.get("/streaming/30")
# verify that the time reported includes the download time of the whole streamed response
self.assertGreater(global_stats.get("/streaming/30", method="GET").avg_response_time, 250)
global_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.assertGreater(global_stats.get("/streaming/30", method="GET").avg_response_time, 0)
self.assertLess(global_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_max_requests_failed_requests(self):
class MyTaskSet(TaskSet):
@task
def my_task(self):
self.client.get("/ultra_fast")
self.client.get("/fail")
self.client.get("/fail")
class MyLocust(HttpLocust):
host = "http://127.0.0.1:%i" % self.port
task_set = MyTaskSet
min_wait = 1
max_wait = 1
try:
from locust.exception import StopLocust
global_stats.clear_all()
global_stats.max_requests = 3
l = MyLocust()
self.assertRaises(StopLocust, lambda: l.task_set(l).run())
self.assertEqual(1, global_stats.num_requests)
self.assertEqual(2, global_stats.num_failures)
global_stats.clear_all()
global_stats.max_requests = 2
self.assertEqual(0, global_stats.num_requests)
self.assertEqual(0, global_stats.num_failures)
l.run()
self.assertEqual(1, global_stats.num_requests)
self.assertEqual(1, global_stats.num_failures)
finally:
global_stats.clear_all()
global_stats.max_requests = None
def test_max_requests_failed_requests(self):
class MyTaskSet(TaskSet):
@task
def my_task(self):
self.client.get("/ultra_fast")
self.client.get("/fail")
self.client.get("/fail")
class MyLocust(HttpLocust):
host = "http://127.0.0.1:%i" % self.port
task_set = MyTaskSet
min_wait = 1
max_wait = 1
try:
from locust.exception import StopLocust
global_stats.clear_all()
global_stats.max_requests = 3
l = MyLocust()
self.assertRaises(StopLocust, lambda: l.task_set(l).run())
self.assertEqual(1, global_stats.num_requests)
self.assertEqual(2, global_stats.num_failures)
global_stats.clear_all()
global_stats.max_requests = 2
self.assertEqual(0, global_stats.num_requests)
self.assertEqual(0, global_stats.num_failures)
l.run()
self.assertEqual(1, global_stats.num_requests)
self.assertEqual(1, global_stats.num_failures)
finally:
global_stats.clear_all()
global_stats.max_requests = None
def test_max_requests(self):
options = config.LocustConfig()
options.update_config({'host': '127.0.0.1', 'port': self.port})
class MyTaskSet(TaskSet):
@task
def my_task(self):
self.client.http.get("/ultra_fast")
class MyLocust(WebLocust):
task_set = MyTaskSet
min_wait = 1
max_wait = 1
try:
from locust.exception import StopLocust
global_stats.clear_all()
global_stats.max_requests = 2
l = MyLocust(options)
self.assertRaises(StopLocust, lambda: l.task_set(l).run())
self.assertEqual(2, global_stats.num_requests)
global_stats.clear_all()
global_stats.max_requests = 2
self.assertEqual(0, global_stats.num_requests)
l.run()
self.assertEqual(2, global_stats.num_requests)
finally:
global_stats.clear_all()
global_stats.max_requests = None
def tearDown(self):
global_stats.clear_all()