def test_single_count(self):
"""
Adds ``request_rate`` samples per call.
"""
calls_per_second = 10
clock = Clock()
request = TestRequest()
scenario = RequestLoadScenario(clock,
request,
request_rate=calls_per_second)
scenario._request_and_measure(1)
self.assertEqual(scenario.rate_measurer.get_metrics()['ok_count'],
calls_per_second)
def test_multiple_count(self):
"""
The count controls how many requests are made.
"""
calls_per_second = 10
seconds = 2
clock = Clock()
request = TestRequest()
scenario = RequestLoadScenario(clock,
request,
request_rate=calls_per_second)
scenario._request_and_measure(seconds)
self.assertEqual(scenario.rate_measurer.get_metrics()['ok_count'],
calls_per_second * seconds)