def test_wait_for_process_failed(self):
''' Raise exception if process doesn't exit cleanly. '''
process = Mock()
process.args = MagicMock()
process.returncode = 1
with self.assertRaises(click.ClickException):
agnostic._wait_for(process)
def test_wait_for_process (self):
''' Wait for process to finish. '''
process = Mock()
process.returncode = 0
agnostic._wait_for(process)
self.assertTrue(process.wait.called)
