[Findit] Cloud Endpoints API to receive reports on flakes.

appengine/findit/test/findit_api_test.py

 # Copyright 2015 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 json
+import mock
+import re
 
 import endpoints
 from google.appengine.api import taskqueue
+import webtest
 
 from testing_utils import testing
 
 from common.waterfall import failure_type
 import findit_api
 from findit_api import FindItApi
 from model.wf_analysis import WfAnalysis
 from model.wf_swarming_task import WfSwarmingTask
 from model.wf_try_job import WfTryJob
 from model import analysis_status
@@ skipping 676 matching lines @@
                 'master_name': master_name,
                 'builder_name': builder_name,
                 'build_number': build_number,
                 'failed_steps': [],
             },
         ]
     }
     self.assertEqual(
         expected_payload_json,
         json.loads(self.taskqueue_requests[0].get('payload')))
+
+  @mock.patch.object(
+      findit_api.flake_analysis_service,
+      'ScheduleAnalysisForFlake', return_value=None)
+  def testUnauthorizedRequestToAnalyzeFlake(self, _mocked_object):
+    self.mock_current_user(user_email='test@chromium.org', is_admin=False)
+
+    flake = {
+        'name': 'suite.test',
+        'is_step': False,
+        'bug_id': 123,
+        'build_steps': [
+            {
+                'master_name': 'm',
+                'builder_name': 'b',
+                'build_number': 456,
+                'step_name': 'name (with patch) on Windows-7-SP1',
+            }
+        ]
+    }
+
+    self.assertRaisesRegexp(
+        webtest.app.AppError,
+        re.compile('.*401 Unauthorized.*',
+                   re.MULTILINE | re.DOTALL),
+        self.call_api, 'AnalyzeFlake', body=flake)
+
+  @mock.patch.object(
+      findit_api.flake_analysis_service,
+      'ScheduleAnalysisForFlake', return_value=True)
+  def testAuthorizedRequestToAnalyzeFlake(self, _mocked_object):
+    self.mock_current_user(user_email='test@chromium.org', is_admin=True)
+
+    flake = {
+        'name': 'suite.test',
+        'is_step': False,
+        'bug_id': 123,
+        'build_steps': [
+            {
+                'master_name': 'm',
+                'builder_name': 'b',
+                'build_number': 456,
+                'step_name': 'name (with patch) on Windows-7-SP1',
+            }
+        ]
+    }
+
+    response = self.call_api('AnalyzeFlake', body=flake)
+    self.assertEqual(200, response.status_int)
+    self.assertTrue(response.json_body.get('analysis_triggered'))