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 libs
import salt.sdb.yaml as sdb
from tests.support.mock import MagicMock, patch

# Import Salt Testing libs
from tests.support.unit import TestCase


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")