def test_main_no_topdir(mock_cli_parse, mock_is_dir, mock_exit, cli_args): mock_cli_parse.return_value = cli_args exc = '{} is not a directory'.format(cli_args.topdir) with pytest.raises(SystemExit): eos_bootstrap.main() mock_exit.assert_called_with({'ok': False, 'error_type': 'args', 'message': exc})
def test_main(mock_cli_parse, mock_time, mock_isdir, mock_exit, mock_wait, mock_ensure_os, cli_args, device): mock_cli_parse.return_value = cli_args mock_wait.return_value = device with pytest.raises(SystemExit): eos_bootstrap.main() mock_exit.assert_called_with({'ok': True}, dev=device) mock_wait.assert_called_with(countdown=cli_args.reload_delay, poll_delay=10) if device.facts['virtual']: assert not mock_ensure_os.called else: mock_ensure_os.assert_called_with(device)