salt/tests/integration/shell/key.py
2014-07-14 15:22:38 -04:00

260 lines
8.0 KiB
Python

# -*- coding: utf-8 -*-
# Import python libs
import os
import yaml
import shutil
import tempfile
# Import Salt Testing libs
from salttesting.helpers import ensure_in_syspath
from salttesting import skipIf
ensure_in_syspath('../../')
# Import salt libs
import integration
class KeyTest(integration.ShellCase, integration.ShellCaseCommonTestsMixIn):
'''
Test salt-key script
'''
_call_binary_ = 'salt-key'
def test_list_accepted_args(self):
'''
test salt-key -l for accepted arguments
'''
for key in ('acc', 'pre', 'un', 'rej'):
# These should not trigger any error
data = self.run_key('-l {0}'.format(key), catch_stderr=True)
self.assertNotIn('error:', '\n'.join(data[1]))
data = self.run_key('-l foo-{0}'.format(key), catch_stderr=True)
self.assertIn('error:', '\n'.join(data[1]))
def test_list_all(self):
'''
test salt-key -L
'''
data = self.run_key('-L')
expect = None
if self.master_opts['transport'] == 'zeromq':
expect = [
'Accepted Keys:',
'minion',
'sub_minion',
'Unaccepted Keys:',
'Rejected Keys:'
]
elif self.master_opts['transport'] == 'raet':
expect = [
'Accepted Keys:',
'master',
'minion',
'sub_minion',
'Unaccepted Keys:',
'Rejected Keys:'
]
self.assertEqual(data, expect)
def test_list_json_out(self):
'''
test salt-key -L --json-out
'''
data = self.run_key('-L --out json')
expect = None
if self.master_opts['transport'] == 'zeromq':
expect = [
'{',
' "minions_rejected": [], ',
' "minions_pre": [], ',
' "minions": [',
' "minion", ',
' "sub_minion"',
' ]',
'}',
]
elif self.master_opts['transport'] == 'raet':
expect = [
'{',
' "accepted": [',
' "master", ',
' "minion", ',
' "sub_minion"',
' ], ',
' "rejected": [], ',
' "pending": []',
'}'
]
self.assertEqual(data, expect)
def test_list_yaml_out(self):
'''
test salt-key -L --yaml-out
'''
data = self.run_key('-L --out yaml')
expect = []
if self.master_opts['transport'] == 'zeromq':
expect = [
'minions:',
'- minion',
'- sub_minion',
'minions_pre: []',
'minions_rejected: []',
]
elif self.master_opts['transport'] == 'raet':
expect = [
'accepted:',
'- master',
'- minion',
'- sub_minion',
'pending: []',
'rejected: []'
]
self.assertEqual(data, expect)
def test_list_raw_out(self):
'''
test salt-key -L --raw-out
'''
data = self.run_key('-L --out raw')
expect = None
if self.master_opts['transport'] == 'zeromq':
expect = [
"{'minions_rejected': [], 'minions_pre': [], "
"'minions': ['minion', 'sub_minion']}"
]
elif self.master_opts['transport'] == 'raet':
expected_txt = (
'{\'accepted\': '
'[\'master\', \'minion\', \'sub_minion\'], '
'\'rejected\': [], \'pending\': []}'
)
expect = [expected_txt]
self.assertEqual(data, expect)
def test_list_acc(self):
'''
test salt-key -l
'''
data = self.run_key('-l acc')
if self.master_opts['transport'] == 'zeromq':
self.assertEqual(
data,
['Accepted Keys:', 'minion', 'sub_minion']
)
elif self.master_opts['transport'] == 'raet':
self.assertEqual(
data,
[
'minions:',
' - master',
' - minion',
' - sub_minion'
]
)
def test_list_un(self):
'''
test salt-key -l
'''
data = self.run_key('-l un')
expect = None
if self.master_opts['transport'] == 'zeromq':
expect = ['Unaccepted Keys:']
elif self.master_opts['transport'] == 'raet':
expect = ['minions_pre:']
self.assertEqual(
data,
expect
)
def test_keys_generation(self):
tempdir = tempfile.mkdtemp(dir=integration.SYS_TMP_DIR)
arg_str = '--gen-keys minibar --gen-keys-dir {0}'.format(tempdir)
self.run_key(arg_str)
try:
key_names = None
if self.master_opts['transport'] == 'zeromq':
key_names = ('minibar.pub', 'minibar.pem')
elif self.master_opts['transport'] == 'raet':
key_names = ('minibar.key',)
for fname in key_names:
self.assertTrue(os.path.isfile(os.path.join(tempdir, fname)))
finally:
shutil.rmtree(tempdir)
@skipIf(os.geteuid() != 0, 'you must be root to run this test')
def test_keys_generation_no_configdir(self):
tempdir = tempfile.mkdtemp(dir=integration.SYS_TMP_DIR)
arg_str = '--gen-keys minibar --gen-keys-dir {0}'.format(tempdir)
self.run_script('salt-key', arg_str)
try:
key_names = None
if self.master_opts['transport'] == 'zeromq':
key_names = ('minibar.pub', 'minibar.pem')
elif self.master_opts['transport'] == 'raet':
key_names = ('minibar.key',)
for fname in key_names:
self.assertTrue(os.path.isfile(os.path.join(tempdir, fname)))
finally:
shutil.rmtree(tempdir)
def test_keys_generation_keysize_minmax(self):
tempdir = tempfile.mkdtemp(dir=integration.SYS_TMP_DIR)
arg_str = '--gen-keys minion --gen-keys-dir {0}'.format(tempdir)
try:
data, error = self.run_key(
arg_str + ' --keysize=1024', catch_stderr=True
)
self.assertIn(
'salt-key: error: The minimum value for keysize is 2048', error
)
data, error = self.run_key(
arg_str + ' --keysize=32769', catch_stderr=True
)
self.assertIn(
'salt-key: error: The maximum value for keysize is 32768',
error
)
finally:
shutil.rmtree(tempdir)
def test_issue_7754(self):
old_cwd = os.getcwd()
config_dir = os.path.join(integration.TMP, 'issue-7754')
if not os.path.isdir(config_dir):
os.makedirs(config_dir)
os.chdir(config_dir)
config_file_name = 'master'
config = yaml.load(
open(self.get_config_file_path(config_file_name), 'r').read()
)
config['log_file'] = 'file:///dev/log/LOG_LOCAL3'
open(os.path.join(config_dir, config_file_name), 'w').write(
yaml.dump(config, default_flow_style=False)
)
ret = self.run_script(
self._call_binary_,
'--config-dir {0} -L'.format(
config_dir
),
timeout=15
)
try:
self.assertIn('minion', '\n'.join(ret))
self.assertFalse(os.path.isdir(os.path.join(config_dir, 'file:')))
finally:
os.chdir(old_cwd)
if os.path.isdir(config_dir):
shutil.rmtree(config_dir)
if __name__ == '__main__':
from integration import run_tests
run_tests(KeyTest)