[Offline Pages] Adding batch scheduling to test harness.

chrome/browser/android/offline_pages/evaluation/run_offline_page_evaluation_test.py

 #!/usr/bin/env python
 #
 # Copyright (c) 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.
 #
 #
 # This script is used to run OfflinePageSavePageLaterEvaluationTests.
 # The test will try to call SavePageLater on the list provided as the input,
 # and generate results of the background offlining. Then it will pull the
 # results to the output directory.
 #
 # Example Steps:
 # 1. Build chrome_public_test_apk
 # 2. Prepare a list of urls.
 # 3. Run the script (use -d when you have more than one device connected.)
 #   run_offline_page_evaluation_test.py --output-directory
 #   ~/offline_eval_short_output/ --user-requested=true -use-test-scheduler=true
 #   $CHROME_SRC/out/Default ~/offline_eval_urls.txt
 # 4. Check the results in the output directory.
 
 import argparse
 import os
 import shutil
 import subprocess
 import sys
 
 DEFAULT_USER_REQUEST = True
 DEFAULT_USE_TEST_SCHEDULER = False
+# 0 means the batch would be the whole list of urls.
+DEFAULT_BATCH_SIZE = 0
 DEFAULT_VERBOSE = False
 CONFIG_FILENAME = 'test_config'
 CONFIG_TEMPLATE = """\
 IsUserRequested = {is_user_requested}
 UseTestScheduler = {use_test_scheduler}
+ScheduleBatchSize = {schedule_batch_size}
 """
 
 
 def main(args):
   # Setting up the argument parser.
   parser = argparse.ArgumentParser()
   parser.add_argument(
       '--output-directory',
       dest='output_dir',
       help='Directory for output. Default is ~/offline_eval_output/')
@@ skipping 11 matching lines @@
       '--use-test-scheduler',
       dest='use_test_scheduler',
       action='store_true',
       help='Use test scheduler to avoid real scheduling')
   parser.add_argument(
       '--not-use-test-scheduler',
       dest='use_test_scheduler',
       action='store_false',
       help='Use GCMNetworkManager for scheduling. Default option.')
   parser.add_argument(
+      '--batch-size',
+      type=int,
+      dest='schedule_batch_size',
+      help='Number of pages to be queued after previous batch completes.')
+  parser.add_argument(
       '-v',
       '--verbose',
       dest='verbose',
       action='store_true',
       help='Make test runner verbose.')
   parser.add_argument(
       '-d',
       '--device',
       type=str,
       dest='device_id',
@@ skipping 30 matching lines @@
 
   # Create the output directory for results, and have a copy of test config
   # there.
   if not os.path.exists(options.output_dir):
     print 'Creating output directory for results... ' + options.output_dir
     os.makedirs(options.output_dir)
   with open(config_output_path, 'w') as config:
     config.write(
         CONFIG_TEMPLATE.format(
             is_user_requested=options.user_request,
-            use_test_scheduler=options.use_test_scheduler))
+            use_test_scheduler=options.use_test_scheduler,
+            schedule_batch_size=options.schedule_batch_size))
 
   print 'Uploading config file and input file onto the device.'
   subprocess.call(
       get_adb_command(
           ['push', config_output_path, external_dir + '/paquete/test_config']))
   subprocess.call(
       get_adb_command([
           'push', options.test_urls_file,
           '/sdcard/paquete/offline_eval_urls.txt'
       ]))
@@ skipping 23 matching lines @@
   subprocess.call(
       get_adb_command([
           'pull', external_dir + '/paquete/offline_eval_logs.txt',
           options.output_dir
       ]))
   print 'Test finished!'
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv[1:]))