[Findit] Adding support for triaging suspected builds from flake analysis

appengine/findit/common/test/base_handler_test.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
+import mock
 import re
 import urllib
 
 import webapp2
 import webtest
 
+from google.appengine.api import users
+
 from testing_utils import testing
 
 from common import base_handler
 from common.base_handler import BaseHandler
 from common.base_handler import Permission
 
 
 class PermissionLevelHandler(BaseHandler):
   PERMISSION_LEVEL = Permission.ANYONE
 
@@ skipping 50 matching lines @@
     PermissionLevelHandler.PERMISSION_LEVEL = Permission.CORP_USER
 
     # Non-member has no access.
     self._VerifyUnauthorizedAccess('test@gmail.com')
     self._VerifyUnauthorizedAccess('test@chromium.org')
 
     # Corp users and admin has access.
     self._VerifyAuthorizedAccess('test@google.com')
     self._VerifyAuthorizedAccess('test@chromium.org', True)
 
+  @mock.patch.object(users, 'is_current_user_admin', return_value=True)
+  def testShowDebugInfoForAdmin(self, _):
+    self.assertTrue(BaseHandler()._ShowDebugInfo())
+
+  @mock.patch.object(users, 'is_current_user_admin', return_value=False)
+  def testShowDebugInfoForNonAdmin(self, _):
+    handler = BaseHandler()
+    handler.request = {}
+    self.assertFalse(handler._ShowDebugInfo())
+
+  @mock.patch.object(users, 'is_current_user_admin', return_value=False)
+  def testShowDebugInfoWithDebugFlag(self, _):
+    handler = BaseHandler()
+    handler.request = {'debug': '1'}
+    self.assertTrue(handler._ShowDebugInfo())
+
   def testAccessByTaskQueue(self):
     for permission in (Permission.ANYONE, Permission.CORP_USER,
                        Permission.ADMIN):
       PermissionLevelHandler.PERMISSION_LEVEL = permission
       # Simulation of task queue request by setting the header requires admin
       # login.
       self._VerifyAuthorizedAccess(
           'test@chromium.org', True, {'X-AppEngine-QueueName': 'task_queue'})
 
   def testUnknownPermissionLevel(self):
@@ skipping 127 matching lines @@
     response = self.test_app.get('/format?format=json&pretty=1')
     self.assertEquals(200, response.status_int)
     self.assertEquals('application/json', response.content_type)
     expected_body = ('{\n  "a": "b", \n  "z": [\n    1, \n    2, \n    3\n  ], '
                      '\n  "b": "%s"\n}' % ('1' * 200))
     self.assertEquals(response.body, expected_body)
 
   def testToJson(self):
     self.assertEqual('{}', base_handler.ToJson({}))
 
+
 class InternalExceptionHandler(BaseHandler):
   PERMISSION_LEVEL = Permission.ANYONE
 
   def HandleGet(self):
     raise Exception('abc')
 
 
 class InternalExceptionTest(testing.AppengineTestCase):
   app_module = webapp2.WSGIApplication([
       ('/exception', InternalExceptionHandler),
   ], debug=True)
 
   def testInternalException(self):
     self.assertRaisesRegexp(
         webtest.app.AppError,
         re.compile('.*500 Internal Server Error.*An internal error occurred.*',
                    re.MULTILINE | re.DOTALL),
         self.test_app.get, '/exception')