Scripts to download files from google storage based on sha1 sums

third_party/gsutil/boto/tests/unit/s3/test_keyfile.py

Index: third_party/gsutil/boto/tests/unit/s3/test_keyfile.py
diff --git a/third_party/gsutil/boto/tests/unit/s3/test_keyfile.py b/third_party/gsutil/boto/tests/unit/s3/test_keyfile.py
new file mode 100644
index 0000000000000000000000000000000000000000..bf90664a7e3a713bd44887e6be8d68b95cecc16e
--- /dev/null
+++ b/third_party/gsutil/boto/tests/unit/s3/test_keyfile.py
@@ -0,0 +1,101 @@
+# Copyright 2013 Google Inc.
+# Copyright 2011, Nexenta Systems 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.
+
+import os
+import unittest
+from boto.s3.keyfile import KeyFile
+from tests.integration.s3.mock_storage_service import MockConnection
+from tests.integration.s3.mock_storage_service import MockBucket
+
+
+class KeyfileTest(unittest.TestCase):
+
+    def setUp(self):
+        service_connection = MockConnection()
+        self.contents = '0123456789'
+        bucket = MockBucket(service_connection, 'mybucket')
+        key = bucket.new_key('mykey')
+        key.set_contents_from_string(self.contents)
+        self.keyfile = KeyFile(key)
+
+    def tearDown(self):
+        self.keyfile.close()
+
+    def testReadFull(self):
+        self.assertEqual(self.keyfile.read(len(self.contents)), self.contents)
+
+    def testReadPartial(self):
+        self.assertEqual(self.keyfile.read(5), self.contents[:5])
+        self.assertEqual(self.keyfile.read(5), self.contents[5:])
+
+    def testTell(self):
+        self.assertEqual(self.keyfile.tell(), 0)
+        self.keyfile.read(4)
+        self.assertEqual(self.keyfile.tell(), 4)
+        self.keyfile.read(6)
+        self.assertEqual(self.keyfile.tell(), 10)
+        self.keyfile.close()
+        try:
+            self.keyfile.tell()
+        except ValueError, e:
+            self.assertEqual(str(e), 'I/O operation on closed file')
+
+    def testSeek(self):
+        self.assertEqual(self.keyfile.read(4), self.contents[:4])
+        self.keyfile.seek(0)
+        self.assertEqual(self.keyfile.read(4), self.contents[:4])
+        self.keyfile.seek(5)
+        self.assertEqual(self.keyfile.read(5), self.contents[5:])
+
+        # Seeking negative should raise.
+        try:
+            self.keyfile.seek(-5)
+        except IOError, e:
+            self.assertEqual(str(e), 'Invalid argument')
+
+        # Reading past end of file is supposed to return empty string.
+        self.keyfile.read(10)
+        self.assertEqual(self.keyfile.read(20), '')
+
+        # Seeking past end of file is supposed to silently work.
+        self.keyfile.seek(50)
+        self.assertEqual(self.keyfile.tell(), 50)
+        self.assertEqual(self.keyfile.read(1), '')
+
+    def testSeekEnd(self):
+        self.assertEqual(self.keyfile.read(4), self.contents[:4])
+        self.keyfile.seek(0, os.SEEK_END)
+        self.assertEqual(self.keyfile.read(1), '')
+        self.keyfile.seek(-1, os.SEEK_END)
+        self.assertEqual(self.keyfile.tell(), 9)
+        self.assertEqual(self.keyfile.read(1), '9')
+        # Test attempt to seek backwards past the start from the end.
+        try:
+            self.keyfile.seek(-100, os.SEEK_END)
+        except IOError, e:
+            self.assertEqual(str(e), 'Invalid argument')
+
+    def testSeekCur(self):
+        self.assertEqual(self.keyfile.read(1), self.contents[0])
+        self.keyfile.seek(1, os.SEEK_CUR)
+        self.assertEqual(self.keyfile.tell(), 2)
+        self.assertEqual(self.keyfile.read(4), self.contents[2:6])