Add support for Android and iOS policy fetch types to the testserver.

chrome/browser/policy/test/policy_testserver.py

 # Copyright (c) 2012 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.
 
 """A bare-bones test server for testing cloud policy support.
 
 This implements a simple cloud policy test server that can be used to test
 chrome's device management service client. The policy information is read from
 the file named device_management in the server's data directory. It contains
 enforced and recommended policies for the device and user scope, and a list
@@ skipping 346 matching lines @@
       A tuple of HTTP status code and response data to send to the client.
     """
     token_info, error = self.CheckToken()
     if not token_info:
       return error
 
     response = dm.DeviceManagementResponse()
     for request in msg.request:
       fetch_response = response.policy_response.response.add()
       if (request.policy_type in
-             ('google/chrome/user',
-              'google/chromeos/user',
+             ('google/android/user',
+              'google/chrome/extension',
               'google/chromeos/device',
               'google/chromeos/publicaccount',
-              'google/chrome/extension')):
+              'google/chromeos/user',
+              'google/chrome/user',
+              'google/ios/user')):
         if request_type != 'policy':
           fetch_response.error_code = 400
           fetch_response.error_message = 'Invalid request type'
         else:
           self.ProcessCloudPolicy(request, token_info, fetch_response)
       else:
         fetch_response.error_code = 400
         fetch_response.error_message = 'Invalid policy_type'
 
     return (200, response.SerializeToString())
@@ skipping 154 matching lines @@
 
     # Response is only given if the scope is specified in the config file.
     # Normally 'google/chromeos/device', 'google/chromeos/user' and
     # 'google/chromeos/publicaccount' should be accepted.
     policy = self.server.GetPolicies()
     policy_value = ''
     policy_key = msg.policy_type
     if msg.settings_entity_id:
       policy_key += '/' + msg.settings_entity_id
     if msg.policy_type in token_info['allowed_policy_types']:
-      if (msg.policy_type == 'google/chromeos/user' or
-          msg.policy_type == 'google/chrome/user' or
-          msg.policy_type == 'google/chromeos/publicaccount'):
+      if msg.policy_type in ('google/android/user',
+                             'google/chromeos/publicaccount',
+                             'google/chromeos/user',
+                             'google/chrome/user',
+                             'google/ios/user'):
         settings = cp.CloudPolicySettings()
         payload = self.server.ReadPolicyFromDataDir(policy_key, settings)
         if payload is None:
           self.GatherUserPolicySettings(settings, policy.get(policy_key, {}))
           payload = settings.SerializeToString()
       elif dp is not None and msg.policy_type == 'google/chromeos/device':
         settings = dp.ChromeDeviceSettingsProto()
         payload = self.server.ReadPolicyFromDataDir(policy_key, settings)
         if payload is None:
           self.GatherDevicePolicySettings(settings, policy.get(policy_key, {}))
@@ skipping 211 matching lines @@
           'google/chrome/extension'
       ],
       dm.DeviceRegisterRequest.USER: [
           'google/chromeos/user',
           'google/chrome/extension'
       ],
       dm.DeviceRegisterRequest.DEVICE: [
           'google/chromeos/device',
           'google/chromeos/publicaccount'
       ],
+      dm.DeviceRegisterRequest.ANDROID_BROWSER: [
+          'google/android/user'
+      ],
+      dm.DeviceRegisterRequest.IOS_BROWSER: [
+          'google/ios/user'
+      ],
       dm.DeviceRegisterRequest.TT: ['google/chromeos/user',
                                     'google/chrome/user'],
     }
     if machine_id in KIOSK_MACHINE_IDS:
       enrollment_mode = dm.DeviceRegisterResponse.RETAIL
     else:
       enrollment_mode = dm.DeviceRegisterResponse.ENTERPRISE
     self._registered_tokens[dmtoken] = {
       'device_id': device_id,
       'device_token': dmtoken,
@@ skipping 180 matching lines @@
     if (self.options.log_to_console):
       logger.addHandler(logging.StreamHandler())
     if (self.options.log_file):
       logger.addHandler(logging.FileHandler(self.options.log_file))
 
     testserver_base.TestServerRunner.run_server(self)
 
 
 if __name__ == '__main__':
   sys.exit(PolicyServerRunner().main())