1 # Copyright (c) 2015 Intel Research and Development Ireland Ltd.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 __author__ = 'gpetralx'
19 import experimental_framework.heat_template_generation as heat_gen
22 import experimental_framework.common as common
27 common.CONF_FILE = None
28 common.DEPLOYMENT_UNIT = None
29 common.ITERATIONS = None
30 common.BASE_DIR = None
31 common.RESULT_DIR = None
32 common.TEMPLATE_DIR = None
33 common.TEMPLATE_NAME = None
34 common.TEMPLATE_FILE_EXTENSION = None
36 common.PKTGEN_DIR = None
37 common.PKTGEN_DPDK_DIRECTORY = None
38 common.PKTGEN_PROGRAM = None
39 common.PKTGEN_COREMASK = None
40 common.PKTGEN_MEMCHANNEL = None
41 common.PKTGEN_BUS_SLOT_NIC_1 = None
42 common.PKTGEN_BUS_SLOT_NIC_2 = None
43 common.INFLUXDB_IP = None
44 common.INFLUXDB_PORT = None
45 common.INFLUXDB_DB_NAME = None
48 class TestGeneratesTemplate(unittest.TestCase):
50 self.deployment_configuration = {
51 'vnic_type': ['normal', 'direct'],
55 self.template_name = 'VTC_base_single_vm_wait.tmp'
61 @mock.patch('experimental_framework.common.LOG')
62 @mock.patch('experimental_framework.common.get_template_dir')
63 def test_generates_template_for_success(self, mock_template_dir,
65 generated_templates_dir = 'tests/data/generated_templates/'
66 mock_template_dir.return_value = generated_templates_dir
67 test_templates = 'tests/data/test_templates/'
68 heat_gen.generates_templates(self.template_name,
69 self.deployment_configuration)
70 for dirname, dirnames, filenames in os.walk(test_templates):
71 for filename in filenames:
72 with open(test_templates + filename) as test:
73 with open(generated_templates_dir + filename) as generated:
74 self.assertListEqual(test.readlines(),
75 generated.readlines())
77 t_name = '/tests/data/generated_templates/VTC_base_single_vm_wait.tmp'
78 self.template_name = "{}{}".format(os.getcwd(), t_name)
79 heat_gen.generates_templates(self.template_name,
80 self.deployment_configuration)
81 for dirname, dirnames, filenames in os.walk(test_templates):
82 for filename in filenames:
83 with open(test_templates + filename) as test:
84 with open(generated_templates_dir + filename) as generated:
85 self.assertListEqual(test.readlines(),
86 generated.readlines())
88 @mock.patch('experimental_framework.common.get_template_dir')
89 def test_get_all_heat_templates_for_success(self, template_dir):
90 generated_templates = 'tests/data/generated_templates/'
91 template_dir.return_value = generated_templates
93 expected = ['experiment_1.yaml', 'experiment_2.yaml']
94 result = heat_gen.get_all_heat_templates(generated_templates,
96 self.assertListEqual(expected, result)