mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 01:18:58 +00:00
0c4e38ced4
* Fix the saltutil.wheel function and add integration tests Fixes #32446 * Provide a more useful message when command is run on non-local minion
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
Integration tests for the saltutil module.
|
|
'''
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing libs
|
|
from salttesting.helpers import ensure_in_syspath
|
|
|
|
ensure_in_syspath('../../')
|
|
|
|
# Import Salt libs
|
|
import integration
|
|
|
|
|
|
class SaltUtilModuleTest(integration.ModuleCase):
|
|
'''
|
|
Testcase for the saltutil execution module
|
|
'''
|
|
|
|
# Tests for the wheel function
|
|
|
|
def test_wheel_just_function(self):
|
|
'''
|
|
Tests using the saltutil.wheel function when passing only a function.
|
|
'''
|
|
ret = self.run_function('saltutil.wheel', ['minions.connected'])
|
|
self.assertEqual(ret, ['minion', 'sub_minion'])
|
|
|
|
def test_wheel_with_arg(self):
|
|
'''
|
|
Tests using the saltutil.wheel function when passing a function and an arg.
|
|
'''
|
|
ret = self.run_function('saltutil.wheel', ['key.list', 'minion'])
|
|
self.assertEqual(ret, {})
|
|
|
|
def test_wheel_no_arg_raise_error(self):
|
|
'''
|
|
Tests using the saltutil.wheel function when passing a function that requires
|
|
an arg, but one isn't supplied.
|
|
'''
|
|
self.assertRaises(TypeError, 'saltutil.wheel', ['key.list'])
|
|
|
|
def test_wheel_with_kwarg(self):
|
|
'''
|
|
Tests using the saltutil.wheel function when passing a function and a kwarg.
|
|
This function just generates a key pair, but doesn't do anything with it. We
|
|
just need this for testing purposes.
|
|
'''
|
|
ret = self.run_function('saltutil.wheel', ['key.gen'], keysize=1024)
|
|
self.assertIn('pub', ret)
|
|
self.assertIn('priv', ret)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(SaltUtilModuleTest)
|