summaryrefslogtreecommitdiff
path: root/vmdb/spec_tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'vmdb/spec_tests.py')
-rw-r--r--vmdb/spec_tests.py109
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)