swarming: add support for cipd on the server side

appengine/swarming/handlers_endpoints_test.py

 #!/usr/bin/env python
 # coding=utf-8
 # Copyright 2015 The LUCI Authors. All rights reserved.
 # Use of this source code is governed by the Apache v2.0 license that can be
 # found in the LICENSE file.
 
 import base64
 import datetime
 import json
 import logging
@@ skipping 130 matching lines @@
     self.mock(random, 'getrandbits', lambda _: 0x88)
     now = datetime.datetime(2010, 1, 2, 3, 4, 5)
     self.mock_now(now)
     str_now = unicode(now.strftime(self.DATETIME_NO_MICRO))
     request = swarming_rpcs.NewTaskRequest(
         expiration_secs=30,
         name='job1',
         priority=200,
         properties=swarming_rpcs.TaskProperties(
             command=['rm', '-rf', '/'],
+            packages=[
+              swarming_rpcs.CipdPackage(package_name='rm', version='latest'),
+            ],
             dimensions=[
               swarming_rpcs.StringPair(key='pool', value='default'),
             ],
             env=[
               swarming_rpcs.StringPair(key='PATH', value='/'),
             ],
             execution_timeout_secs=30,
             io_timeout_secs=30),
         tags=['foo:bar'],
         user='joe@localhost',
         pubsub_topic='projects/abc/topics/def',
         pubsub_auth_token='secret that must not be shown',
         pubsub_userdata='userdata')
     expected = {
       u'request': {
         u'authenticated': u'user:user@example.com',
         u'created_ts': str_now,
         u'expiration_secs': u'30',
         u'name': u'job1',
         u'priority': u'200',
         u'properties': {
           u'command': [u'rm', u'-rf', u'/'],
+          u'packages': [{
+            u'package_name': u'rm',
+            u'version': u'latest',
+          }],
           u'dimensions': [
             {u'key': u'pool', u'value': u'default'},
           ],
           u'env': [
             {u'key': u'PATH', u'value': u'/'},
           ],
           u'execution_timeout_secs': u'30',
           u'grace_period_secs': u'30',
           u'idempotent': False,
           u'io_timeout_secs': u'30',
@@ skipping 34 matching lines @@
         expiration_secs=30,
         name='job1',
         priority=200,
         properties=swarming_rpcs.TaskProperties(
             command=['python', 'run_test.py'],
             dimensions=[
               swarming_rpcs.StringPair(key='os', value='Amiga'),
               swarming_rpcs.StringPair(key='pool', value='default'),
             ],
             execution_timeout_secs=3600,
+            idempotent=True,
             io_timeout_secs=1200,
-            idempotent=True),
+            packages=[
+              swarming_rpcs.CipdPackage(
+                  package_name='rm',
+                  version=test_env_handlers.PINNED_PACKAGE_VERSION),
+            ]),
         tags=['foo:bar'],
         user='joe@localhost')
     expected = {
       u'request': {
         u'authenticated': u'user:user@example.com',
         u'created_ts': str_now_30,
         u'expiration_secs': u'30',
         u'name': u'job1',
         u'priority': u'200',
         u'properties': {
           u'command': [u'python', u'run_test.py'],
           u'dimensions': [
             {u'key': u'os', u'value': u'Amiga'},
             {u'key': u'pool', u'value': u'default'},
           ],
           u'execution_timeout_secs': u'3600',
           u'grace_period_secs': u'30',
           u'idempotent': True,
           u'io_timeout_secs': u'1200',
+          u'packages': [{
+            u'package_name': u'rm',
+            u'version': test_env_handlers.PINNED_PACKAGE_VERSION,
+          }],
         },
         u'tags': [
           u'foo:bar',
           u'os:Amiga',
           u'pool:default',
           u'priority:200',
           u'user:joe@localhost',
         ],
         u'user': u'joe@localhost',
       },
@@ skipping 107 matching lines @@
           u'properties': {
             u'command': [u'python', u'run_test.py'],
             u'dimensions': [
               {u'key': u'os', u'value': u'Amiga'},
               {u'key': u'pool', u'value': u'default'},
             ],
             u'execution_timeout_secs': u'3600',
             u'grace_period_secs': u'30',
             u'idempotent': True,
             u'io_timeout_secs': u'1200',
+            u'packages': [{
+              u'package_name': u'rm',
+              u'version': test_env_handlers.PINNED_PACKAGE_VERSION,
+            }],
           },
           u'tags': [
             u'foo:bar',
             u'os:Amiga',
             u'pool:default',
             u'priority:200',
             u'user:joe@localhost',
           ],
           u'user': u'joe@localhost',
         },
         {
           u'authenticated': u'user:user@example.com',
           u'created_ts': str_now,
           u'expiration_secs': u'86400',
           u'name': u'task',
           u'priority': u'10',
           u'properties': {
             u'command': [u'python', u'run_test.py'],
             u'dimensions': [
               {u'key': u'os', u'value': u'Amiga'},
               {u'key': u'pool', u'value': u'default'},
             ],
             u'execution_timeout_secs': u'3600',
             u'grace_period_secs': u'30',
             u'idempotent': True,
             u'io_timeout_secs': u'1200',
+            u'packages': [{
+              u'package_name': u'rm',
+              u'version': test_env_handlers.PINNED_PACKAGE_VERSION,
+            }],
           },
           u'tags': [
             u'commit:post',
             u'os:Amiga',
             u'os:Win',
             u'pool:default',
             u'priority:10',
             u'project:yay',
             u'user:joe@localhost',
           ],
@@ skipping 609 matching lines @@
       u'properties': {
         u'command': [u'python', u'run_test.py'],
         u'dimensions': [
           {u'key': u'os', u'value': u'Amiga'},
           {u'key': u'pool', u'value': u'default'},
         ],
         u'execution_timeout_secs': u'3600',
         u'grace_period_secs': u'30',
         u'idempotent': False,
         u'io_timeout_secs': u'1200',
+        u'packages': [{
+          u'package_name': u'rm',
+          u'version': test_env_handlers.PINNED_PACKAGE_VERSION,
+        }],
       },
       u'tags': [
         u'os:Amiga',
         u'pool:default',
         u'priority:10',
         u'user:joe@localhost',
       ],
       u'user': u'joe@localhost',
     }
     self.assertEqual(expected, response.json)
@@ skipping 295 matching lines @@
     self.assertEqual(expected, response.json)
 
 
 if __name__ == '__main__':
   if '-v' in sys.argv:
     unittest.TestCase.maxDiff = None
     logging.basicConfig(level=logging.DEBUG)
   else:
     logging.basicConfig(level=logging.CRITICAL)
   unittest.main()