def prompt(self, args=None):
"""Return the text to be shown as prompt or handle the prompt and return None.
:param args: optional argument passed from switch_screen calls
:type args: anything
:return: returns an instance of Prompt with text to be shown next to the prompt
for input or None to skip further input processing
:rtype: Prompt instance|None
"""
prompt = Prompt()
prompt.add_refresh_option()
prompt.add_continue_option()
prompt.add_quit_option()
return prompt
def test_quit_option(self):
# quit option
p = Prompt()
p.add_quit_option()
self._check_default_option(p, Prompt.QUIT, Prompt.QUIT_DESCRIPTION)
# test add with description
p = Prompt()
p.add_quit_option("Other quit")
self._check_default_option(p, Prompt.QUIT, "Other quit")
# change existing description
p.add_quit_option("New quit")
self._check_default_option(p, Prompt.QUIT, "New quit")
