salt/tests/unit/modules/test_aliases.py
rallytime 3273bbdab7
Merge branch '2017.7' into '2018.3'
Conflicts:
  - doc/ref/configuration/master.rst
  - doc/ref/modules/all/index.rst
  - doc/topics/grains/index.rst
  - doc/topics/releases/2016.3.4.rst
  - doc/topics/spm/spm_formula.rst
  - doc/topics/tutorials/cron.rst
  - doc/topics/tutorials/index.rst
  - doc/topics/tutorials/stormpath.rst
  - salt/engines/slack.py
  - salt/log/handlers/fluent_mod.py
  - salt/modules/cyg.py
  - salt/modules/junos.py
  - salt/modules/namecheap_dns.py
  - salt/modules/namecheap_domains.py
  - salt/modules/namecheap_ns.py
  - salt/modules/namecheap_ssl.py
  - salt/modules/namecheap_users.py
  - salt/modules/reg.py
  - salt/modules/tomcat.py
  - salt/modules/vault.py
  - salt/modules/win_file.py
  - salt/modules/zpool.py
  - salt/output/highstate.py
  - salt/renderers/pass.py
  - salt/runners/cache.py
  - salt/states/boto_apigateway.py
  - salt/states/boto_iam.py
  - salt/states/boto_route53.py
  - salt/states/msteams.py
  - salt/states/reg.py
  - salt/states/win_iis.py
  - tests/integration/modules/test_cmdmod.py
  - tests/integration/states/test_user.py
  - tests/support/helpers.py
  - tests/unit/cloud/clouds/test_openstack.py
  - tests/unit/fileserver/test_gitfs.py
  - tests/unit/modules/test_junos.py
  - tests/unit/pillar/test_git.py
  - tests/unit/states/test_win_path.py
  - tests/unit/test_pillar.py
  - tests/unit/utils/test_format_call.py
  - tests/unit/utils/test_utils.py
  - tests/unit/utils/test_warnings.py
2018-06-01 14:54:12 -04:00

170 lines
6.0 KiB
Python

# -*- coding: utf-8 -*-
'''
:codeauthor: Nicole Thomas <nicole@saltstack.com>
'''
# Import python libs
from __future__ import absolute_import, print_function, unicode_literals
# Import Salt Libs
import salt.modules.aliases as aliases
from salt.exceptions import SaltInvocationError
# Import Salt Testing Libs
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.unit import TestCase, skipIf
from tests.support.mock import MagicMock, patch, NO_MOCK, NO_MOCK_REASON
@skipIf(NO_MOCK, NO_MOCK_REASON)
class AliasesTestCase(TestCase, LoaderModuleMockMixin):
'''
TestCase for salt.modules.aliases module
'''
mock_alias = [('foo', 'bar@example.com', '')]
mock_alias_mult = [('foo', 'bar@example.com', ''),
('hello', 'world@earth.com, earth@world.com', '')]
def setup_loader_modules(self):
return {aliases: {}}
def test_list_aliases(self):
'''
Tests the return of a file containing one alias
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias)):
ret = {'foo': 'bar@example.com'}
self.assertEqual(aliases.list_aliases(), ret)
def test_list_aliases_mult(self):
'''
Tests the return of a file containing multiple aliases
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias_mult)):
ret = {'foo': 'bar@example.com',
'hello': 'world@earth.com, earth@world.com'}
self.assertEqual(aliases.list_aliases(), ret)
def test_get_target(self):
'''
Tests the target returned by an alias with one target
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias)):
ret = 'bar@example.com'
self.assertEqual(aliases.get_target('foo'), ret)
def test_get_target_mult(self):
'''
Tests the target returned by an alias with multiple targets
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias_mult)):
ret = 'world@earth.com, earth@world.com'
self.assertEqual(aliases.get_target('hello'), ret)
def test_get_target_no_alias(self):
'''
Tests return of an alias doesn't exist
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias)):
self.assertEqual(aliases.get_target('pizza'), '')
def test_has_target(self):
'''
Tests simple return known alias and target
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias)):
ret = aliases.has_target('foo', 'bar@example.com')
self.assertTrue(ret)
def test_has_target_no_alias(self):
'''
Tests return of empty alias and known target
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias)):
ret = aliases.has_target('', 'bar@example.com')
self.assertFalse(ret)
def test_has_target_no_target(self):
'''
Tests return of known alias and empty target
'''
self.assertRaises(SaltInvocationError, aliases.has_target, 'foo', '')
def test_has_target_mult(self):
'''
Tests return of multiple targets to one alias
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias_mult)):
ret = aliases.has_target('hello',
'world@earth.com, earth@world.com')
self.assertTrue(ret)
def test_has_target_mult_differs(self):
'''
Tests return of multiple targets to one alias in opposite order
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias_mult)):
ret = aliases.has_target('hello',
'earth@world.com, world@earth.com')
self.assertFalse(ret)
def test_has_target_list_mult(self):
'''
Tests return of target as same list to know alias
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias_mult)):
ret = aliases.has_target('hello', ['world@earth.com',
'earth@world.com'])
self.assertTrue(ret)
def test_has_target_list_mult_differs(self):
'''
Tests return of target as differing list to known alias
'''
with patch('salt.modules.aliases.__parse_aliases',
MagicMock(return_value=self.mock_alias_mult)):
ret = aliases.has_target('hello', ['world@earth.com',
'mars@space.com'])
self.assertFalse(ret)
def test_set_target_equal(self):
'''
Tests return when target is already present
'''
with patch('salt.modules.aliases.get_target', MagicMock(return_value='bar@example.com')):
alias = 'foo'
target = 'bar@example.com'
ret = aliases.set_target(alias, target)
self.assertTrue(ret)
def test_set_target_empty_alias(self):
'''
Tests return of empty alias
'''
self.assertRaises(SaltInvocationError, aliases.set_target, '', 'foo')
def test_set_target_empty_target(self):
'''
Tests return of known alias and empty target
'''
self.assertRaises(SaltInvocationError, aliases.set_target, 'foo', '')
def test_rm_alias_absent(self):
'''
Tests return when alias is not present
'''
with patch('salt.modules.aliases.get_target', MagicMock(return_value='')):
ret = aliases.rm_alias('foo')
self.assertTrue(ret)