def test__run_scenario_aborted(self): runner_obj = constant.ConstantScenarioRunner(self.task, self.config) runner_obj.abort() runner_obj._run_scenario(fakes.FakeScenario, "do_it", self.context, self.args) self.assertEqual(len(runner_obj.result_queue), 0)
def test__run_scenario(self): runner_obj = constant.ConstantScenarioRunner(self.task, self.config) runner_obj._run_scenario( fakes.FakeScenario, "do_it", self.context, self.args) self.assertEqual(len(runner_obj.result_queue), self.config["times"]) for result in runner_obj.result_queue: self.assertIsNotNone(runner.ScenarioRunnerResult(result))
def test__run_scenario_exception(self): runner_obj = constant.ConstantScenarioRunner(self.task, self.config) runner_obj._run_scenario(fakes.FakeScenario, "something_went_wrong", self.context, self.args) self.assertEqual(len(runner_obj.result_queue), self.config["times"]) for result in runner_obj.result_queue: self.assertIsNotNone(runner.ScenarioRunnerResult(result)) self.assertIn("error", runner_obj.result_queue[0])
def test_abort(self): runner_obj = constant.ConstantScenarioRunner(self.task, self.config) self.assertFalse(runner_obj.aborted.is_set()) runner_obj.abort() self.assertTrue(runner_obj.aborted.is_set())
def test_that_cpu_count_is_adjusted_properly(self, mock__join_processes, mock__create_process_pool, mock__log_debug_info, mock_cpu_count, mock_queue): samples = [ { "input": { "times": 20, "concurrency": 20, "type": "constant", "max_cpu_count": 1 }, "real_cpu": 2, "expected": { # max_cpu_used equals to min(max_cpu_count, real_cpu) "max_cpu_used": 1, # processes_to_start equals to # min(max_cpu_used, times, concurrency)) "processes_to_start": 1, "concurrency_per_worker": 20, "concurrency_overhead": 0, } }, { "input": { "times": 20, "concurrency": 15, "type": "constant", "max_cpu_count": 3 }, "real_cpu": 2, "expected": { "max_cpu_used": 2, "processes_to_start": 2, "concurrency_per_worker": 7, "concurrency_overhead": 1, } }, { "input": { "times": 20, "concurrency": 1, "type": "constant", "max_cpu_count": 3 }, "real_cpu": 2, "expected": { "max_cpu_used": 2, "processes_to_start": 1, "concurrency_per_worker": 1, "concurrency_overhead": 0, } }, { "input": { "times": 2, "concurrency": 5, "type": "constant", "max_cpu_count": 4 }, "real_cpu": 4, "expected": { "max_cpu_used": 4, "processes_to_start": 2, "concurrency_per_worker": 2, "concurrency_overhead": 1, } } ] for sample in samples: mock__log_debug_info.reset_mock() mock_cpu_count.reset_mock() mock__create_process_pool.reset_mock() mock__join_processes.reset_mock() mock_queue.reset_mock() mock_cpu_count.return_value = sample["real_cpu"] runner_obj = constant.ConstantScenarioRunner( self.task, sample["input"]) runner_obj._run_scenario(fakes.FakeScenario, "do_it", self.context, self.args) mock_cpu_count.assert_called_once_with() mock__log_debug_info.assert_called_once_with( times=sample["input"]["times"], concurrency=sample["input"]["concurrency"], timeout=0, max_cpu_used=sample["expected"]["max_cpu_used"], processes_to_start=sample["expected"]["processes_to_start"], concurrency_per_worker=( sample["expected"]["concurrency_per_worker"]), concurrency_overhead=( sample["expected"]["concurrency_overhead"])) args, kwargs = mock__create_process_pool.call_args self.assertIn(sample["expected"]["processes_to_start"], args) self.assertIn(constant._worker_process, args) mock__join_processes.assert_called_once_with( mock__create_process_pool.return_value, mock_queue.return_value, mock_queue.return_value)