| # Copyright (C) 2010 Research in Motion Ltd. All rights reserved. |
| # |
| # Redistribution and use in source and binary forms, with or without |
| # modification, are permitted provided that the following conditions are |
| # met: |
| # |
| # * Redistributions of source code must retain the above copyright |
| # notice, this list of conditions and the following disclaimer. |
| # * Redistributions in binary form must reproduce the above |
| # copyright notice, this list of conditions and the following disclaimer |
| # in the documentation and/or other materials provided with the |
| # distribution. |
| # * Neither the name of Research in Motion Ltd. nor the names of its |
| # contributors may be used to endorse or promote products derived from |
| # this software without specific prior written permission. |
| # |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| import unittest |
| |
| from webkitpy.common.system.output_capture import OutputCapture |
| from webkitpy.common.system.user import User |
| |
| |
| class UserTest(unittest.TestCase): |
| |
| example_user_response = 'example user response' |
| |
| def test_prompt_repeat(self): |
| self.repeatsRemaining = 2 |
| |
| def mock_raw_input(message): |
| self.repeatsRemaining -= 1 |
| if not self.repeatsRemaining: |
| return UserTest.example_user_response |
| return None |
| self.assertEqual(User.prompt('input', repeat=self.repeatsRemaining, |
| raw_input=mock_raw_input), UserTest.example_user_response) |
| |
| def test_prompt_when_exceeded_repeats(self): |
| self.repeatsRemaining = 2 |
| |
| def mock_raw_input(message): |
| self.repeatsRemaining -= 1 |
| return None |
| self.assertIsNone(User.prompt('input', repeat=self.repeatsRemaining, raw_input=mock_raw_input)) |
| |
| def test_prompt_with_multiple_lists(self): |
| def run_prompt_test(inputs, expected_result, can_choose_multiple=False): |
| def mock_raw_input(message): |
| return inputs.pop(0) |
| output_capture = OutputCapture() |
| actual_result = output_capture.assert_outputs( |
| self, |
| User.prompt_with_multiple_lists, |
| args=['title', ['subtitle1', 'subtitle2'], [['foo', 'bar'], ['foobar', 'barbaz', 'foobaz']]], |
| kwargs={'can_choose_multiple': can_choose_multiple, 'raw_input': mock_raw_input}, |
| expected_stdout='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) |
| |
| run_prompt_test(['1'], 'foo') |
| run_prompt_test(['badinput', '2'], 'bar') |
| run_prompt_test(['3'], 'foobar') |
| run_prompt_test(['4'], 'barbaz') |
| run_prompt_test(['5'], 'foobaz') |
| |
| run_prompt_test(['1,2'], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test(['1-3'], ['foo', 'bar', 'foobar'], can_choose_multiple=True) |
| run_prompt_test(['1-2,3'], ['foo', 'bar', 'foobar'], can_choose_multiple=True) |
| run_prompt_test(['2-1,3'], ['foobar'], can_choose_multiple=True) |
| run_prompt_test([' 1, 2 '], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test(['all'], ['foo', 'bar', 'foobar', 'barbaz', 'foobaz'], can_choose_multiple=True) |
| run_prompt_test([''], ['foo', 'bar', 'foobar', 'barbaz', 'foobaz'], can_choose_multiple=True) |
| run_prompt_test([' '], ['foo', 'bar', 'foobar', 'barbaz', 'foobaz'], can_choose_multiple=True) |
| run_prompt_test(['badinput', 'all'], ['foo', 'bar', 'foobar', 'barbaz', 'foobaz'], can_choose_multiple=True) |
| |
| def test_prompt_with_list(self): |
| def run_prompt_test(inputs, expected_result, can_choose_multiple=False): |
| def mock_raw_input(message): |
| return inputs.pop(0) |
| output_capture = OutputCapture() |
| actual_result = output_capture.assert_outputs( |
| self, |
| User.prompt_with_list, |
| args=['title', ['foo', 'bar']], |
| kwargs={'can_choose_multiple': can_choose_multiple, 'raw_input': mock_raw_input}, |
| expected_stdout='title\n 1. foo\n 2. bar\n') |
| self.assertEqual(actual_result, expected_result) |
| self.assertEqual(len(inputs), 0) |
| |
| run_prompt_test(['1'], 'foo') |
| run_prompt_test(['badinput', '2'], 'bar') |
| |
| run_prompt_test(['1,2'], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test([' 1, 2 '], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test(['all'], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test([''], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test([' '], ['foo', 'bar'], can_choose_multiple=True) |
| run_prompt_test(['badinput', 'all'], ['foo', 'bar'], can_choose_multiple=True) |
| |
| def test_confirm(self): |
| test_cases = ( |
| (('Continue? [Y/n]: ', True), (User.DEFAULT_YES, 'y')), |
| (('Continue? [Y/n]: ', False), (User.DEFAULT_YES, 'n')), |
| (('Continue? [Y/n]: ', True), (User.DEFAULT_YES, '')), |
| (('Continue? [Y/n]: ', False), (User.DEFAULT_YES, 'q')), |
| (('Continue? [y/N]: ', True), (User.DEFAULT_NO, 'y')), |
| (('Continue? [y/N]: ', False), (User.DEFAULT_NO, 'n')), |
| (('Continue? [y/N]: ', False), (User.DEFAULT_NO, '')), |
| (('Continue? [y/N]: ', False), (User.DEFAULT_NO, 'q')), |
| ) |
| for test_case in test_cases: |
| expected, inputs = test_case |
| |
| def mock_raw_input(message): |
| self.assertEqual(expected[0], message) |
| return inputs[1] |
| |
| result = User().confirm(default=inputs[0], |
| raw_input=mock_raw_input) |
| self.assertEqual(expected[1], result) |