| Index: third_party/WebKit/Tools/Scripts/webkitpy/common/system/user_unittest.py
|
| diff --git a/third_party/WebKit/Tools/Scripts/webkitpy/common/system/user_unittest.py b/third_party/WebKit/Tools/Scripts/webkitpy/common/system/user_unittest.py
|
| index 087f6d81b9365bc71d4ceb42c91bf09e11f7003e..915fc9adec6809a0d2a1198117889fa68ce213a2 100644
|
| --- a/third_party/WebKit/Tools/Scripts/webkitpy/common/system/user_unittest.py
|
| +++ b/third_party/WebKit/Tools/Scripts/webkitpy/common/system/user_unittest.py
|
| @@ -31,21 +31,25 @@ import unittest
|
| from webkitpy.common.system.outputcapture 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)
|
| + 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
|
|
|