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]) |