sandwich: Implement SandwichTaskBuilder

tools/android/loading/sandwich_metrics_unittest.py

 # Copyright 2016 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.
 
 import copy
 import json
 import os
 import shutil
 import subprocess
 import tempfile
 import unittest
 
 import loading_trace
 import page_track
 import sandwich_metrics as puller
+import sandwich_runner
 import request_track
 import tracing
 
 
 _BLINK_CAT = 'blink.user_timing'
 _MEM_CAT = 'disabled-by-default-memory-infra'
 _START='requestStart'
 _LOADS='loadEventStart'
 _LOADE='loadEventEnd'
 _UNLOAD='unloadEventEnd'
@@ skipping 70 matching lines @@
         {'pid': 123, 'ts':  4000, 'cat': _MEM_CAT, 'ph': 'v', 'name': 'foo'},
         {'pid': 123, 'ts':  5000, 'cat': _MEM_CAT, 'ph': 'v', 'name': NAME},
         {'pid': 123, 'ts':  6000, 'cat': _MEM_CAT, 'ph': 'V'},
         {'pid': 672, 'ts':  7000, 'cat': _MEM_CAT, 'ph': 'v', 'name': NAME},
         {'pid': 354, 'ts':  8000, 'cat': _MEM_CAT, 'ph': 'v', 'name': 'foo'},
         {'pid': 123, 'ts':  9000, 'cat': 'whatever1', 'ph': 'v', 'name': NAME},
         {'pid': 123, 'ts': 10000, 'cat': _MEM_CAT, 'ph': 'v', 'name': NAME},
         {'pid': 354, 'ts': 11000, 'cat': 'whatever0', 'ph': 'R'},
         {'pid': 672, 'ts': 12000, 'cat': _MEM_CAT, 'ph': 'v', 'name': NAME}]
 
-    self.assertTrue(_MEM_CAT in puller.CATEGORIES)
+    self.assertTrue(_MEM_CAT in sandwich_runner.CATEGORIES)
 
     bump_events = RunHelper(TRACE_EVENTS, 123)
     self.assertEquals(2, len(bump_events))
     self.assertEquals(5, bump_events[0].start_msec)
     self.assertEquals(10, bump_events[1].start_msec)
 
     bump_events = RunHelper(TRACE_EVENTS, 354)
     self.assertEquals(1, len(bump_events))
     self.assertEquals(1, bump_events[0].start_msec)
 
     bump_events = RunHelper(TRACE_EVENTS, 672)
     self.assertEquals(3, len(bump_events))
     self.assertEquals(3, bump_events[0].start_msec)
     self.assertEquals(7, bump_events[1].start_msec)
     self.assertEquals(12, bump_events[2].start_msec)
 
     with self.assertRaises(ValueError):
       RunHelper(TRACE_EVENTS, 895)
 
   def testGetWebPageTrackedEvents(self):
-    self.assertTrue(_BLINK_CAT in puller.CATEGORIES)
+    self.assertTrue(_BLINK_CAT in sandwich_runner.CATEGORIES)
 
     trace_events = puller._GetWebPageTrackedEvents(TracingTrack([
         {'ph': 'R', 'ts':  0000, 'args': {},             'cat': 'whatever',
             'name': _START},
         {'ph': 'R', 'ts':  1000, 'args': {'frame': '0'}, 'cat': 'whatever',
             'name': _LOADS},
         {'ph': 'R', 'ts':  2000, 'args': {'frame': '0'}, 'cat': 'whatever',
             'name': _LOADE},
         {'ph': 'R', 'ts':  3000, 'args': {},             'cat': _BLINK_CAT,
             'name': _START},
@@ skipping 87 matching lines @@
       point(230, 1.0),
     ]
     with self.assertRaises(ValueError):
       puller.ComputeSpeedIndex(completness_record)
 
   def testCommandLine(self):
     tmp_dir = tempfile.mkdtemp()
     for dirname in ['1', '2', 'whatever']:
       os.mkdir(os.path.join(tmp_dir, dirname))
       LoadingTrace(_MINIMALIST_TRACE_EVENTS).ToJsonFile(
-          os.path.join(tmp_dir, dirname, 'trace.json'))
+          os.path.join(tmp_dir, dirname, sandwich_runner.TRACE_FILENAME))
 
     process = subprocess.Popen(['python', puller.__file__, tmp_dir])
     process.wait()
     shutil.rmtree(tmp_dir)
 
     self.assertEquals(0, process.returncode)
 
 
 if __name__ == '__main__':
   unittest.main()