[Findit] Adding central config test class for unit tests

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
 
 from handlers import config
 from model import wf_config
 from testing_utils import testing
-
+from google.appengine.api import users
 
 _MOCK_STEPS_FOR_MASTERS_RULES_OLD_FORMAT = {
     'master1': ['unsupported_step1', 'unsupported_step2'],
     'master2': ['unsupported_step3', 'unsupported_step4'],
 }
 
 _MOCK_STEPS_FOR_MASTERS_RULES = {
     'supported_masters': {
         'master1': {
             # supported_steps override global.
@@ skipping 45 matching lines @@
 
 _MOCK_VERSION_NUMBER = 12
 
 
 class ConfigTest(testing.AppengineTestCase):
   app_module = webapp2.WSGIApplication([
       ('/config', config.Configuration),
   ], debug=True)
 
   def testGetConfigurationSettings(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
     }
-    wf_config.FinditConfig.Get().Update(**config_data)
+
+    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,
         '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
     }
-    wf_config.FinditConfig.Get().Update(**config_data)
+    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,
         '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):
-    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
     }
-    wf_config.FinditConfig.Get().Update(**config_data)
+    self.mock_current_user(user_email='test@chromium.org', is_admin=True)
+
+    wf_config.FinditConfig.Get().Update(users.GetCurrentUser(), True,
+                                        **config_data)
 
     self.assertRaisesRegexp(
         webtest.app.AppError,
         re.compile('The requested version is invalid or not found.',
                    re.MULTILINE | re.DOTALL),
         self.test_app.get, '/config', params={'version': 0, 'format': 'json'})
     self.assertRaisesRegexp(
         webtest.app.AppError,
         re.compile('The requested version is invalid or not found.',
                    re.MULTILINE | re.DOTALL),
@@ skipping 459 matching lines @@
         'builders': _MOCK_BUILDERS_TO_TRYBOTS,
         'try_job_settings': _MOCK_TRY_JOB_SETTINGS,
         'swarming_settings': _MOCK_SWARMING_SETTINGS,
         'version': 1,
         'latest_version': 1,
         'updated_by': 'test',
         'updated_ts': response.json_body.get('updated_ts')
     }
 
     self.assertEquals(expected_response, response.json_body)