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')