Add CIPD pin reporting to swarming.

appengine/swarming/server/task_result_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 os
 import random
 import sys
@@ skipping 132 matching lines @@
     self.assertEqual('Deduped', task_result.state_to_string(f))
 
   def test_new_result_summary(self):
     request = mkreq(_gen_request())
     actual = task_result.new_result_summary(request)
     expected = {
       'abandoned_ts': None,
       'bot_dimensions': None,
       'bot_id': None,
       'bot_version': None,
+      'cipd_pins': None,
       'children_task_ids': [],
       'completed_ts': None,
       'costs_usd': [],
       'cost_saved_usd': None,
       'created_ts': self.now,
       'deduped_from': None,
       'duration': None,
       'exit_code': None,
       'failure': False,
       'id': '1d69b9f088008810',
@@ skipping 32 matching lines @@
     request = mkreq(_gen_request())
     actual = task_result.new_run_result(
         request, 1, 'localhost', 'abc',
         {'id': ['localhost'], 'foo': ['bar', 'biz']})
     expected = {
       'abandoned_ts': None,
       'bot_dimensions': {'id': ['localhost'], 'foo': ['bar', 'biz']},
       'bot_id': 'localhost',
       'bot_version': 'abc',
       'children_task_ids': [],
+      'cipd_pins': None,
       'completed_ts': None,
       'cost_usd': 0.,
       'duration': None,
       'exit_code': None,
       'failure': False,
       'id': '1d69b9f088008811',
       'internal_failure': False,
       'modified_ts': None,
       'outputs_ref': None,
       'server_versions': ['v1a'],
@@ skipping 12 matching lines @@
     request = mkreq(_gen_request())
     result_summary = task_result.new_result_summary(request)
     to_run = task_to_run.new_task_to_run(request)
     result_summary.modified_ts = utils.utcnow()
     ndb.transaction(lambda: ndb.put_multi([result_summary, to_run]))
     expected = {
       'abandoned_ts': None,
       'bot_dimensions': None,
       'bot_id': None,
       'bot_version': None,
+      'cipd_pins': None,
       'children_task_ids': [],
       'completed_ts': None,
       'costs_usd': [],
       'cost_saved_usd': None,
       'created_ts': self.now,
       'deduped_from': None,
       'duration': None,
       'exit_code': None,
       'failure': False,
       'id': '1d69b9f088008810',
@@ skipping 27 matching lines @@
     to_run.put()
     run_result = task_result.new_run_result(request, 1, 'localhost', 'abc', {})
     run_result.modified_ts = utils.utcnow()
     result_summary.set_from_run_result(run_result, request)
     ndb.transaction(lambda: ndb.put_multi((result_summary, run_result)))
     expected = {
       'abandoned_ts': None,
       'bot_dimensions': {},
       'bot_id': u'localhost',
       'bot_version': u'abc',
+      'cipd_pins': None,
       'children_task_ids': [],
       'completed_ts': None,
       'costs_usd': [0.],
       'cost_saved_usd': None,
       'created_ts': self.now,
       'deduped_from': None,
       'duration': None,
       'exit_code': None,
       'failure': False,
       'id': '1d69b9f088008810',
@@ skipping 34 matching lines @@
         isolated_upload=task_result.OperationStats(
             duration=0.01, items_cold='foo')).put()
     ndb.transaction(lambda: ndb.put_multi(run_result.append_output('foo', 0)))
     result_summary.set_from_run_result(run_result, request)
     ndb.transaction(lambda: ndb.put_multi((result_summary, run_result)))
     expected = {
       'abandoned_ts': None,
       'bot_dimensions': {},
       'bot_id': u'localhost',
       'bot_version': u'abc',
+      'cipd_pins': None,
       'children_task_ids': [],
       'completed_ts': complete_ts,
       'costs_usd': [0.],
       'cost_saved_usd': None,
       'created_ts': self.now,
       'deduped_from': None,
       'duration': 0.1,
       'exit_code': 0,
       'failure': False,
       'id': '1d69b9f088008810',
@@ skipping 343 matching lines @@
         task_result.TaskOutput.CHUNK_SIZE * '\x00' + 'Baz\x00Bar\x00FooWow',
         self.run_result.get_output())
     self.assertTaskOutputChunk(
         [{'chunk': 'Baz\x00Bar\x00FooWow', 'gaps': [3, 4, 7, 8]}])
 
 
 if __name__ == '__main__':
   if '-v' in sys.argv:
     unittest.TestCase.maxDiff = None
   unittest.main()