def test_slave_stats_report_median(self):
import mock
class MyTestLocust(Locust):
pass
with mock.patch("locust.rpc.rpc.Server", mocked_rpc_server()) as server:
master = MasterLocustRunner(MyTestLocust, 10, 10, None)
server.mocked_send(Message("client_ready", None, "fake_client"))
sleep(0)
RequestStats.get("GET", "/").log(100, 23455)
RequestStats.get("GET", "/").log(800, 23455)
RequestStats.get("GET", "/").log(700, 23455)
data = {"user_count":1}
events.report_to_master.fire("fake_client", data)
RequestStats.reset_all()
server.mocked_send(Message("stats", data, "fake_client"))
sleep(0)
s = RequestStats.get("GET", "/")
self.assertEqual(700, s.median_response_time)
def reset_stats():
RequestStats.reset_all()
return "ok"
