salt/tests/unit/sdb/test_yaml.py

# -*- coding: utf-8 -*-
'''
Test case for the YAML SDB module
'''

# Import python libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Testing libs
from tests.support.unit import skipIf, TestCase
from tests.support.mock import (
    NO_MOCK,
    NO_MOCK_REASON,
    MagicMock,
    patch)

# Import Salt libs
import salt.sdb.yaml as sdb


@skipIf(NO_MOCK, NO_MOCK_REASON)
class TestYamlRenderer(TestCase):
    '''
    Test case for the YAML SDB module
    '''

    def test_plaintext(self):
        '''
        Retrieve a value from the top level of the dictionary
        '''
        plain = {'foo': 'bar'}
        with patch('salt.sdb.yaml._get_values', MagicMock(return_value=plain)):
            self.assertEqual(sdb.get('foo'), 'bar')

    def test_nested(self):
        '''
        Retrieve a value from a nested level of the dictionary
        '''
        plain = {'foo': {'bar': 'baz'}}
        with patch('salt.sdb.yaml._get_values', MagicMock(return_value=plain)):
            self.assertEqual(sdb.get('foo:bar'), 'baz')

    def test_encrypted(self):
        '''
        Assume the content is plaintext if GPG is not configured
        '''
        plain = {'foo': 'bar'}
        with patch('salt.sdb.yaml._decrypt', MagicMock(return_value=plain)):
            with patch('salt.sdb.yaml._get_values', MagicMock(return_value=None)):
                self.assertEqual(sdb.get('foo', profile={'gpg': True}), 'bar')