salt/tests/unit/modules/test_shadow.py

73 lines
1.9 KiB
Python

# -*- coding: utf-8 -*-
'''
:codeauthor: :email:`Erik Johnson <erik@saltstack.com>`
'''
# Import Pytohn libs
from __future__ import absolute_import, unicode_literals, print_function
# Import Salt Testing libs
import salt.utils.platform
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.unit import TestCase, skipIf
from tests.support.helpers import skip_if_not_root
# Import salt libs
try:
import salt.modules.shadow as shadow
HAS_SHADOW = True
except ImportError:
HAS_SHADOW = False
# Import 3rd-party libs
from salt.ext import six
_PASSWORD = 'lamepassword'
# Not testing blowfish as it is not available on most Linux distros
_HASHES = dict(
md5=dict(
pw_salt='TgIp9OTu',
pw_hash='$1$TgIp9OTu$.d0FFP6jVi5ANoQmk6GpM1'
),
sha256=dict(
pw_salt='3vINbSrC',
pw_hash='$5$3vINbSrC$hH8A04jAY3bG123yU4FQ0wvP678QDTvWBhHHFbz6j0D'
),
sha512=dict(
pw_salt='PiGA3V2o',
pw_hash='$6$PiGA3V2o$/PrntRYufz49bRV/V5Eb1V6DdHaS65LB0fu73Tp/xxmDFr6HWJKptY2TvHRDViXZugWpnAcOnrbORpOgZUGTn.'
),
)
@skipIf(not salt.utils.platform.is_linux(), 'minion is not Linux')
@skipIf(not HAS_SHADOW, 'shadow module is not available')
class LinuxShadowTest(TestCase, LoaderModuleMockMixin):
def setup_loader_modules(self):
return {shadow: {}}
def test_gen_password(self):
'''
Test shadow.gen_password
'''
self.assertTrue(HAS_SHADOW)
for algorithm, hash_info in six.iteritems(_HASHES):
self.assertEqual(
shadow.gen_password(
_PASSWORD,
crypt_salt=hash_info['pw_salt'],
algorithm=algorithm
),
hash_info['pw_hash']
)
@skip_if_not_root
def test_list_users(self):
'''
Test if it returns a list of all users
'''
self.assertTrue(shadow.list_users())