def test_interactive_menu_with_failures(self): with mock.patch('custom.prompt', side_effect=['a', 'do', '2']): choices = ['a', 'b', 'c'] result = select_interactive_menu(choices) self.assertEquals('c', result)
def test_interactive_menu(self): with mock.patch('custom.prompt', return_value='1'): choices = ['a', 'b', 'c'] result = select_interactive_menu(choices) self.assertEquals('b', result)