Index: client/tests/auth_server_test.py |
diff --git a/client/tests/auth_server_test.py b/client/tests/auth_server_test.py |
index 2f18f6fe0b348a3a840cd5781659c8f60ff3d5cf..0e09d8dfd8644f69c856d6d2d0fd2622e24250f3 100755 |
--- a/client/tests/auth_server_test.py |
+++ b/client/tests/auth_server_test.py |
@@ -116,34 +116,22 @@ class LocalAuthServerTest(auto_stub.TestCase): |
self.assertEqual([('acc_1', ('A', 'B', 'C'))], calls) |
def test_handles_token_errors(self): |
- fatal = False |
- code = 123 |
+ calls = [] |
def token_gen(_account_id, _scopes): |
- raise auth_server.TokenError(code, 'error message', fatal=fatal) |
+ calls.append(1) |
+ raise auth_server.TokenError(123, 'error message') |
with local_auth_server(token_gen, 'acc_1'): |
self.assertEqual( |
{u'error_code': 123, u'error_message': u'error message'}, |
call_rpc('acc_1', ['B', 'B', 'A', 'C'])) |
+ self.assertEqual(1, len(calls)) |
- # Non-fatal errors aren't cached. |
- code = 456 |
- self.assertEqual( |
- {u'error_code': 456, u'error_message': u'error message'}, |
- call_rpc('acc_1', ['B', 'B', 'A', 'C'])) |
- |
- # Fatal errors are cached. |
- fatal = True |
- code = 789 |
+ # Errors are cached. Same error is returned. |
self.assertEqual( |
- {u'error_code': 789, u'error_message': u'error message'}, |
- call_rpc('acc_1', ['B', 'B', 'A', 'C'])) |
- |
- # Same cached error. |
- code = 111 |
- self.assertEqual( |
- {u'error_code': 789, u'error_message': u'error message'}, |
+ {u'error_code': 123, u'error_message': u'error message'}, |
call_rpc('acc_1', ['B', 'B', 'A', 'C'])) |
+ self.assertEqual(1, len(calls)) |
def test_http_level_errors(self): |
def token_gen(_account_id, _scopes): |