def test_echo_confirm_default(self, tty_mock):
echo.confirm("message", default="the new default")
self.click_confirm.mock.assert_called_once_with(
"message",
default="the new default",
abort=False,
prompt_suffix=": ",
show_default=True,
err=False,
)
def register_decorator(self, *args, snap_name: str, **kwargs):
try:
return method(self, *args, snap_name=snap_name, **kwargs)
except storeapi.errors.StoreUploadError as upload_error:
if "resource-not-found" not in upload_error.error_list:
raise
echo.wrapped(
"You are required to register this snap before continuing. "
"Refer to 'snapcraft help register' for more options.")
if echo.confirm(
"Would you like to register {!r} with the Snap Store?".
format(snap_name)):
self.register(snap_name=snap_name)
return method(self, *args, snap_name=snap_name, **kwargs)
else:
raise
def prompt_for_value(self, ctx):
default = self.get_default(ctx)
# If this is a prompt for a flag we need to handle this
# differently.
if self.is_bool_flag:
return confirm(self.prompt, default)
return prompt(
self.prompt,
default=default,
type=self.type,
hide_input=self.hide_input,
show_choices=self.show_choices,
confirmation_prompt=self.confirmation_prompt,
value_proc=lambda x: self.process_value(ctx, x),
)
def test_echo_confirm_is_not_tty(self, tty_mock):
echo.confirm("message")
self.click_confirm.mock.assert_not_called()
