def test_prompt_for_ec2_keyname__no_keys_exist(
self, get_boolean_response_mock, validate_action_mock,
get_key_pairs_mock, generate_and_upload_keypair_mock):
get_key_pairs_mock.return_value = []
get_boolean_response_mock.return_value = True
sshops.prompt_for_ec2_keyname('my-environment')
generate_and_upload_keypair_mock.assert_called_once_with([])
validate_action_mock.assert_called_once()
def test_prompt_for_ec2_keyname__get_boolean_response_to_confirm_termination(
self, get_boolean_response_mock, get_key_pairs_mock,
generate_and_upload_keypair_mock, prompt_for_item_in_list_mock):
get_key_pairs_mock.return_value = mock_responses.DESCRIBE_KEY_PAIRS_RESPONSE[
'KeyPairs']
prompt_for_item_in_list_mock.return_value = 'key_pair_2'
get_boolean_response_mock.return_value = True
sshops.prompt_for_ec2_keyname()
generate_and_upload_keypair_mock.assert_not_called()
def test_prompt_for_ec2_keyname__choose_existing_key(
self, validate_action_mock, get_key_pairs_mock,
generate_and_upload_keypair_mock, prompt_for_item_in_list_mock):
get_key_pairs_mock.return_value = mock_responses.DESCRIBE_KEY_PAIRS_RESPONSE[
'KeyPairs']
prompt_for_item_in_list_mock.return_value = 'key_pair_2'
sshops.prompt_for_ec2_keyname('my-environment')
validate_action_mock.assert_called_once_with(
'To confirm, type the environment name', 'my-environment')
generate_and_upload_keypair_mock.assert_not_called()
def get_keyname(self, default=None):
keyname = self.app.pargs.keyname
if not keyname:
keyname = default
# Get keyname from config file, if exists
if not keyname:
try:
keyname = commonops.get_default_keyname()
except NotInitializedError:
keyname = None
if self.force_non_interactive and not self.interactive:
return keyname
if keyname is None or \
(self.interactive and not self.app.pargs.keyname):
# Prompt for one
keyname = sshops.prompt_for_ec2_keyname()
elif keyname != -1:
commonops.upload_keypair_if_needed(keyname)
return keyname
def get_keyname(self, default=None):
keyname = self.app.pargs.keyname
if not keyname:
keyname = default
# Get keyname from config file, if exists
if not keyname:
try:
keyname = commonops.get_default_keyname()
except NotInitializedError:
keyname = None
if self.force_non_interactive and not self.interactive:
return keyname
if keyname is None or \
(self.interactive and not self.app.pargs.keyname):
# Prompt for one
keyname = sshops.prompt_for_ec2_keyname()
elif keyname != -1:
commonops.upload_keypair_if_needed(keyname)
return keyname
def get_keyname(self, keyname):
# Get keyname from config file, if exists
if not keyname:
try:
keyname = commonops.get_default_keyname()
except NotInitializedError:
keyname = None
if keyname is None:
# Prompt for one
keyname = sshops.prompt_for_ec2_keyname(message=prompts['platforminit.ssh'])
elif keyname != -1:
commonops.upload_keypair_if_needed(keyname)
return keyname
def get_keyname(self, keyname):
# Get keyname from config file, if exists
if not keyname:
try:
keyname = commonops.get_default_keyname()
except NotInitializedError:
keyname = None
if keyname is None:
# Prompt for one
keyname = sshops.prompt_for_ec2_keyname(
message=prompts['platforminit.ssh'])
elif keyname != -1:
commonops.upload_keypair_if_needed(keyname)
return keyname
def get_keyname(keyname, keyname_of_existing_app, interactive,
force_non_interactive):
keyname_passed_through_command_line = not not keyname
keyname = keyname or keyname_of_existing_app
if not keyname:
try:
keyname = commonops.get_default_keyname()
except NotInitializedError:
keyname = None
if force_non_interactive and not interactive:
return keyname
if ((interactive and not keyname_passed_through_command_line)
or (not keyname and not force_non_interactive)):
keyname = sshops.prompt_for_ec2_keyname()
elif keyname != -1:
commonops.upload_keypair_if_needed(keyname)
return keyname
def get_keyname():
return commonops.get_default_keyname() or sshops.prompt_for_ec2_keyname(
message=prompts['platforminit.ssh'])
