Update checked in version of gsutil to version 4.6

gslib/tests/test_thread_pool.py

-# Copyright 2011 Google Inc. All Rights Reserved.
-#
-# Permission is hereby granted, free of charge, to any person obtaining a
-# copy of this software and associated documentation files (the
-# "Software"), to deal in the Software without restriction, including
-# without limitation the rights to use, copy, modify, merge, publish, dis-
-# tribute, sublicense, and/or sell copies of the Software, and to permit
-# persons to whom the Software is furnished to do so, subject to the fol-
-# lowing conditions:
-#
-# The above copyright notice and this permission notice shall be included
-# in all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
-# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
-# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
-# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
-# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
-# IN THE SOFTWARE.
-
-"""Unit tests for gsutil thread pool."""
-
-import threading
-
-import gslib.tests.testcase as testcase
-import gslib.thread_pool as thread_pool
-
-
-class GsutilThreadPoolTests(testcase.GsUtilUnitTestCase):
-  """gsutil thread pool test suite."""
-
-  def _TestThreadPool(self, threads):
-    """Tests pool with specified threads from end to end."""
-    pool = thread_pool.ThreadPool(threads)
-
-    self.actual_call_count = 0
-    expected_call_count = 10000
-
-    self.data = xrange(expected_call_count)
-
-    self.actual_result = 0
-    expected_result = sum(self.data)
-
-    stats_lock = threading.Lock()
-
-    def _Dummy(num):
-      stats_lock.acquire()
-      self.actual_call_count += 1
-      self.actual_result += num
-      stats_lock.release()
-
-    for data in xrange(expected_call_count):
-      pool.AddTask(_Dummy, data)
-
-    pool.Shutdown()
-    self.assertEqual(self.actual_call_count, expected_call_count)
-    self.assertEqual(self.actual_result, expected_result)
-
-    for thread in pool.threads:
-      self.assertFalse(thread.is_alive())
-
-  def testSingleThreadPool(self):
-    """Tests thread pool with a single thread."""
-    self._TestThreadPool(1)
-
-  def testThirtyThreadPool(self):
-    """Tests thread pool with 30 threads."""
-    self._TestThreadPool(30)
-
-  def testThreadPoolExceptionHandler(self):
-    """Tests thread pool with exceptions."""
-    self.exception_raised = False
-
-    def _ExceptionHandler(e):
-      """Verify an exception is raised and that it's the correct one."""
-      self.assertTrue(isinstance(e, TypeError))
-      self.assertEqual(e[0], 'gsutil')
-      self.exception_raised = True
-
-    pool = thread_pool.ThreadPool(1, exception_handler=_ExceptionHandler)
-
-    def _Dummy():
-      raise TypeError('gsutil')
-
-    pool.AddTask(_Dummy)
-    pool.Shutdown()
-
-    self.assertTrue(self.exception_raised)