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 |