saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Added tests for salt.pillar.extra_minion_data_in_pillar

Browse source
This commit is contained in:
Alexandru Bleotu 2017-09-28 06:31:35 -04:00
parent 1a12d5cb30
commit 998c4a95fa
1 changed files with 55 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
tests/unit/pillar/test_extra_minion_data_in_pillar.py Normal file
View file

@ -0,0 +1,55 @@
# -*- coding: utf-8 -*-
# Import python libs
from __future__ import absolute_import
# Import Salt Testing libs
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.unit import TestCase, skipIf
from tests.support.mock import NO_MOCK, NO_MOCK_REASON, MagicMock
# Import Salt Libs
from salt.pillar import extra_minion_data_in_pillar
@skipIf(NO_MOCK, NO_MOCK_REASON)
class ExtraMinionDataInPillarTestCase(TestCase, LoaderModuleMockMixin):
'''
Test cases for salt.pillar.extra_minion_data_in_pillar
'''
def setup_loader_modules(self):
return {
extra_minion_data_in_pillar : {
'__virtual__': True,
}
}
def setUp(self):
self.pillar = MagicMock()
self.extra_minion_data = {'key1': {'subkey1': 'value1'},
'key2': {'subkey2': {'subsubkey2': 'value2'}},
'key3': 'value3',
'key4': {'subkey4': 'value4'}}
def test_extra_values_none_or_empty(self):
ret = extra_minion_data_in_pillar.ext_pillar('fake_id', self.pillar,
'fake_include', None)
self.assertEqual(ret, {})
ret = extra_minion_data_in_pillar.ext_pillar('fake_id', self.pillar,
'fake_include', {})
self.assertEqual(ret, {})
def test_include_all(self):
ret = extra_minion_data_in_pillar.ext_pillar(
'fake_id', self.pillar, '<all>', self.extra_minion_data)
self.assertEqual(ret, self.extra_minion_data)
def test_include_specific_keys(self):
# Tests partially existing key, key with and without subkey,
ret = extra_minion_data_in_pillar.ext_pillar(
'fake_id', self.pillar,
include=['key1:subkey1', 'key2:subkey3', 'key3', 'key4'],
extra_minion_data=self.extra_minion_data)
self.assertEqual(ret, {'key1': {'subkey1': 'value1'},
'key3': 'value3',
'key4': {'subkey4': 'value4'}})