mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
8cdb9ea54f
This makes the 2.x usage invalid syntax and forces the use of print as a function. This adds the import to the files which I've updated in the last couple of days but forgot to add it.
117 lines
3.1 KiB
Python
117 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
# Import Salt Testing libs
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mixins import AdaptedConfigurationTestCaseMixin
|
|
|
|
# Import Salt libs
|
|
import salt.auth
|
|
import salt.wheel
|
|
import salt.utils.platform
|
|
|
|
|
|
class WheelModuleTest(TestCase, AdaptedConfigurationTestCaseMixin):
|
|
|
|
eauth_creds = {
|
|
'username': 'saltdev_auto',
|
|
'password': 'saltdev',
|
|
'eauth': 'auto',
|
|
}
|
|
|
|
def setUp(self):
|
|
'''
|
|
Configure an eauth user to test with
|
|
'''
|
|
self.wheel = salt.wheel.Wheel(dict(self.get_config('client_config')))
|
|
|
|
def tearDown(self):
|
|
del self.wheel
|
|
|
|
def test_master_call(self):
|
|
'''
|
|
Test executing master_call with lowdata
|
|
|
|
The choice of using key.list_all for this is arbitrary and should be
|
|
changed to some mocked function that is more testing friendly.
|
|
'''
|
|
low = {
|
|
'client': 'wheel',
|
|
'fun': 'key.list_all',
|
|
'print_event': False
|
|
}
|
|
low.update(self.eauth_creds)
|
|
|
|
self.wheel.master_call(**low)
|
|
|
|
def test_token(self):
|
|
'''
|
|
Test executing master_call with lowdata
|
|
|
|
The choice of using key.list_all for this is arbitrary and should be
|
|
changed to some mocked function that is more testing friendly.
|
|
'''
|
|
auth = salt.auth.LoadAuth(dict(self.get_config('client_config')))
|
|
token = auth.mk_token(self.eauth_creds)
|
|
|
|
token = auth.mk_token({
|
|
'username': 'saltdev_auto',
|
|
'password': 'saltdev',
|
|
'eauth': 'auto',
|
|
})
|
|
|
|
self.wheel.master_call(**{
|
|
'client': 'wheel',
|
|
'fun': 'key.list_all',
|
|
'token': token['token'],
|
|
'print_event': False,
|
|
})
|
|
|
|
def test_cmd_sync(self):
|
|
low = {
|
|
'client': 'wheel',
|
|
'fun': 'key.list_all',
|
|
'print_event': False,
|
|
}
|
|
low.update(self.eauth_creds)
|
|
|
|
self.wheel.cmd_sync(low)
|
|
|
|
# Remove this skipIf when Issue #39616 is resolved
|
|
# https://github.com/saltstack/salt/issues/39616
|
|
@skipIf(salt.utils.platform.is_windows(),
|
|
'Causes pickling error on Windows: Issue #39616')
|
|
def test_cmd_async(self):
|
|
low = {
|
|
'client': 'wheel_async',
|
|
'fun': 'key.list_all',
|
|
'print_event': False,
|
|
}
|
|
low.update(self.eauth_creds)
|
|
|
|
self.wheel.cmd_async(low)
|
|
|
|
def test_cmd_sync_w_arg(self):
|
|
low = {
|
|
'fun': 'key.finger',
|
|
'match': '*',
|
|
'print_event': False,
|
|
}
|
|
low.update(self.eauth_creds)
|
|
|
|
ret = self.wheel.cmd_sync(low)
|
|
self.assertIn('return', ret.get('data', {}))
|
|
|
|
def test_wildcard_auth(self):
|
|
low = {
|
|
'username': 'the_s0und_of_t3ch',
|
|
'password': 'willrockyou',
|
|
'eauth': 'auto',
|
|
'fun': 'key.list_all',
|
|
'print_event': False,
|
|
}
|
|
|
|
self.wheel.cmd_sync(low)
|