mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
# Import python libs
|
|
import sys
|
|
|
|
# Import salt libs
|
|
from saltunittest import TestLoader, TextTestRunner
|
|
import integration
|
|
from integration import TestDaemon
|
|
|
|
|
|
class StdTest(integration.ModuleCase):
|
|
'''
|
|
Test standard client calls
|
|
'''
|
|
def test_cli(self):
|
|
'''
|
|
Test cli function
|
|
'''
|
|
cmd_iter = self.client.cmd_cli(
|
|
'minion',
|
|
'test.kwarg',
|
|
['foo=bar', 'baz=quo'],
|
|
kwarg={'qux': 'quux'}
|
|
)
|
|
for ret in cmd_iter:
|
|
data = ret['minion']['ret']
|
|
self.assertIn('foo', data)
|
|
self.assertIn('baz', data)
|
|
self.assertIn('qux', data)
|
|
self.assertEqual(data['foo'], 'bar')
|
|
self.assertEqual(data['baz'], 'quo')
|
|
self.assertEqual(data['qux'], 'quux')
|
|
|
|
def test_iter(self):
|
|
'''
|
|
test cmd_iter
|
|
'''
|
|
cmd_iter = self.client.cmd_iter(
|
|
'minion',
|
|
'test.kwarg',
|
|
['foo=bar', 'baz=quo'],
|
|
kwarg={'qux': 'quux'}
|
|
)
|
|
for ret in cmd_iter:
|
|
data = ret['minion']['ret']
|
|
self.assertIn('foo', data)
|
|
self.assertIn('baz', data)
|
|
self.assertIn('qux', data)
|
|
self.assertEqual(data['foo'], 'bar')
|
|
self.assertEqual(data['baz'], 'quo')
|
|
self.assertEqual(data['qux'], 'quux')
|
|
|
|
def test_iter_no_block(self):
|
|
'''
|
|
test cmd_iter_no_block
|
|
'''
|
|
cmd_iter = self.client.cmd_iter_no_block(
|
|
'minion',
|
|
'test.kwarg',
|
|
['foo=bar', 'baz=quo'],
|
|
kwarg={'qux': 'quux'}
|
|
)
|
|
for ret in cmd_iter:
|
|
if ret is None:
|
|
continue
|
|
data = ret['minion']['ret']
|
|
self.assertIn('foo', data)
|
|
self.assertIn('baz', data)
|
|
self.assertIn('qux', data)
|
|
self.assertEqual(data['foo'], 'bar')
|
|
self.assertEqual(data['baz'], 'quo')
|
|
self.assertEqual(data['qux'], 'quux')
|
|
|
|
def test_full_returns(self):
|
|
'''
|
|
test cmd_iter
|
|
'''
|
|
ret = self.client.cmd_full_return(
|
|
'minion',
|
|
'test.kwarg',
|
|
['foo=bar', 'baz=quo'],
|
|
kwarg={'qux': 'quux'}
|
|
)
|
|
data = ret['minion']
|
|
self.assertIn('foo', data['ret'])
|
|
self.assertIn('baz', data['ret'])
|
|
self.assertIn('qux', data['ret'])
|
|
self.assertEqual(data['ret']['foo'], 'bar')
|
|
self.assertEqual(data['ret']['baz'], 'quo')
|
|
self.assertEqual(data['ret']['qux'], 'quux')
|
|
|
|
if __name__ == "__main__":
|
|
loader = TestLoader()
|
|
tests = loader.loadTestsFromTestCase(StdTest)
|
|
print('Setting up Salt daemons to execute tests')
|
|
with TestDaemon():
|
|
runner = TextTestRunner(verbosity=1).run(tests)
|
|
sys.exit(runner.wasSuccessful())
|