mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 17:33:54 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
# coding: utf-8
|
|
|
|
# Import Salt Testing libs
|
|
from __future__ import absolute_import
|
|
import integration
|
|
|
|
# Import Salt libs
|
|
import salt.wheel
|
|
|
|
|
|
class KeyWheelModuleTest(integration.TestCase, integration.AdaptedConfigurationTestCaseMixIn):
|
|
def setUp(self):
|
|
self.wheel = salt.wheel.Wheel(dict(self.get_config('client_config')))
|
|
|
|
def test_list_all(self):
|
|
ret = self.wheel.call_func('key.list_all')
|
|
self.assertEqual(ret, {
|
|
'local': [
|
|
'master.pem',
|
|
'master.pub',
|
|
'minion.pem',
|
|
'minion.pub',
|
|
'minion_master.pub',
|
|
'syndic_master.pub'
|
|
],
|
|
'minions_rejected': [],
|
|
'minions_pre': [],
|
|
'minions_denied': [],
|
|
'minions': ['minion', 'sub_minion'],
|
|
})
|
|
|
|
def test_gen(self):
|
|
ret = self.wheel.call_func('key.gen', id_='soundtechniciansrock')
|
|
|
|
self.assertIn('pub', ret)
|
|
self.assertIn('priv', ret)
|
|
self.assertTrue(
|
|
ret.get('pub', '').startswith('-----BEGIN PUBLIC KEY-----'))
|
|
self.assertTrue(
|
|
ret.get('priv', '').startswith('-----BEGIN RSA PRIVATE KEY-----'))
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(KeyWheelModuleTest, needs_daemon=True)
|