def test___init__(self, mock_heat, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = 'tests/data/results/' common.TEMPLATE_FILE_EXTENSION = '.ext' common.RESULT_DIR = 'tests/data/results/' heat_template_name = 'name' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = ['bench_1', 'bench_2'] bu = BenchmarkingUnit(heat_template_name, openstack_credentials, heat_template_parameters, iterations, benchmarks) self.assertEqual(bu.required_benchmarks, benchmarks) bu.heat_template_parameters = heat_template_parameters # mock_data_manager.assert_called_once_with('tests/data/results/12345') mock_dep_unit.assert_called_once_with(openstack_credentials) mock_heat.assert_called_once_with('tests/data/results/', '.ext')
def test_run_benchmarks_for_success(self, mock_heat, mock_common_dep_unit, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time, mock_rfc2544, mock_log, mock_influx): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = 'tests/data/test_templates/' common.TEMPLATE_FILE_EXTENSION = '.yaml' common.RESULT_DIR = 'tests/data/results/' common.INFLUXDB_IP = 'InfluxIP' common.INFLUXDB_PORT = '8086' common.INFLUXDB_DB_NAME = 'test_db' heat_template_name = 'VTC_base_single_vm_wait_' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = [ { 'name': 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark', 'params': dict() } ] bu = BenchmarkingUnit(heat_template_name, openstack_credentials, heat_template_parameters, iterations, benchmarks) # bu.data_manager = DummyDataManager('tests/data/results/12345') bu.template_files = ['VTC_base_single_vm_wait_1.yaml', 'VTC_base_single_vm_wait_2.yaml'] bu.benchmarks = [Dummy_2544('dummy', {'param1': 'val1'})] bu.run_benchmarks() self.assertEqual(bu.benchmarks[0].init(True), 2) self.assertEqual(bu.benchmarks[0].finalize(True), 2) self.assertEqual(bu.benchmarks[0].run(True), 2)
def test_get_benchmark_name_for_success( self, mock_heat, mock_common_dep_unit, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time, mock_rfc2544, mock_log): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = 'tests/data/test_templates/' common.TEMPLATE_FILE_EXTENSION = '.yaml' common.RESULT_DIR = 'tests/data/results/' heat_template_name = 'VTC_base_single_vm_wait_' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = [ { 'name': 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark', 'params': dict() } ] bu = BenchmarkingUnit(heat_template_name, openstack_credentials, heat_template_parameters, iterations, benchmarks) expected = 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark_0' output = bu.get_benchmark_name( 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark') self.assertEqual(expected, output) expected = 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark_1' output = bu.get_benchmark_name( 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark') self.assertEqual(expected, output)
def test_get_required_benchmarks_for_success( self, mock_heat, mock_common_dep_unit, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time, mock_rfc2544, mock_log): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = 'tests/data/test_templates/' common.TEMPLATE_FILE_EXTENSION = '.yaml' common.RESULT_DIR = 'tests/data/results/' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = [ { 'name': 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark', 'params': dict() } ] bu = BenchmarkingUnit('', openstack_credentials, heat_template_parameters, iterations, benchmarks) req_benchs = \ ['rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark'] output = bu.get_required_benchmarks(req_benchs) self.assertEqual(len(req_benchs), 1) self.assertEqual(output[0].__class__, Dummy_2544)
def test_run_benchmarks_for_success(self, mock_heat, mock_common_dep_unit, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time, mock_rfc2544, mock_log, mock_influx): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = os.path.join(APEX_LAKE_ROOT, 'tests/data/test_templates/') common.TEMPLATE_FILE_EXTENSION = '.yaml' common.RESULT_DIR = 'tests/data/results/' common.INFLUXDB_IP = 'InfluxIP' common.INFLUXDB_PORT = '8086' common.INFLUXDB_DB_NAME = 'test_db' heat_template_name = 'VTC_base_single_vm_wait_' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = [ { 'name': 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark', 'params': dict() } ] bu = BenchmarkingUnit(heat_template_name, openstack_credentials, heat_template_parameters, iterations, benchmarks) # bu.data_manager = DummyDataManager('tests/data/results/12345') bu.template_files = ['VTC_base_single_vm_wait_1.yaml', 'VTC_base_single_vm_wait_2.yaml'] bu.benchmarks = [Dummy_2544('dummy', {'param1': 'val1'})] bu.run_benchmarks() self.assertEqual(bu.benchmarks[0].init(True), 2) self.assertEqual(bu.benchmarks[0].finalize(True), 2) self.assertEqual(bu.benchmarks[0].run(True), 2)
def test_run_benchmarks_2_for_success( self, mock_heat, mock_common_dep_unit, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time, mock_rfc2544, mock_log): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = 'tests/data/test_templates/' common.TEMPLATE_FILE_EXTENSION = '.yaml' common.RESULT_DIR = 'tests/data/results/' heat_template_name = 'VTC_base_single_vm_wait_' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = [ { 'name': 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark', 'params': dict() } ] bu = BenchmarkingUnit(heat_template_name, openstack_credentials, heat_template_parameters, iterations, benchmarks) # bu.data_manager = DummyDataManager('tests/data/results/12345') bu.template_files = ['VTC_base_single_vm_wait_1.yaml', 'VTC_base_single_vm_wait_2.yaml'] bu.benchmarks = [Dummy_2544('dummy', dict())] common.DEPLOYMENT_UNIT = DummyDeploymentUnit(dict()) bu.run_benchmarks() self.assertEqual(bu.benchmarks[0].init(True), 2) self.assertEqual(bu.benchmarks[0].finalize(True), 0) self.assertEqual(bu.benchmarks[0].run(True), 0)
def test_initialize_for_success(self, mock_heat, mock_dep_unit, # mock_data_manager, mock_temp_dir, mock_time, mock_rfc2544): mock_heat.return_value = list() mock_time.return_value = '12345' mock_temp_dir.return_value = 'tests/data/test_templates/' common.TEMPLATE_FILE_EXTENSION = '.yaml' common.RESULT_DIR = 'tests/data/results/' heat_template_name = 'VTC_base_single_vm_wait_' openstack_credentials = { 'name': 'aaa', 'surname': 'bbb' } heat_template_parameters = { 'param_1': 'name_1', 'param_2': 'name_2' } iterations = 1 benchmarks = [ { 'name': 'rfc2544_throughput_benchmark.RFC2544ThroughputBenchmark', 'params': dict() } ] bu = BenchmarkingUnit(heat_template_name, openstack_credentials, heat_template_parameters, iterations, benchmarks) self.assertEqual(bu.required_benchmarks, benchmarks) bu.heat_template_parameters = heat_template_parameters bu.template_files = ['VTC_base_single_vm_wait_1.yaml', 'VTC_base_single_vm_wait_2.yaml'] bu.initialize() self.assertTrue(len(bu.benchmarks) == 1) self.assertEqual(bu.benchmarks[0].__class__, Dummy_2544)
def __init__(self): BenchmarkingUnit.__init__(self)
def __init__(self): BenchmarkingUnit.__init__(self)