def test_define_task(self):
cmd_mock = Mock(return_value=True)
with patch.dict(kapacitor.__salt__, {'cmd.retcode': cmd_mock}):
kapacitor.define_task('taskname', '/tmp/script.tick')
cmd_mock.assert_called_once_with(
'kapacitor define -name taskname -tick /tmp/script.tick -type stream'
)
def test_screen_state(self):
config = {'screen_event': True, 'user': '******'}
mock = Mock(return_value=0)
with patch.dict(glxinfo.__salt__, {'cmd.retcode': mock}):
ret = glxinfo.beacon(config)
self.assertEqual(ret, [{'tag': 'screen_event', 'screen_available': True}])
mock.assert_called_once_with('DISPLAY=:0 glxinfo', runas='frank', python_shell=True)
def test_with_user(self):
config = {'states': ['device'], 'user': '******'}
mock = Mock(return_value='* daemon started successfully *\nList of devices attached\nHTC\tdevice',)
with patch.dict(adb.__salt__, {'cmd.run': mock}):
ret = adb.beacon(config)
mock.assert_called_once_with('adb devices', runas='fred')
self.assertEqual(ret, [{'device': 'HTC', 'state': 'device', 'tag': 'device'}])
def test_screen_state(self):
config = {'screen_event': True, 'user': '******'}
mock = Mock(return_value=0)
with patch.dict(glxinfo.__salt__, {'cmd.retcode': mock}):
ret = glxinfo.beacon(config)
self.assertEqual(ret, [{
'tag': 'screen_event',
'screen_available': True
}])
mock.assert_called_once_with('DISPLAY=:0 glxinfo',
runas='frank',
python_shell=True)
def test_with_user(self):
config = {'states': ['device'], 'user': '******'}
mock = Mock(
return_value=
'* daemon started successfully *\nList of devices attached\nHTC\tdevice',
)
with patch.dict(adb.__salt__, {'cmd.run': mock}):
ret = adb.beacon(config)
mock.assert_called_once_with('adb devices', runas='fred')
self.assertEqual(ret, [{
'device': 'HTC',
'state': 'device',
'tag': 'device'
}])
def test_delete_task(self):
cmd_mock = Mock(return_value=True)
with patch.dict(kapacitor.__salt__, {'cmd.retcode': cmd_mock}):
kapacitor.delete_task('taskname')
cmd_mock.assert_called_once_with('kapacitor delete tasks taskname')
def test_disable_task(self):
cmd_mock = Mock(return_value={'retcode': 0})
with patch.dict(kapacitor.__salt__, {'cmd.run_all': cmd_mock}):
kapacitor.disable_task('taskname')
cmd_mock.assert_called_once_with('kapacitor disable taskname')
