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)