71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
# Copyright 2017, Major Hayden <major@mhtx.net>
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Tests for the process plugin."""
|
|
|
|
import mock
|
|
|
|
from monitorstack.plugins import process
|
|
|
|
import tests.unit # Import the test base module
|
|
|
|
|
|
class TestUptime(object):
|
|
"""Tests for the uptime monitor class."""
|
|
|
|
def test_run_failure(self):
|
|
"""Ensure the run() method works."""
|
|
process_name = 'dont-go-chasing-waterfalls'
|
|
result = tests.runner('process', extra_args=[process_name])
|
|
assert result['variables'] == {process_name: 0}
|
|
assert result['exit_code'] == 1
|
|
|
|
def test_run_success(self):
|
|
"""Ensure the run() method works."""
|
|
process_name = '/'
|
|
result = tests.runner('process', extra_args=[process_name])
|
|
assert result['variables'] == {process_name: 1}
|
|
assert result['exit_code'] == 0
|
|
|
|
def test_check_process_success(self, monkeypatch):
|
|
"""Ensure the check_process() method works."""
|
|
def mock_get_cmdlines():
|
|
return ['process1', 'process2']
|
|
|
|
monkeypatch.setattr(process, 'get_cmdlines', mock_get_cmdlines)
|
|
result = process.check_process('process2')
|
|
assert result
|
|
|
|
def test_get_cmdlines(self):
|
|
"""Ensure the get_cmdlines() method works."""
|
|
assert isinstance(process.get_cmdlines(), list)
|
|
|
|
def test_get_cmdlines_exception(self, monkeypatch):
|
|
"""Ensure the get_cmdlines() method works."""
|
|
class _RaisePid(object):
|
|
pid = 'not-a-pid'
|
|
|
|
@staticmethod
|
|
def cmdline():
|
|
raise process.psutil.NoSuchProcess('not-a-pid')
|
|
|
|
def _mock_process_iter():
|
|
return [_RaisePid, _RaisePid, _RaisePid]
|
|
|
|
with mock.patch('psutil.process_iter') as MockClass:
|
|
MockClass.return_value = _mock_process_iter()
|
|
process_name = 'dont-go-chasing-waterfalls'
|
|
result = tests.runner('process', extra_args=[process_name])
|
|
assert result['variables'] == {process_name: 0}
|
|
assert result['exit_code'] == 1
|