saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

more work on the wheeltest script

Browse source
This commit is contained in:
Thomas S Hatch 2012-10-29 22:22:19 -06:00
parent 87cef23b34
commit 0ccb4d9b9b
1 changed files with 19 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
tests/wheeltest.py
View file

@ -30,7 +30,13 @@ def parse():
options, args = parser.parse_opts()
cli = options.__dict__()
for arg in args:
if '=' in arg:
comps = arg.split('=')
cli[comps[0]] = comps[1]
return cli
class Wheeler(object):
@ -38,6 +44,7 @@ class Wheeler(object):
Set up communication with the wheel interface
'''
def __init__(self, cli):
self.cli = cli
self.opts = salt.config.master_config('/etc/salt')
self.wheel = salt.wheel.Wheel(self.opts)
@ -45,20 +52,17 @@ class Wheeler(object):
'''
Fill in the blanks for the eauth system
'''
if self.options.eauth:
resolver = salt.auth.Resolver(self.config)
res = resolver.cli(self.options.eauth)
if self.options.mktoken and res:
tok = resolver.token_cli(
self.options.eauth,
res
)
if tok:
kwargs['token'] = tok.get('token', '')
if not res:
sys.exit(2)
kwargs.update(res)
kwargs['eauth'] = self.options.eauth
if self.cli['eauth']:
res = resolver.cli(self.options.eauth)
self.cli.update(res)
def run(self):
'''
Execute the wheel call
'''
print wheel.master(self.cli['fun'], **self.cli)
print wheel.master_call('key.list_all', username='thatch', eauth='pam', password='idfsuhgsklfdn')
if __name__ == '__main__':
wheeler = Wheeler(parse())
wheeler.run()