def testPrompt(self, raw_input_mock, print_mock):
raw_input_mock.side_effect = ['', '42']
self.assertEqual(
cli_helpers.Prompt(
'What is the ultimate meaning of life, universe and everything?'
), '42')
self.assertEqual(raw_input_mock.call_count, 2)
self.assertEqual(print_mock.call_count, 3)
def _EnsureEditor():
if 'EDITOR' not in os.environ:
os.environ['EDITOR'] = cli_helpers.Prompt(
'Looks like EDITOR environment varible is not defined. Please enter '
'the command to view logs: ')