Added affiliation IDs for the new affiliation determination.

chrome/browser/policy/test/policy_testserver.py

Index: chrome/browser/policy/test/policy_testserver.py
diff --git a/chrome/browser/policy/test/policy_testserver.py b/chrome/browser/policy/test/policy_testserver.py
index 09f3c4c62d00e99405ccd20573f47f5efd235b33..480124cf9414b512973030fcf899a6ca1242ce67 100644
--- a/chrome/browser/policy/test/policy_testserver.py
+++ b/chrome/browser/policy/test/policy_testserver.py
@@ -812,6 +812,16 @@ class PolicyRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
       # username from the configuration file or use a default.
       policy_data.username = policy.get('policy_user', 'user@example.com')
     policy_data.device_id = token_info['device_id']
+
+    # Set affiliation IDs so that user was managed on the device.
+    device_affiliation_ids = policy.get('device_affiliation_ids')
+    if device_affiliation_ids:
+      policy_data.device_affiliation_ids.extend(device_affiliation_ids)
+
+    user_affiliation_ids = policy.get('user_affiliation_ids')
+    if user_affiliation_ids:
+      policy_data.user_affiliation_ids.extend(user_affiliation_ids)
+
     signed_data = policy_data.SerializeToString()
 
     response.policy_data = signed_data