mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 09:23:56 +00:00
100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt testing libs
|
|
from salttesting import TestCase
|
|
from salttesting.mock import Mock, patch, mock_open
|
|
|
|
# Import Salt libs
|
|
from salt.states import kapacitor
|
|
|
|
kapacitor.__opts__ = {'test': False}
|
|
kapacitor.__salt__ = {}
|
|
kapacitor.__env__ = 'test'
|
|
|
|
|
|
def _present(name='testname',
|
|
tick_script='/tmp/script.tick',
|
|
task_type='stream',
|
|
database='testdb',
|
|
retention_policy='default',
|
|
enable=True,
|
|
task=None,
|
|
define_result=True,
|
|
enable_result=True,
|
|
disable_result=True,
|
|
script='test'):
|
|
'''
|
|
Run a "kapacitor.present" state after setting up mocks, and return the
|
|
state return value as well as the mocks to make assertions.
|
|
'''
|
|
get_mock = Mock(return_value=task)
|
|
|
|
if isinstance(define_result, bool):
|
|
define_result = {'success': define_result}
|
|
define_mock = Mock(return_value=define_result)
|
|
|
|
if isinstance(enable_result, bool):
|
|
enable_result = {'success': enable_result}
|
|
enable_mock = Mock(return_value=enable_result)
|
|
|
|
if isinstance(disable_result, bool):
|
|
disable_result = {'success': disable_result}
|
|
disable_mock = Mock(return_value=disable_result)
|
|
|
|
with patch.dict(kapacitor.__salt__, {
|
|
'kapacitor.get_task': get_mock,
|
|
'kapacitor.define_task': define_mock,
|
|
'kapacitor.enable_task': enable_mock,
|
|
'kapacitor.disable_task': disable_mock,
|
|
}):
|
|
with patch('salt.utils.fopen', mock_open(read_data=script)) as open_mock:
|
|
retval = kapacitor.task_present(name, tick_script, task_type=task_type,
|
|
database=database, retention_policy=retention_policy, enable=enable)
|
|
|
|
return retval, get_mock, define_mock, enable_mock, disable_mock
|
|
|
|
|
|
class KapacitorTestCase(TestCase):
|
|
def test_task_present_new_task(self):
|
|
ret, get_mock, define_mock, enable_mock, _ = _present()
|
|
get_mock.assert_called_once_with('testname')
|
|
define_mock.assert_called_once_with('testname', '/tmp/script.tick',
|
|
database='testdb', retention_policy='default', task_type='stream')
|
|
enable_mock.assert_called_once_with('testname')
|
|
self.assertIn('TICKscript diff', ret['changes'])
|
|
self.assertIn('enabled', ret['changes'])
|
|
self.assertEqual(True, ret['changes']['enabled']['new'])
|
|
|
|
def test_task_present_existing_task(self):
|
|
old_task = {'script': 'old_task', 'enabled': True}
|
|
ret, get_mock, define_mock, enable_mock, _ = _present(task=old_task)
|
|
get_mock.assert_called_once_with('testname')
|
|
define_mock.assert_called_once_with('testname', '/tmp/script.tick',
|
|
database='testdb', retention_policy='default', task_type='stream')
|
|
self.assertEqual(False, enable_mock.called)
|
|
self.assertIn('TICKscript diff', ret['changes'])
|
|
self.assertNotIn('enabled', ret['changes'])
|
|
|
|
def test_task_present_existing_task_not_enabled(self):
|
|
old_task = {'script': 'test', 'enabled': False}
|
|
ret, get_mock, define_mock, enable_mock, _ = _present(task=old_task)
|
|
get_mock.assert_called_once_with('testname')
|
|
self.assertEqual(False, define_mock.called)
|
|
enable_mock.assert_called_once_with('testname')
|
|
self.assertNotIn('diff', ret['changes'])
|
|
self.assertIn('enabled', ret['changes'])
|
|
self.assertEqual(True, ret['changes']['enabled']['new'])
|
|
|
|
def test_task_present_disable_existing_task(self):
|
|
old_task = {'script': 'test', 'enabled': True}
|
|
ret, get_mock, define_mock, _, disable_mock = _present(task=old_task, enable=False)
|
|
get_mock.assert_called_once_with('testname')
|
|
self.assertEqual(False, define_mock.called)
|
|
disable_mock.assert_called_once_with('testname')
|
|
self.assertNotIn('diff', ret['changes'])
|
|
self.assertIn('enabled', ret['changes'])
|
|
self.assertEqual(False, ret['changes']['enabled']['new'])
|