diff --git a/setup.py b/setup.py index 13bee08e..1fbf760e 100644 --- a/setup.py +++ b/setup.py @@ -5,6 +5,7 @@ """ import inspect +import logging import os import sys @@ -132,12 +133,8 @@ class PyTest(Command): except Exception: raise RuntimeError('py.test is not installed, run: pip install pytest') - import logging - silence = logging.WARNING - logging.basicConfig(format='%(asctime)s %(levelname)s: %(message)s', level=os.getenv('LOGLEVEL', silence)) - args = ['--verbose', 'tests', '--doctest-modules', MAIN_PACKAGE] +\ - ['-s' if logging.getLogger().getEffectiveLevel() < silence else '--capture=fd'] + ['-s' if logging.getLogger().getEffectiveLevel() < logging.WARNING else '--capture=fd'] if self.cov: args += self.cov @@ -162,6 +159,8 @@ def read(fname): def setup_package(version): + logging.basicConfig(format='%(message)s', level=os.getenv('LOGLEVEL', logging.WARNING)) + # Assemble additional setup commands cmdclass = {'test': PyTest, 'flake8': Flake8} diff --git a/tests/test_cancellable.py b/tests/test_cancellable.py index b555472b..cd078fa6 100644 --- a/tests/test_cancellable.py +++ b/tests/test_cancellable.py @@ -27,8 +27,8 @@ class TestCancellableSubprocess(unittest.TestCase): def test_cancel(self): self.c._process = Mock() self.c._process.is_running.return_value = True - self.c._process.children.side_effect = psutil.Error() - self.c._process.suspend.side_effect = psutil.Error() + self.c._process.children.side_effect = psutil.NoSuchProcess(123) + self.c._process.suspend.side_effect = psutil.AccessDenied() self.c.cancel() self.c._process.is_running.side_effect = [True, False] self.c.cancel() diff --git a/tests/test_postmaster.py b/tests/test_postmaster.py index 4e4d1397..07f09de2 100644 --- a/tests/test_postmaster.py +++ b/tests/test_postmaster.py @@ -73,7 +73,7 @@ class TestPostmasterProcess(unittest.TestCase): # all processes successfully stopped mock_children.return_value = [Mock()] - mock_children.return_value[0].kill.side_effect = psutil.Error + mock_children.return_value[0].kill.side_effect = psutil.NoSuchProcess(123) self.assertTrue(proc.signal_kill()) # postmaster has gone before suspend @@ -81,17 +81,17 @@ class TestPostmasterProcess(unittest.TestCase): self.assertTrue(proc.signal_kill()) # postmaster has gone before we got a list of children - mock_suspend.side_effect = psutil.Error() + mock_suspend.side_effect = psutil.AccessDenied() mock_children.side_effect = psutil.NoSuchProcess(123) self.assertTrue(proc.signal_kill()) # postmaster has gone after we got a list of children - mock_children.side_effect = psutil.Error() + mock_children.side_effect = psutil.AccessDenied() mock_kill.side_effect = psutil.NoSuchProcess(123) self.assertTrue(proc.signal_kill()) # failed to kill postmaster - mock_kill.side_effect = psutil.AccessDenied(123) + mock_kill.side_effect = psutil.AccessDenied() self.assertFalse(proc.signal_kill()) @patch('psutil.Process.__init__', Mock())