The Tor Browser / file revision

 # Copyright (c) 2012 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.

 """A module that contains a queue for running sharded tests."""

 import multiprocessing

 class ShardedTestsQueue(object):

   """A queue for managing pending tests across different runners.

   This class should only be used when sharding.

   Attributes:

     num_devices: an integer; the number of attached Android devices.

     tests: a list of tests to be run.

     tests_queue: if sharding, a JoinableQueue object that holds tests from

         |tests|. Otherwise, a list holding tests.

     results_queue: a Queue object to hold TestResults objects.

   """

   _STOP_SENTINEL = 'STOP'  # sentinel value for iter()

   def __init__(self, num_devices, tests):

     self.num_devices = num_devices

     self.tests_queue = multiprocessing.Queue()

     for test in tests:

       self.tests_queue.put(test)

     for _ in xrange(self.num_devices):

       self.tests_queue.put(ShardedTestsQueue._STOP_SENTINEL)

   def __iter__(self):

     """Returns an iterator with the test cases."""

     return iter(self.tests_queue.get, ShardedTestsQueue._STOP_SENTINEL)