def test_prompt_for_ami_noconfig():
'''
Test when called with no config, should return value from text_prompt
'''
instance = Instance()
with mock.patch('__builtin__.raw_input', return_value='foo'):
from touchpaper.prompts import prompt_for_ami
prompt_for_ami(instance)
assert instance.ami == 'foo'
def test_prompt_for_ami_config_freetext():
'''
Test when called with config and string entered, returns string
'''
instance = Instance()
with mock.patch('__builtin__.raw_input', return_value='foo'):
from touchpaper.prompts import prompt_for_ami
config.data = SAMPLE_AMI_CONFIG
prompt_for_ami(instance)
assert instance.ami == 'foo'
def test_prompt_for_ami_config_selection():
'''
Test when called with config and int entered, returns matching ami entry
'''
instance = Instance()
with mock.patch('__builtin__.raw_input', return_value=0):
from touchpaper.prompts import prompt_for_ami
config.data = SAMPLE_AMI_CONFIG
prompt_for_ami(instance)
for k, v in config.data['favourite_amis'].iteritems():
first = k
break
assert instance.ami == k
