Scripts to download files from google storage based on sha1 sums

third_party/gsutil/gslib/test_plurality_checkable_iterator.py

+#!/usr/bin/env python
+#
+# Copyright 2012 Google Inc.
+#
+# 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 PluralityCheckableIterator"""
+
+import unittest
+
+from plurality_checkable_iterator import PluralityCheckableIterator
+
+class PluralityCheckableIteratorTests(unittest.TestCase):
+
+  def GetSuiteDescription(self):
+    return 'Unit tests for PluralityCheckableIterator'
+
+  @classmethod
+  def SetUpClass(cls):
+    """Creates class level artifacts useful to multiple tests."""
+    pass
+
+  @classmethod
+  def TearDownClass(cls):
+    """Cleans up any artifacts created by SetUpClass."""
+    pass
+
+  def TestPluralityCheckableIteratorWith0Elems(self):
+    """Tests empty PluralityCheckableIterator."""
+    input_list = range(0)
+    it = iter(input_list)
+    pcit = PluralityCheckableIterator(it)
+    self.assertTrue(pcit.is_empty())
+    self.assertFalse(pcit.has_plurality())
+    output_list = list(pcit)
+    self.assertEqual(input_list, output_list)
+
+  def TestPluralityCheckableIteratorWith1Elem(self):
+    """Tests PluralityCheckableIterator with 1 element."""
+    input_list = range(1)
+    it = iter(input_list)
+    pcit = PluralityCheckableIterator(it)
+    self.assertFalse(pcit.is_empty())
+    self.assertFalse(pcit.has_plurality())
+    output_list = list(pcit)
+    self.assertEqual(input_list, output_list)
+
+  def TestPluralityCheckableIteratorWith2Elems(self):
+    """Tests PluralityCheckableIterator with 2 elements."""
+    input_list = range(2)
+    it = iter(input_list)
+    pcit = PluralityCheckableIterator(it)
+    self.assertFalse(pcit.is_empty())
+    self.assertTrue(pcit.has_plurality())
+    output_list = list(pcit)
+    self.assertEqual(input_list, output_list)
+
+  def TestPluralityCheckableIteratorWith3Elems(self):
+    """Tests PluralityCheckableIterator with 3 elements."""
+    input_list = range(3)
+    it = iter(input_list)
+    pcit = PluralityCheckableIterator(it)
+    self.assertFalse(pcit.is_empty())
+    self.assertTrue(pcit.has_plurality())
+    output_list = list(pcit)
+    self.assertEqual(input_list, output_list)
+
+if __name__ == '__main__':
+  test_loader = unittest.TestLoader()
+  test_loader.testMethodPrefix = 'Test'
+  suite = test_loader.loadTestsFromTestCase(PluralityCheckableIteratorTests)
+  # Seems like there should be a cleaner way to find the test_class.
+  test_class = suite.__getattribute__('_tests')[0]
+  # We call SetUpClass() and TearDownClass() ourselves because we
+  # don't assume the user has Python 2.7 (which supports classmethods
+  # that do it, with camelCase versions of these names).
+  try:
+    print 'Setting up %s...' % test_class.GetSuiteDescription()
+    test_class.SetUpClass()
+    print 'Running %s...' % test_class.GetSuiteDescription()
+    unittest.TextTestRunner(verbosity=2).run(suite)
+  finally:
+    print 'Cleaning up after %s...' % test_class.GetSuiteDescription()
+    test_class.TearDownClass()
+    print ''