def test_can_replace_default_prompt_for_string_input(self):
handler = InputHandler()
handler.start(string_prompt='a')
handler.input_string()
handler.input_string('b')
handler.end()
self.m1.assert_any_call('a')
self.m1.assert_any_call('b')
def test_can_set_several_default_prompts_for_inputs(self):
handler = InputHandler()
handler.start(number_prompt='a',
password_prompt='b',
string_prompt='c')
handler.input_number()
handler.input_password()
handler.input_string()
handler.end()
self.m1.assert_any_call('a')
self.m1.assert_any_call('c')
self.m2.assert_called_once_with('b')
def check(s):
self.m1.reset_mock()
self.m1.return_value = s
handler = InputHandler()
handler.start()
res = handler.input_string()
handler.end()
return res
def test_can_set_nondefault_prompt_for_string_input(self):
handler = InputHandler()
handler.start()
handler.input_string('a')
handler.end()
self.m1.assert_called_once_with('a')