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()