diff options
| -rwxr-xr-x | yardstick/vTC/apexlake/experimental_framework/bin/run_tests.sh | 1 | ||||
| -rw-r--r-- | yardstick/vTC/apexlake/experimental_framework/tests/generates_template_test.py | 71 |
2 files changed, 72 insertions, 0 deletions
diff --git a/yardstick/vTC/apexlake/experimental_framework/bin/run_tests.sh b/yardstick/vTC/apexlake/experimental_framework/bin/run_tests.sh new file mode 100755 index 000000000..01592ddc7 --- /dev/null +++ b/yardstick/vTC/apexlake/experimental_framework/bin/run_tests.sh @@ -0,0 +1 @@ +nosetests --with-coverage --cover-erase --cover-package experimental_framework diff --git a/yardstick/vTC/apexlake/experimental_framework/tests/generates_template_test.py b/yardstick/vTC/apexlake/experimental_framework/tests/generates_template_test.py new file mode 100644 index 000000000..85435db6a --- /dev/null +++ b/yardstick/vTC/apexlake/experimental_framework/tests/generates_template_test.py @@ -0,0 +1,71 @@ +# Copyright (c) 2015 Intel Research and Development Ireland Ltd. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import unittest +import experimental_framework.heat_template_generation as heat_gen +import mock +import os +import experimental_framework.common as common + + +class TestGeneratesTemplate(unittest.TestCase): + def setUp(self): + self.deployment_configuration = { + 'vnic_type': ['normal', 'direct'], + 'ram': ['1024'], + 'vcpus': ['2'] + } + self.template_name = 'VTC_base_single_vm_wait.tmp' + common.init() + + def test_dummy(self): + self.assertTrue(True) + + def tearDown(self): + pass + + @mock.patch('experimental_framework.common.get_template_dir') + def test_generates_template_for_success(self, mock_template_dir): + generated_templates_dir = 'tests/data/generated_templates/' + mock_template_dir.return_value = generated_templates_dir + test_templates = 'tests/data/test_templates/' + heat_gen.generates_templates(self.template_name, + self.deployment_configuration) + for dirname, dirnames, filenames in os.walk(test_templates): + for filename in filenames: + with open(test_templates + filename) as test: + with open(generated_templates_dir + filename) as generated: + self.assertListEqual(test.readlines(), + generated.readlines()) + + t_name = '/tests/data/generated_templates/VTC_base_single_vm_wait.tmp' + self.template_name = os.getcwd() + t_name + heat_gen.generates_templates(self.template_name, + self.deployment_configuration) + for dirname, dirnames, filenames in os.walk(test_templates): + for filename in filenames: + with open(test_templates + filename) as test: + with open(generated_templates_dir + filename) as generated: + self.assertListEqual(test.readlines(), + generated.readlines()) + + @mock.patch('experimental_framework.common.get_template_dir') + def test_get_all_heat_templates_for_success(self, template_dir): + generated_templates = 'tests/data/generated_templates/' + template_dir.return_value = generated_templates + extension = '.yaml' + expected = ['experiment_1.yaml', 'experiment_2.yaml'] + result = heat_gen.get_all_heat_templates(generated_templates, + extension) + self.assertListEqual(expected, result) |