mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
123 lines
3.7 KiB
Python
123 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
Unit tests for salt.utils.yamlloader.SaltYamlSafeLoader
|
|
'''
|
|
|
|
# Import python libs
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
import textwrap
|
|
|
|
# Import Salt Libs
|
|
from yaml.constructor import ConstructorError
|
|
from salt.utils.yamlloader import SaltYamlSafeLoader
|
|
import salt.utils.files
|
|
from salt.ext import six
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import patch, NO_MOCK, NO_MOCK_REASON, mock_open
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class YamlLoaderTestCase(TestCase):
|
|
'''
|
|
TestCase for salt.utils.yamlloader module
|
|
'''
|
|
|
|
@staticmethod
|
|
def _render_yaml(data):
|
|
'''
|
|
Takes a YAML string, puts it into a mock file, passes that to the YAML
|
|
SaltYamlSafeLoader and then returns the rendered/parsed YAML data
|
|
'''
|
|
if six.PY2:
|
|
# On Python 2, data read from a filehandle will not already be
|
|
# unicode, so we need to encode it first to properly simulate
|
|
# reading from a file. This is because unicode_literals is imported
|
|
# and all of the data to be used in mock_open will be a unicode
|
|
# type. Encoding will make it a str.
|
|
data = salt.utils.data.encode(data)
|
|
with patch('salt.utils.files.fopen', mock_open(read_data=data)) as mocked_file:
|
|
with salt.utils.files.fopen(mocked_file) as mocked_stream:
|
|
return SaltYamlSafeLoader(mocked_stream).get_data()
|
|
|
|
def test_yaml_basics(self):
|
|
'''
|
|
Test parsing an ordinary path
|
|
'''
|
|
self.assertEqual(
|
|
self._render_yaml(textwrap.dedent('''\
|
|
p1:
|
|
- alpha
|
|
- beta''')),
|
|
{'p1': ['alpha', 'beta']}
|
|
)
|
|
|
|
def test_yaml_merge(self):
|
|
'''
|
|
Test YAML anchors
|
|
'''
|
|
# Simple merge test
|
|
self.assertEqual(
|
|
self._render_yaml(textwrap.dedent('''\
|
|
p1: &p1
|
|
v1: alpha
|
|
p2:
|
|
<<: *p1
|
|
v2: beta''')),
|
|
{'p1': {'v1': 'alpha'}, 'p2': {'v1': 'alpha', 'v2': 'beta'}}
|
|
)
|
|
|
|
# Test that keys/nodes are overwritten
|
|
self.assertEqual(
|
|
self._render_yaml(textwrap.dedent('''\
|
|
p1: &p1
|
|
v1: alpha
|
|
p2:
|
|
<<: *p1
|
|
v1: new_alpha''')),
|
|
{'p1': {'v1': 'alpha'}, 'p2': {'v1': 'new_alpha'}}
|
|
)
|
|
|
|
# Test merging of lists
|
|
self.assertEqual(
|
|
self._render_yaml(textwrap.dedent('''\
|
|
p1: &p1
|
|
v1: &v1
|
|
- t1
|
|
- t2
|
|
p2:
|
|
v2: *v1''')),
|
|
{"p2": {"v2": ["t1", "t2"]}, "p1": {"v1": ["t1", "t2"]}}
|
|
)
|
|
|
|
def test_yaml_duplicates(self):
|
|
'''
|
|
Test that duplicates still throw an error
|
|
'''
|
|
with self.assertRaises(ConstructorError):
|
|
self._render_yaml(textwrap.dedent('''\
|
|
p1: alpha
|
|
p1: beta'''))
|
|
|
|
with self.assertRaises(ConstructorError):
|
|
self._render_yaml(textwrap.dedent('''\
|
|
p1: &p1
|
|
v1: alpha
|
|
p2:
|
|
<<: *p1
|
|
v2: beta
|
|
v2: betabeta'''))
|
|
|
|
def test_yaml_with_unicode_literals(self):
|
|
'''
|
|
Test proper loading of unicode literals
|
|
'''
|
|
self.assertEqual(
|
|
self._render_yaml(textwrap.dedent('''\
|
|
foo:
|
|
a: Д
|
|
b: {'a': u'\\u0414'}''')),
|
|
{'foo': {'a': u'\u0414', 'b': {'a': u'\u0414'}}}
|
|
)
|