def test_help(capsys):
with pytest.raises(SystemExit):
main(['--help'])
out, err = capsys.readouterr()
assert out.startswith('usage:')
assert '--yes' in out
assert '--skip' in out
def test_abort(caplog):
with mock.patch('piwheels.terminal.yes_no_prompt') as prompt_mock:
prompt_mock.return_value = False
assert main(['foo', '0.1']) == 2
assert find_message(caplog.records, message='User aborted removal')
def test_version(capsys):
with pytest.raises(SystemExit):
main(['--version'])
out, err = capsys.readouterr()
assert out.strip() == __version__
def capture_exc(self, args):
try:
self.exitcode = main(args)
except Exception as e:
self.exception = e