def test_patch_id_from_hash__no_matches(capsys):
rpc = mock.Mock()
rpc.patch_get_by_project_hash.return_value = {}
with pytest.raises(SystemExit):
patches.patch_id_from_hash(rpc, 'foo', '698fa7f')
captured = capsys.readouterr()
assert 'No patch has the hash provided' in captured.err
assert captured.out == ''
def test_patch_id_from_hash__invalid_id(capsys):
rpc = mock.Mock()
rpc.patch_get_by_project_hash.return_value = {'id': 'xyz'}
with pytest.raises(SystemExit):
patches.patch_id_from_hash(rpc, 'foo', '698fa7f')
captured = capsys.readouterr()
assert 'Invalid patch ID obtained from server' in captured.err
assert captured.out == ''
def test_patch_id_from_hash():
rpc = mock.Mock()
rpc.patch_get_by_project_hash.return_value = {'id': '1'}
result = patches.patch_id_from_hash(rpc, 'foo', '698fa7f')
assert result == 1
rpc.patch_get_by_project_hash.assert_called_once_with('foo', '698fa7f')
rpc.patch_get_by_hash.assert_not_called()
def test_patch_id_from_hash__legacy_function():
rpc = mock.Mock()
rpc.patch_get_by_project_hash.side_effect = xmlrpc.xmlrpclib.Fault(1, 'x')
rpc.patch_get_by_hash.return_value = {'id': '1'}
result = patches.patch_id_from_hash(rpc, 'foo', '698fa7f')
assert result == 1
rpc.patch_get_by_project_hash.assert_called_once_with('foo', '698fa7f')
rpc.patch_get_by_hash.assert_called_once_with('698fa7f')
def main():
action_parser = parser.get_parser()
if len(sys.argv) < 2:
action_parser.print_help()
sys.exit(0)
args = action_parser.parse_args()
args = dict(vars(args))
action = args.get('subcmd')
# set defaults
filt = Filter()
commit_str = None
url = DEFAULT_URL
use_hashes = args.get('hash')
archived_str = args.get('a')
state_str = args.get('s')
project_str = args.get('p')
submitter_str = args.get('w')
delegate_str = args.get('d')
format_str = args.get('f')
patch_ids = args.get('id') or []
msgid_str = args.get('m')
commit_str = args.get('c')
# update multiple IDs with a single commit-hash does not make sense
if commit_str and len(patch_ids) > 1 and action == 'update':
sys.stderr.write("Declining update with COMMIT-REF on multiple IDs")
sys.exit(1)
if state_str is None and archived_str is None and action == 'update':
sys.stderr.write(
'Must specify one or more update options (-a or -s)\n')
sys.exit(1)
if args.get('n'):
try:
filt.add('max_count', args.get('n'))
except:
sys.stderr.write("Invalid maximum count '%s'\n" % args.get('n'))
sys.exit(1)
if args.get('N'):
try:
filt.add('max_count', 0 - args.get('N'))
except:
sys.stderr.write("Invalid maximum count '%s'\n" % args.get('N'))
sys.exit(1)
do_signoff = args.get('signoff')
do_three_way = args.get('3way')
# grab settings from config files
config = ConfigParser.ConfigParser()
config.read([CONFIG_FILE])
if not config.has_section('options') and os.path.exists(CONFIG_FILE):
utils.migrate_old_config_file(CONFIG_FILE, config)
sys.exit(1)
if not project_str:
try:
project_str = config.get('options', 'default')
except:
sys.stderr.write('No default project configured in %s\n' %
CONFIG_FILE)
sys.exit(1)
if not config.has_section(project_str):
sys.stderr.write('No section for project %s in %s\n' %
(CONFIG_FILE, project_str))
sys.exit(1)
if not config.has_option(project_str, 'url'):
sys.stderr.write('No URL for project %s in %s\n' %
(CONFIG_FILE, project_str))
sys.exit(1)
if not do_signoff and config.has_option('options', 'signoff'):
do_signoff = config.getboolean('options', 'signoff')
if not do_signoff and config.has_option(project_str, 'signoff'):
do_signoff = config.getboolean(project_str, 'signoff')
if not do_three_way and config.has_option('options', '3way'):
do_three_way = config.getboolean('options', '3way')
if not do_three_way and config.has_option(project_str, '3way'):
do_three_way = config.getboolean(project_str, '3way')
url = config.get(project_str, 'url')
transport = Transport(url)
if action in auth_actions:
if config.has_option(project_str, 'username') and \
config.has_option(project_str, 'password'):
transport.set_credentials(config.get(project_str, 'username'),
config.get(project_str, 'password'))
else:
sys.stderr.write("The %s action requires authentication, but no "
"username or password\nis configured\n" % action)
sys.exit(1)
if project_str:
filt.add("project", project_str)
if state_str:
filt.add("state", state_str)
if archived_str:
filt.add("archived", archived_str == 'yes')
if msgid_str:
filt.add("msgid", msgid_str)
try:
rpc = xmlrpclib.Server(url, transport=transport)
except:
sys.stderr.write("Unable to connect to %s\n" % url)
sys.exit(1)
if use_hashes:
patch_ids = [
patch_id_from_hash(rpc, project_str, x) for x in patch_ids
]
else:
try:
patch_ids = [int(x) for x in patch_ids]
except ValueError:
sys.stderr.write('Patch IDs must be integers')
sys.exit(1)
if action == 'list' or action == 'search':
if args.get('patch_name') is not None:
filt.add("name__icontains", args.get('patch_name'))
action_list(rpc, filt, submitter_str, delegate_str, format_str)
elif action.startswith('project'):
action_projects(rpc)
elif action.startswith('state'):
action_states(rpc)
elif action == 'view':
pager = os.environ.get('PAGER')
if pager:
pager = subprocess.Popen(pager.split(), stdin=subprocess.PIPE)
if pager:
i = list()
for patch_id in patch_ids:
s = rpc.patch_get_mbox(patch_id)
if len(s) > 0:
i.append(s)
if len(i) > 0:
pager.communicate(input="\n".join(i).encode("utf-8"))
pager.stdin.close()
else:
for patch_id in patch_ids:
s = rpc.patch_get_mbox(patch_id)
if len(s) > 0:
print(s)
elif action == 'info':
for patch_id in patch_ids:
action_info(rpc, patch_id)
elif action == 'get':
for patch_id in patch_ids:
action_get(rpc, patch_id)
elif action == 'apply':
for patch_id in patch_ids:
ret = action_apply(rpc, patch_id)
if ret:
sys.stderr.write("Apply failed with exit status %d\n" % ret)
sys.exit(1)
elif action == 'git_am':
cmd = ['git', 'am']
if do_signoff:
cmd.append('-s')
if do_three_way:
cmd.append('-3')
for patch_id in patch_ids:
ret = action_apply(rpc, patch_id, cmd)
if ret:
sys.stderr.write("'git am' failed with exit status %d\n" % ret)
sys.exit(1)
elif action == 'update':
for patch_id in patch_ids:
action_update_patch(rpc,
patch_id,
state=state_str,
archived=archived_str,
commit=commit_str)
elif action == 'check_list':
action_check_list(rpc)
elif action == 'check_info':
check_id = args['check_id']
action_check_info(rpc, check_id)
elif action == 'check_create':
for patch_id in patch_ids:
action_check_create(rpc, patch_id, args['c'], args['s'], args['u'],
args['d'])
else:
sys.stderr.write("Unknown action '%s'\n" % action)
action_parser.print_help()
sys.exit(1)