swarming: Add server-side implementation for supplemental bot_config

appengine/swarming/test_env_handlers.py

 # coding: utf-8
 # Copyright 2014 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
 """Base class for handlers_*_test.py"""
 
 import base64
 import json
 import os
@@ skipping 123 matching lines @@
         '/auth/api/v1/accounts/self/xsrf_token',
         headers={'X-XSRF-Token-Request': '1'}).json
     return resp['xsrf_token'].encode('ascii')
 
   def post_json(self, url, params, **kwargs):
     """Does an HTTP POST with a JSON API and return JSON response."""
     return self.app.post_json(url, params=params, **kwargs).json
 
   # Bot
 
-  def do_handshake(self, bot='bot1'):
+  def do_handshake(self, bot='bot1', status=200):
     """Performs bot handshake, returns data to be sent to bot handlers.
 
     Also populates self.bot_version.
     """
     params = {
       'dimensions': {
         'id': [bot],
         'os': ['Amiga'],
         'pool': ['default'],
       },
       'state': {
         'running_time': 1234.0,
         'sleep_streak': 0,
         'started_ts': 1410990411.111,
       },
       'version': '123',
     }
     response = self.app.post_json(
         '/swarming/api/v1/bot/handshake',
-        params=params).json
+        params=params, status=status)
+    if status != 200:
+      return
+    response = response.json
     self.bot_version = response['bot_version']
     params['version'] = self.bot_version
     params['state']['bot_group_cfg_version'] = response['bot_group_cfg_version']
+    # A bit hackish but fine for unit testing purpose.
+    if response.get('bot_config'):
+      params['bot_config'] = response['bot_config']
     return params
 
   def bot_poll(self, bot='bot1'):
     """Simulates a bot that polls for task."""
     params = self.do_handshake(bot)
     return self.post_json('/swarming/api/v1/bot/poll', params)
 
   def bot_complete_task(self, **kwargs):
     # Emulate an isolated task.
     params = {
@@ skipping 95 matching lines @@
 
   def client_create_task_raw(self, properties=None, **kwargs):
     """Creates a raw command TaskRequest via the Cloud Endpoints API."""
     properties = (properties or {}).copy()
     properties['command'] = ['python', 'run_test.py']
     return self._client_create_task(properties, **kwargs)
 
   def client_get_results(self, task_id):
     return self.endpoint_call(
         handlers_endpoints.SwarmingTaskService, 'result', {'task_id': task_id})