[Findit] Moving check flake parameters to config

appengine/findit/waterfall/test/wf_testcase.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 google.appengine.api import users
 
 import copy
 
 from common.findit_testcase import FinditTestCase
 from model.wf_config import FinditConfig
@@ skipping 70 matching lines @@
 
 
 _DEFAULT_SWARMING_SETTINGS = {
     'server_host': 'chromium-swarm.appspot.com',
     'default_request_priority': 150,
     'request_expiration_hours': 20,
     'server_query_interval_seconds': 60,
     'task_timeout_hours': 23,
     'isolated_server': 'https://isolateserver.appspot.com',
     'isolated_storage_url': 'isolateserver.storage.googleapis.com',
-    'iterations_to_rerun': 10
+    'iterations_to_rerun': 10,
+    'get_swarming_task_id_timeout_seconds': 5 * 60,  # 5 minutes.
+    'get_swarming_task_id_wait_seconds': 10,
 }
 
 
 _DEFAULT_DOWNLOAD_BUILD_DATA_SETTINGS = {
     'download_interval_seconds': 10,
     'memcache_master_download_expiration_seconds': 3600,
     'use_chrome_build_extract': True
 }
 
 
 _DEFAULT_ACTION_SETTINGS = {
     'cr_notification_build_threshold': 2,
     'cr_notification_latency_limit_minutes': 30,
 }
 
 
+_DEFAULT_CHECK_FLAKE_SETTINGS = {
+    'lower_flake_threshold': 0.02,
+    'upper_flake_threshold': 0.98,
+    'max_flake_in_a_row': 4,
+    'max_stable_in_a_row': 4,
+    'iterations_to_rerun': 100
+}
+
+
 DEFAULT_CONFIG_DATA = {
     'steps_for_masters_rules': _DEFAULT_STEPS_FOR_MASTERS_RULES,
     'builders_to_trybots': _DEFAULT_TRY_BOT_MAPPING,
     'try_job_settings': _DEFAULT_TRY_JOB_SETTINGS,
     'swarming_settings': _DEFAULT_SWARMING_SETTINGS,
     'download_build_data_settings': _DEFAULT_DOWNLOAD_BUILD_DATA_SETTINGS,
     'action_settings': _DEFAULT_ACTION_SETTINGS,
+    'check_flake_settings': _DEFAULT_CHECK_FLAKE_SETTINGS
 }
 
 
 class WaterfallTestCase(FinditTestCase):  # pragma: no cover.
 
   def UpdateUnitTestConfigSettings(self, config_property=None,
                                    override_data=None):
     """Sets up Findit's config for unit tests.
 
     Args:
       config_property: The name of the config property to update.
       override_data: A dict to override any default settings.
     """
     config_data = DEFAULT_CONFIG_DATA
 
     if config_property and override_data:
       config_data = copy.deepcopy(DEFAULT_CONFIG_DATA)
       config_data[config_property].update(override_data)
 
     FinditConfig.Get().Update(users.User(email='admin@chromium.org'), True,
                               **config_data)
 
   def setUp(self):
     super(WaterfallTestCase, self).setUp()
     self.UpdateUnitTestConfigSettings()