[Findit] Add settings for actions after culprits or suspects are identified.

appengine/findit/handlers/test/config_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 datetime
 import json
 import re
 import webapp2
 import webtest
 
@@ skipping 57 matching lines @@
     'iterations_to_rerun': 10
 }
 
 
 _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS = {
     'download_interval_seconds': 10,
     'memcache_master_download_expiration_seconds': 3600,
     'use_chrome_build_extract': True
 }
 
+_MOCK_ACTION_SETTINGS = {
+    'cr_notification_build_threshold': 2,
+    'cr_notification_latency_limit_minutes': 1000,
+}
+
 _MOCK_VERSION_NUMBER = 12
 
 
 class ConfigTest(testing.AppengineTestCase):
   app_module = webapp2.WSGIApplication([
       ('/config', config.Configuration),
   ], debug=True)
 
   def testGetConfigurationSettings(self):
     config_data = {
         'steps_for_masters_rules': _MOCK_STEPS_FOR_MASTERS_RULES,
         'builders_to_trybots': _MOCK_BUILDERS_TO_TRYBOTS,
         'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
         'swarming_settings': _MOCK_SWARMING_SETTINGS,
-        'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS
+        'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS,
+        'action_settings': _MOCK_ACTION_SETTINGS,
     }
 
     self.mock_current_user(user_email='test@chromium.org', is_admin=True)
 
     wf_config.FinditConfig.Get().Update(users.GetCurrentUser(), True,
                                         **config_data)
 
     response = self.test_app.get('/config', params={'format': 'json'})
     self.assertEquals(response.status_int, 200)
 
     expected_response = {
         'masters': _MOCK_STEPS_FOR_MASTERS_RULES,
         'builders': _MOCK_BUILDERS_TO_TRYBOTS,
         'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
         'swarming_settings': _MOCK_SWARMING_SETTINGS,
         'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS,
+        'action_settings': _MOCK_ACTION_SETTINGS,
         'version': 1,
         'latest_version': 1,
         'updated_by': 'test',
         'updated_ts': response.json_body.get('updated_ts')
     }
 
     self.assertEquals(expected_response, response.json_body)
 
   def testGetVersionOfConfigurationSettings(self):
     self.mock_current_user(user_email='test@chromium.org', is_admin=True)
 
     config_data = {
         'steps_for_masters_rules': _MOCK_STEPS_FOR_MASTERS_RULES,
         'builders_to_trybots': _MOCK_BUILDERS_TO_TRYBOTS,
         'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
         'swarming_settings': _MOCK_SWARMING_SETTINGS,
-        'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS
+        'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS,
+        'action_settings': _MOCK_ACTION_SETTINGS,
     }
     wf_config.FinditConfig.Get().Update(users.GetCurrentUser(), True,
                                         **config_data)
 
     response = self.test_app.get(
         '/config', params={'version': 1, 'format': 'json'})
     self.assertEquals(response.status_int, 200)
 
     expected_response = {
         'masters': _MOCK_STEPS_FOR_MASTERS_RULES,
         'builders': _MOCK_BUILDERS_TO_TRYBOTS,
         'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
         'swarming_settings': _MOCK_SWARMING_SETTINGS,
         'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS,
+        'action_settings': _MOCK_ACTION_SETTINGS,
         'version': 1,
         'latest_version': 1,
         'updated_by': 'test',
         'updated_ts': response.json_body.get('updated_ts')
     }
 
     self.assertEquals(expected_response, response.json_body)
 
   def testGetOutOfBoundsVersionOfConfigurationSettings(self):
     config_data = {
@@ skipping 486 matching lines @@
                         'check_global': False
                     }
                 },
                 'global': {
                     'unsupported_steps': ['1']
                 }
             },
             'builders_to_trybots': _MOCK_BUILDERS_TO_TRYBOTS,
             'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
             'swarming_settings': _MOCK_SWARMING_SETTINGS,
-            'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS
+            'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS,
+            'action_settings': _MOCK_ACTION_SETTINGS,
         })
     }
 
     response = self.test_app.post('/config', params=params)
 
     expected_response = {
         'masters': {
             'supported_masters': {
                 'a': {
                 },
                 'b': {
                     'supported_steps': ['1'],
                     'unsupported_steps': ['2', '3', '4'],
                 },
                 'c': {
                     'supported_steps': ['5'],
                     'check_global': False
                 }
             },
             'global': {
                 'unsupported_steps': ['1']
             }
         },
         'builders': _MOCK_BUILDERS_TO_TRYBOTS,
         'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
         'swarming_settings': _MOCK_SWARMING_SETTINGS,
         'download_build_data_settings': _MOCK_DOWNLOAD_BUILD_DATA_SETTINGS,
+        'action_settings': _MOCK_ACTION_SETTINGS,
         'version': 1,
         'latest_version': 1,
         'updated_by': 'test',
         'updated_ts': response.json_body.get('updated_ts')
     }
 
     self.assertEquals(expected_response, response.json_body)
+
+  def testValidateActionSettings(self):
+    self.assertFalse(config._ValidateActionSettings({}))
+    self.assertTrue(config._ValidateActionSettings(
+        {
+            'cr_notification_build_threshold': 2,
+            'cr_notification_latency_limit_minutes': 1000,
+        }))