diff options
Diffstat (limited to 'vmdb/spec_tests.py')
-rw-r--r-- | vmdb/spec_tests.py | 109 |
1 files changed, 0 insertions, 109 deletions
diff --git a/vmdb/spec_tests.py b/vmdb/spec_tests.py deleted file mode 100644 index 635079a..0000000 --- a/vmdb/spec_tests.py +++ /dev/null @@ -1,109 +0,0 @@ -# Copyright 2017 Lars Wirzenius -# -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU General Public License as published by -# the Free Software Foundation, either version 3 of the License, or -# (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with this program. If not, see <http://www.gnu.org/licenses/>. -# -# =*= License: GPL-3+ =*= - - -import io -import os -import tempfile -import unittest - -import yaml - -import vmdb - - -class SpecTests(unittest.TestCase): - - spec_yaml = ''' - steps: - - step: foo - arg: "{{ var1 }}" - - step: bar - ''' - - def setUp(self): - self.filename = write_temp_file(bytes(self.spec_yaml, 'ascii')) - self.spec = vmdb.Spec() - - def tearDown(self): - if os.path.exists(self.filename): - os.remove(self.filename) - - def test_loads_spec(self): - self.spec.load_file(self.filename) - self.assertEqual(self.spec.as_dict(), as_dict(self.spec_yaml)) - - def test_expands_templates(self): - self.spec.load_file(self.filename) - params = { - 'var1': 'value1', - } - steps = self.spec.get_steps(params) - self.assertEqual( - steps, - [ - { - 'step': 'foo', - 'arg': 'value1', - }, - { - 'step': 'bar', - }, - ] - ) - -class ExpandTemplatesTests(unittest.TestCase): - - def test_raises_assert_if_given_incomprehensible_value(self): - with self.assertRaises(AssertionError): - vmdb.expand_templates(None, {}) - - def test_returns_same_given_string_without_template(self): - self.assertEqual(vmdb.expand_templates('foo', {}), 'foo') - - def test_expands_simple_string_template(self): - params = { - 'foo': 'bar', - } - self.assertEqual(vmdb.expand_templates('{{ foo }}', params), 'bar') - - def test_expands_list_of_templates(self): - params = { - 'foo': 'bar', - } - self.assertEqual(vmdb.expand_templates(['{{ foo }}'], params), ['bar']) - - def test_expands_dict_of_templates(self): - params = { - 'foo': 'bar', - } - self.assertEqual( - vmdb.expand_templates({'key': '{{ foo }}'}, params), - {'key': 'bar'} - ) - - -def write_temp_file(data): - fd, filename = tempfile.mkstemp() - os.write(fd, data) - os.close(fd) - return filename - - -def as_dict(yaml_text): - with io.StringIO(yaml_text) as f: - return yaml.safe_load(f) |