Patch mock signercli.make_delegation prompts.

This commit is contained in:
dachshund 2013-06-23 13:08:26 +08:00
parent 47abec8cde
commit 42e9dffd24
3 changed files with 10 additions and 4 deletions

View file

@ -141,8 +141,10 @@ def _mock_prompt(msg, junk):
return delegated_targets_dir
elif msg.startswith('\nChoose and enter the parent'):
return parent_role
elif msg.endswith('\nEnter the delegated role\'s name: '):
elif msg.startswith('\nEnter the delegated role\'s name: '):
return delegated_role_name
elif msg.startswith('Recursively walk the given directory? (Y)es/(N)o: '):
return 'N'
else:
error_msg = ('Prompt: '+'\''+msg+'\''+
' did not match any predefined mock prompts.')

View file

@ -1,3 +1,5 @@
#!/usr/bin/env python
"""
<Program Name>
test_extraneous_dependencies_attack.py
@ -179,4 +181,4 @@ def _write_rogue_metadata():
try:
test_extraneous_dependencies_attack()
except ExtraneousDependenciesAttackAlert, error:
print 'error'
print 'error'

View file

@ -549,8 +549,10 @@ def _mock_prompt(msg, junk, targets_path=delegated_targets_path,
return targets_path
elif msg.startswith('\nChoose and enter the parent'):
return parent_role
elif msg.endswith('\nEnter the delegated role\'s name: '):
elif msg.startswith('\nEnter the delegated role\'s name: '):
return new_role_name
elif msg.startswith('Recursively walk the given directory? (Y)es/(N)o: '):
return 'N'
else:
error_msg = ('Prompt: '+'\''+msg+'\''+
' did not match any predefined mock prompts.')
@ -585,4 +587,4 @@ def _mock_get_keyid(junk, keyid=keyid):
signercli._get_keyids = original_get_keyids
signercli._get_password = original_get_password
signercli._prompt = original_prompt
signercli._get_metadata_directory = original_get_metadata_directory
signercli._get_metadata_directory = original_get_metadata_directory