def test_ask(self):
"""
DialogHelper.ask() behaves properly
"""
dialog = DialogHelper()
dialog.set_input_stream(self.get_input_stream('\n8AM\n'))
self.assertEqual('2PM', dialog.ask(self.get_output_stream(), 'What time is it?', '2PM'))
output = self.get_output_stream()
self.assertEqual('8AM', dialog.ask(output, 'What time is it?', '2PM'))
output.get_stream().seek(0)
self.assertEqual('What time is it?', output.get_stream().read().decode())
def test_ask_with_string(self):
"""
DialogHelper.ask() behaves properly with string
"""
dialog = DialogHelper()
dialog.set_input_stream(self.get_string_input_stream('\n8AM\n'))
self.assertEqual(
'2PM',
dialog.ask(self.get_output_stream(), 'What time is it?', '2PM'))
output = self.get_output_stream()
self.assertEqual('8AM', dialog.ask(output, 'What time is it?', '2PM'))
output.get_stream().seek(0)
self.assertEqual('What time is it?',
output.get_stream().read().decode())
