Add CIPD pin reporting to swarming.

appengine/swarming/server/task_request_test.py

 #!/usr/bin/env python
 # 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.
 
 import datetime
 import logging
 import random
 import sys
 import unittest
@@ skipping 620 matching lines @@
     self.assertEqual(
         datetime.datetime(2012, 1, 2, 3, 4, 5, 123000),
         task_request.request_key_to_datetime(key))
 
   def test_request_id_to_key(self):
     # Simple XOR.
     self.assertEqual(
         ndb.Key(task_request.TaskRequest, 0x7f14acec2fcfffff),
         task_request.request_id_to_key(0xeb5313d0300000))
 
+  def test_packages_grouped_by_path(self):
+    def cp(name, vers, path):
+      return task_request.CipdPackage(
+        package_name=name, version=vers, path=path)
+
+    deadbeef = 'deadbeef' * 5
+    badc0fee = 'badc0fee' * 5
+
+    ci = task_request.CipdInput(packages=[
+      cp('unpinned/${platform}', 'latest', 'a'),
+      cp('partial/windows-amd64', 'latest', 'b'),
+      cp('fully_specified/windows', deadbeef, 'b'),
+    ])
+
+    self.assertEqual(
+      ci.packages_grouped_by_path(None),
+      [
+        (
+          "a",
+          [
+            task_request.PinInfo(
+              pkg=cp('unpinned/${platform}', 'latest', 'a'), pin=None),
+          ]
+        ),
+        (
+          "b",
+          [
+            task_request.PinInfo(
+              pkg=cp('fully_specified/windows', deadbeef, 'b'), pin=None),
+            task_request.PinInfo(
+              pkg=cp('partial/windows-amd64', 'latest', 'b'), pin=None),
+          ]
+        )
+      ]
+    )
+
+    self.assertEqual(
+      ci.packages_grouped_by_path([
+        cp('unpinned/windows', 'instance_id1', 'a'),
+        cp('partial/windows-amd64', badc0fee, 'b'),
+        cp('fully_specified/windows', deadbeef, 'b'),
+      ]),
+      [
+        (
+          "a",
+          [
+            task_request.PinInfo(
+              pkg=cp('unpinned/${platform}', 'latest', 'a'),
+              pin=cp('unpinned/windows', 'instance_id1', 'a')),
+          ]
+        ),
+        (
+          "b",
+          [
+            task_request.PinInfo(
+              pkg=cp('fully_specified/windows', deadbeef, 'b'), pin=None),
+            task_request.PinInfo(
+              pkg=cp('partial/windows-amd64', 'latest', 'b'),
+              pin=cp(None, badc0fee, 'b')),
+          ]
+        )
+      ]
+    )
+
 
 if __name__ == '__main__':
   if '-v' in sys.argv:
     unittest.TestCase.maxDiff = None
   logging.basicConfig(
       level=logging.DEBUG if '-v' in sys.argv else logging.ERROR)
   unittest.main()