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 from __future__ import absolute_import
17 import experimental_framework.heat_template_generation as heat_gen
20 import experimental_framework.common as common
21 from experimental_framework import APEX_LAKE_ROOT
23 __author__ = 'gpetralx'
28 common.CONF_FILE = None
29 common.DEPLOYMENT_UNIT = None
30 common.ITERATIONS = None
31 common.BASE_DIR = None
32 common.RESULT_DIR = None
33 common.TEMPLATE_DIR = None
34 common.TEMPLATE_NAME = None
35 common.TEMPLATE_FILE_EXTENSION = None
37 common.PKTGEN_DIR = None
38 common.PKTGEN_DPDK_DIRECTORY = None
39 common.PKTGEN_PROGRAM = None
40 common.PKTGEN_COREMASK = None
41 common.PKTGEN_MEMCHANNEL = None
42 common.PKTGEN_BUS_SLOT_NIC_1 = None
43 common.PKTGEN_BUS_SLOT_NIC_2 = None
44 common.INFLUXDB_IP = None
45 common.INFLUXDB_PORT = None
46 common.INFLUXDB_DB_NAME = None
49 class TestGeneratesTemplate(unittest.TestCase):
52 self.deployment_configuration = {
53 'vnic_type': ['normal', 'direct'],
57 self.template_name = 'VTC_base_single_vm_wait.tmp'
63 @mock.patch('experimental_framework.common.LOG')
64 @mock.patch('experimental_framework.common.get_template_dir')
65 def test_generates_template_for_success(self, mock_template_dir,
67 generated_templates_dir = os.path.join(
68 APEX_LAKE_ROOT, 'tests/data/generated_templates/')
69 mock_template_dir.return_value = generated_templates_dir
70 test_templates = os.path.join(APEX_LAKE_ROOT,
71 'tests/data/test_templates/')
72 heat_gen.generates_templates(self.template_name,
73 self.deployment_configuration)
74 for dirname, dirnames, filenames in os.walk(test_templates):
75 for filename in filenames:
76 with open(test_templates + filename) as test:
77 with open(generated_templates_dir + filename) as generated:
78 self.assertListEqual(test.readlines(),
79 generated.readlines())
81 self.template_name = os.path.join(
83 'tests/data/generated_templates/VTC_base_single_vm_wait.tmp')
84 heat_gen.generates_templates(self.template_name,
85 self.deployment_configuration)
86 for dirname, dirnames, filenames in os.walk(test_templates):
87 for filename in filenames:
88 with open(test_templates + filename) as test:
89 with open(generated_templates_dir + filename) as generated:
90 self.assertListEqual(test.readlines(),
91 generated.readlines())
93 @mock.patch('experimental_framework.common.get_template_dir')
94 def test_get_all_heat_templates_for_success(self, template_dir):
95 generated_templates = os.path.join(APEX_LAKE_ROOT,
96 'tests/data/generated_templates/')
97 template_dir.return_value = generated_templates
99 expected = ['experiment_1.yaml', 'experiment_2.yaml']
100 result = heat_gen.get_all_heat_templates(generated_templates,
102 self.assertListEqual(expected, result)