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'):
get_mock = Mock(return_value=task)
define_mock = Mock(return_value=define_result)
enable_mock = Mock(return_value=enable_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
def _present(
name="testname",
tick_script="/tmp/script.tick",
task_type="stream",
database="testdb",
retention_policy="default",
dbrps=None,
enable=True,
task=None,
define_result=True,
enable_result=True,
disable_result=True,
script="testscript",
):
"""
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.files.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,
dbrps=dbrps,
)
return retval, get_mock, define_mock, enable_mock, disable_mock
def _present(name='testname',
tick_script='/tmp/script.tick',
task_type='stream',
database='testdb',
retention_policy='default',
dbrps=None,
enable=True,
task=None,
define_result=True,
enable_result=True,
disable_result=True,
script='testscript'):
'''
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.files.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,
dbrps=dbrps)
return retval, get_mock, define_mock, enable_mock, disable_mock
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