def run_prompt_test(inputs,
expected_result,
can_choose_multiple=False):
def mock_raw_input(message):
return inputs.pop(0)
with OutputCapture() as captured:
actual_result = User.prompt_with_multiple_lists(
'title',
['subtitle1', 'subtitle2'],
[['foo', 'bar'], ['foobar', 'barbaz', 'foobaz']],
can_choose_multiple=can_choose_multiple,
raw_input=mock_raw_input,
)
self.assertEqual(
captured.stdout.getvalue(),
'title\n\nsubtitle1\n 1. foo\n 2. bar\n\nsubtitle2\n 3. foobar\n 4. barbaz\n 5. foobaz\n',
)
self.assertEqual(actual_result, expected_result)
self.assertEqual(len(inputs), 0)
