[Findit] Adding metadata for manually vs automatically triggered flake analyses

appengine/findit/waterfall/flake/test/flake_analysis_service_test.py

 # Copyright 2016 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.
 
 from datetime import datetime
 import mock
 
 from common import constants
 from model.flake.flake_analysis_request import BuildStep
 from model.flake.flake_analysis_request import FlakeAnalysisRequest
 from waterfall.flake import flake_analysis_service
+from waterfall.flake import triggering_sources
 from waterfall.test import wf_testcase
 
 
 class FlakeAnalysisServiceTest(wf_testcase.WaterfallTestCase):
 
   def testCheckFlakeSwarmedAndSupportedWhenNotSupported(self):
     request = FlakeAnalysisRequest.Create('flake', False, 123)
     step1 = BuildStep.Create('m', 'b1', 10, 's', datetime(2016, 10, 01))
     step1.swarmed = False
     step1.supported = False
     step2 = BuildStep.Create('m', 'b2', 10, 's', datetime(2016, 10, 01))
     step2.swarmed = False
     step2.supported = False
     request.build_steps = [step1, step2]
 
     self.assertEqual(
         (False, False, None),
         flake_analysis_service._CheckFlakeSwarmedAndSupported(request))
 
-
   def testNeedNewAnalysisWhenNoPreviousOneAndNotStepLevelFlake(self):
     request = FlakeAnalysisRequest.Create('flake', False, 123)
     step1 = BuildStep.Create('m', 'b1', 10, 's', datetime(2016, 10, 01))
     step1.swarmed = False
     step1.supported = False
     step2 = BuildStep.Create('m', 'b2', 10, 's', datetime(2016, 10, 01))
     step2.swarmed = True
     step2.supported = True
     request.build_steps = [step1, step2]
 
@@ skipping 144 matching lines @@
     self.assertEqual(1, version)
     self.assertIsNotNone(step)
     self.assertEqual(80, step.build_number)
 
   def testUnauthorizedAccess(self):
     request = FlakeAnalysisRequest.Create('flake', False, 123)
     step = BuildStep.Create('m', 'b2', 80, 's', datetime(2016, 10, 20))
     request.build_steps = [step]
 
     self.assertIsNone(flake_analysis_service.ScheduleAnalysisForFlake(
-        request, 'test@chromium.org', False))
+        request, 'test@chromium.org', False, triggering_sources.FINDIT_UI))
 
   @mock.patch.object(
-      flake_analysis_service, '_CheckForNewAnalysis', return_value= (0, None))
+      flake_analysis_service, '_CheckForNewAnalysis', return_value=(0, None))
   @mock.patch.object(
       flake_analysis_service.step_mapper, 'FindMatchingWaterfallStep')
   def testAuthorizedAccessButNoNewAnalysisNeeded(self, _mock1, _mock2):
     request = FlakeAnalysisRequest.Create('flake', False, 123)
     step = BuildStep.Create('m', 'b2', 80, 's', datetime(2016, 10, 20))
     request.build_steps = [step]
 
     self.assertFalse(flake_analysis_service.ScheduleAnalysisForFlake(
-        request, 'test@chromium.org', True))
+        request, 'test@chromium.org', True, triggering_sources.FINDIT_UI))
 
   @mock.patch.object(
       flake_analysis_service.step_mapper, 'FindMatchingWaterfallStep')
   def testAuthorizedAccessAndNewAnalysisNeededAndTriggered(self, _mock):
     step = BuildStep.Create('m', 'b', 80, 's', datetime(2016, 10, 20))
     request = FlakeAnalysisRequest.Create('flake', False, 123)
     request.build_steps = [step]
+    user_email = 'test@chromium.org'
+    triggering_source = triggering_sources.FINDIT_UI
 
     def CheckForNewAnalysis(*_):
       step.wf_master_name = 'wf_m'
       step.wf_builder_name = 'wf_b'
       step.wf_build_number = 100
       step.wf_step_name = 'wf_s'
       return 1, step
 
     mocked_analysis = mock.Mock(key='key')
     mocked_request = mock.Mock()
 
     with mock.patch.object(
         flake_analysis_service, '_CheckForNewAnalysis',
         side_effect=CheckForNewAnalysis) as (
             mocked_CheckForNewAnalysis), mock.patch.object(
-          flake_analysis_service.initialize_flake_pipeline,
-          'ScheduleAnalysisIfNeeded', return_value=mocked_analysis) as (
-              mocked_ScheduleAnalysisIfNeeded), mock.patch.object(
-            flake_analysis_service.FlakeAnalysisRequest,
-            'GetVersion', return_value=mocked_request) as mocked_GetVersion:
+                flake_analysis_service.initialize_flake_pipeline,
+                'ScheduleAnalysisIfNeeded', return_value=mocked_analysis) as (
+                    mocked_ScheduleAnalysisIfNeeded), mock.patch.object(
+                        flake_analysis_service.FlakeAnalysisRequest,
+                        'GetVersion', return_value=mocked_request) as (
+                            mocked_GetVersion):
       self.assertTrue(flake_analysis_service.ScheduleAnalysisForFlake(
-          request, 'test@chromium.org', True))
+          request, user_email, True, triggering_source))
       mocked_CheckForNewAnalysis.assert_called_once_with(request)
       mocked_ScheduleAnalysisIfNeeded.assert_called_once_with(
           'wf_m', 'wf_b', 100, 'wf_s', 'flake',
           allow_new_analysis=True, manually_triggered=False,
+          user_email=user_email, triggering_source=triggering_source,
           queue_name=constants.WATERFALL_ANALYSIS_QUEUE)
       mocked_GetVersion.assert_called_once_with(key='flake', version=1)
       mocked_request.assert_has_calls([
           mock.call.analyses.append('key'),
           mock.call.put(),
       ])
 
   @mock.patch.object(
       flake_analysis_service.step_mapper, 'FindMatchingWaterfallStep')
   def testAuthorizedAccessAndNewAnalysisNeededButNotTriggered(self, _mock):
     step = BuildStep.Create('m', 'b', 80, 's', datetime(2016, 10, 20))
     request = FlakeAnalysisRequest.Create('flake', False, 123)
     request.build_steps = [step]
+    user_email = 'test@chromium.org'
+    triggering_source = triggering_sources.FINDIT_UI
 
     def CheckForNewAnalysis(*_):
       step.wf_master_name = 'wf_m'
       step.wf_builder_name = 'wf_b'
       step.wf_build_number = 100
       step.wf_step_name = 'wf_s'
       return 1, step
 
     with mock.patch.object(
         flake_analysis_service, '_CheckForNewAnalysis',
         side_effect=CheckForNewAnalysis) as (
             mocked_CheckForNewAnalysis), mock.patch.object(
-          flake_analysis_service.initialize_flake_pipeline,
-          'ScheduleAnalysisIfNeeded', return_value=None) as (
-              mocked_ScheduleAnalysisIfNeeded), mock.patch.object(
-            flake_analysis_service.FlakeAnalysisRequest,
-            'GetVersion', return_value=None) as mocked_GetVersion:
+                flake_analysis_service.initialize_flake_pipeline,
+                'ScheduleAnalysisIfNeeded', return_value=None) as (
+                    mocked_ScheduleAnalysisIfNeeded), mock.patch.object(
+                        flake_analysis_service.FlakeAnalysisRequest,
+                        'GetVersion', return_value=None) as mocked_GetVersion:
       self.assertFalse(flake_analysis_service.ScheduleAnalysisForFlake(
-          request, 'test@chromium.org', True))
+          request, user_email, True, triggering_sources.FINDIT_UI))
       mocked_CheckForNewAnalysis.assert_called_once_with(request)
       mocked_ScheduleAnalysisIfNeeded.assert_called_once_with(
           'wf_m', 'wf_b', 100, 'wf_s', 'flake',
           allow_new_analysis=True, manually_triggered=False,
+          user_email=user_email, triggering_source=triggering_source,
           queue_name=constants.WATERFALL_ANALYSIS_QUEUE)
       mocked_GetVersion.assert_not_called()