Consolidate saltclass pillar unit tests into single module

This commit is contained in:
Erik Johnson 2018-08-10 12:58:36 -05:00
parent a642619fa9
commit 330c21179a
No known key found for this signature in database
GPG Key ID: 5E5583C437808F3F

View File

@ -50,3 +50,31 @@ class SaltclassPillarTestCase(TestCase, LoaderModuleMockMixin):
def test_succeeds(self):
ret = ['default.users', 'default.motd', 'default.empty', 'default', 'roles.app']
self._runner(ret)
@skipIf(NO_MOCK, NO_MOCK_REASON)
class SaltclassPillarTestCaseListExpansion(TestCase, LoaderModuleMockMixin):
'''
Tests for salt.pillar.saltclass variable expansion in list
'''
def setup_loader_modules(self):
return {saltclass: {'__opts__': fake_opts,
'__salt__': fake_salt,
'__grains__': fake_grains
}}
def _runner(self, expected_ret):
full_ret = {}
parsed_ret = []
try:
full_ret = saltclass.ext_pillar(fake_minion_id, fake_pillar, fake_args)
parsed_ret = full_ret['test_list']
# Fail the test if we hit our NoneType error
except TypeError as err:
self.fail(err)
# Else give the parsed content result
self.assertListEqual(parsed_ret, expected_ret)
def test_succeeds(self):
ret = [{'a': '192.168.10.10'}, '192.168.10.20']
self._runner(ret)