def test_install_cancel(self):
"""Verify installation can be cancelled."""
def side_effect(*args): # pylint: disable=W0613
"""First call: return None."""
def second_call(*args): # pylint: disable=W0613
"""Second call: raise KeyboardInterrupt."""
raise KeyboardInterrupt()
subprocess.Popen.poll.side_effect = second_call
return None
with patch('subprocess.Popen.poll', Mock(side_effect=side_effect)):
self.assertFalse(installer.install(['testpackage']))
def test_install(self):
"""Verify install can be called."""
self.assertTrue(installer.install(['testpackage']))
def install(*args, **kwargs):
"""Wrap main installer entry point."""
return installer.install(*args, **kwargs)
