예제 #1
0
 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'
     )
예제 #2
0
    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)
예제 #3
0
    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'}])
예제 #4
0
    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)
예제 #5
0
    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'
            }])
예제 #6
0
 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')
예제 #7
0
 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')