Scripts to download files from google storage based on sha1 sums

third_party/gsutil/boto/tests/unit/dynamodb/test_layer2.py

Index: third_party/gsutil/boto/tests/unit/dynamodb/test_layer2.py
diff --git a/third_party/gsutil/boto/tests/unit/dynamodb/test_layer2.py b/third_party/gsutil/boto/tests/unit/dynamodb/test_layer2.py
new file mode 100644
index 0000000000000000000000000000000000000000..ff6ba9ed38df5d8e2fc734617c77cce6f7c3722f
--- /dev/null
+++ b/third_party/gsutil/boto/tests/unit/dynamodb/test_layer2.py
@@ -0,0 +1,119 @@
+#!/usr/bin/env python
+# Copyright (c) 2012 Amazon.com, Inc. or its affiliates.  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.
+#
+
+from tests.unit import unittest
+from mock import Mock
+
+from boto.dynamodb.layer2 import Layer2
+from boto.dynamodb.table import Table, Schema
+
+
+DESCRIBE_TABLE = {
+    "Table": {
+        "CreationDateTime": 1.353526122785E9, "ItemCount":1,
+        "KeySchema": {
+            "HashKeyElement":{"AttributeName": "foo", "AttributeType": "N"}},
+        "ProvisionedThroughput": {
+            "NumberOfDecreasesToday": 0,
+            "ReadCapacityUnits": 5,
+            "WriteCapacityUnits": 5},
+        "TableName": "footest",
+        "TableSizeBytes": 21,
+        "TableStatus": "ACTIVE"}
+}
+
+
+class TestTableConstruction(unittest.TestCase):
+    def setUp(self):
+        self.layer2 = Layer2('access_key', 'secret_key')
+        self.api = Mock()
+        self.layer2.layer1 = self.api
+
+    def test_get_table(self):
+        self.api.describe_table.return_value = DESCRIBE_TABLE
+        table = self.layer2.get_table('footest')
+        self.assertEqual(table.name, 'footest')
+        self.assertEqual(table.create_time, 1353526122.785)
+        self.assertEqual(table.status, 'ACTIVE')
+        self.assertEqual(table.item_count, 1)
+        self.assertEqual(table.size_bytes, 21)
+        self.assertEqual(table.read_units, 5)
+        self.assertEqual(table.write_units, 5)
+        self.assertEqual(table.schema, Schema.create(hash_key=('foo', 'N')))
+
+    def test_create_table_without_api_call(self):
+        table = self.layer2.table_from_schema(
+            name='footest',
+            schema=Schema.create(hash_key=('foo', 'N')))
+        self.assertEqual(table.name, 'footest')
+        self.assertEqual(table.schema, Schema.create(hash_key=('foo', 'N')))
+        # describe_table is never called.
+        self.assertEqual(self.api.describe_table.call_count, 0)
+
+    def test_create_schema_with_hash_and_range(self):
+        schema = self.layer2.create_schema('foo', int, 'bar', str)
+        self.assertEqual(schema.hash_key_name, 'foo')
+        self.assertEqual(schema.hash_key_type, 'N')
+        self.assertEqual(schema.range_key_name, 'bar')
+        self.assertEqual(schema.range_key_type, 'S')
+
+    def test_create_schema_with_hash(self):
+        schema = self.layer2.create_schema('foo', str)
+        self.assertEqual(schema.hash_key_name, 'foo')
+        self.assertEqual(schema.hash_key_type, 'S')
+        self.assertIsNone(schema.range_key_name)
+        self.assertIsNone(schema.range_key_type)
+
+
+class TestSchemaEquality(unittest.TestCase):
+    def test_schema_equal(self):
+        s1 = Schema.create(hash_key=('foo', 'N'))
+        s2 = Schema.create(hash_key=('foo', 'N'))
+        self.assertEqual(s1, s2)
+
+    def test_schema_not_equal(self):
+        s1 = Schema.create(hash_key=('foo', 'N'))
+        s2 = Schema.create(hash_key=('bar', 'N'))
+        s3 = Schema.create(hash_key=('foo', 'S'))
+        self.assertNotEqual(s1, s2)
+        self.assertNotEqual(s1, s3)
+
+    def test_equal_with_hash_and_range(self):
+        s1 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'S'))
+        s2 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'S'))
+        self.assertEqual(s1, s2)
+
+    def test_schema_with_hash_and_range_not_equal(self):
+        s1 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'S'))
+        s2 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'N'))
+        s3 = Schema.create(hash_key=('foo', 'S'), range_key=('baz', 'N'))
+        s4 = Schema.create(hash_key=('bar', 'N'), range_key=('baz', 'N'))
+        self.assertNotEqual(s1, s2)
+        self.assertNotEqual(s1, s3)
+        self.assertNotEqual(s1, s4)
+        self.assertNotEqual(s2, s4)
+        self.assertNotEqual(s3, s4)
+
+
+if __name__ == '__main__':
+    unittest.main()