OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 Amazon.com, Inc. or its affiliates. All Rights Reserved |
| 3 # |
| 4 # Permission is hereby granted, free of charge, to any person obtaining a |
| 5 # copy of this software and associated documentation files (the |
| 6 # "Software"), to deal in the Software without restriction, including |
| 7 # without limitation the rights to use, copy, modify, merge, publish, dis- |
| 8 # tribute, sublicense, and/or sell copies of the Software, and to permit |
| 9 # persons to whom the Software is furnished to do so, subject to the fol- |
| 10 # lowing conditions: |
| 11 # |
| 12 # The above copyright notice and this permission notice shall be included |
| 13 # in all copies or substantial portions of the Software. |
| 14 # |
| 15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
| 16 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- |
| 17 # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT |
| 18 # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
| 19 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| 20 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
| 21 # IN THE SOFTWARE. |
| 22 # |
| 23 |
| 24 from tests.unit import unittest |
| 25 from mock import Mock |
| 26 |
| 27 from boto.dynamodb.layer2 import Layer2 |
| 28 from boto.dynamodb.table import Table, Schema |
| 29 |
| 30 |
| 31 DESCRIBE_TABLE = { |
| 32 "Table": { |
| 33 "CreationDateTime": 1.353526122785E9, "ItemCount":1, |
| 34 "KeySchema": { |
| 35 "HashKeyElement":{"AttributeName": "foo", "AttributeType": "N"}}, |
| 36 "ProvisionedThroughput": { |
| 37 "NumberOfDecreasesToday": 0, |
| 38 "ReadCapacityUnits": 5, |
| 39 "WriteCapacityUnits": 5}, |
| 40 "TableName": "footest", |
| 41 "TableSizeBytes": 21, |
| 42 "TableStatus": "ACTIVE"} |
| 43 } |
| 44 |
| 45 |
| 46 class TestTableConstruction(unittest.TestCase): |
| 47 def setUp(self): |
| 48 self.layer2 = Layer2('access_key', 'secret_key') |
| 49 self.api = Mock() |
| 50 self.layer2.layer1 = self.api |
| 51 |
| 52 def test_get_table(self): |
| 53 self.api.describe_table.return_value = DESCRIBE_TABLE |
| 54 table = self.layer2.get_table('footest') |
| 55 self.assertEqual(table.name, 'footest') |
| 56 self.assertEqual(table.create_time, 1353526122.785) |
| 57 self.assertEqual(table.status, 'ACTIVE') |
| 58 self.assertEqual(table.item_count, 1) |
| 59 self.assertEqual(table.size_bytes, 21) |
| 60 self.assertEqual(table.read_units, 5) |
| 61 self.assertEqual(table.write_units, 5) |
| 62 self.assertEqual(table.schema, Schema.create(hash_key=('foo', 'N'))) |
| 63 |
| 64 def test_create_table_without_api_call(self): |
| 65 table = self.layer2.table_from_schema( |
| 66 name='footest', |
| 67 schema=Schema.create(hash_key=('foo', 'N'))) |
| 68 self.assertEqual(table.name, 'footest') |
| 69 self.assertEqual(table.schema, Schema.create(hash_key=('foo', 'N'))) |
| 70 # describe_table is never called. |
| 71 self.assertEqual(self.api.describe_table.call_count, 0) |
| 72 |
| 73 def test_create_schema_with_hash_and_range(self): |
| 74 schema = self.layer2.create_schema('foo', int, 'bar', str) |
| 75 self.assertEqual(schema.hash_key_name, 'foo') |
| 76 self.assertEqual(schema.hash_key_type, 'N') |
| 77 self.assertEqual(schema.range_key_name, 'bar') |
| 78 self.assertEqual(schema.range_key_type, 'S') |
| 79 |
| 80 def test_create_schema_with_hash(self): |
| 81 schema = self.layer2.create_schema('foo', str) |
| 82 self.assertEqual(schema.hash_key_name, 'foo') |
| 83 self.assertEqual(schema.hash_key_type, 'S') |
| 84 self.assertIsNone(schema.range_key_name) |
| 85 self.assertIsNone(schema.range_key_type) |
| 86 |
| 87 |
| 88 class TestSchemaEquality(unittest.TestCase): |
| 89 def test_schema_equal(self): |
| 90 s1 = Schema.create(hash_key=('foo', 'N')) |
| 91 s2 = Schema.create(hash_key=('foo', 'N')) |
| 92 self.assertEqual(s1, s2) |
| 93 |
| 94 def test_schema_not_equal(self): |
| 95 s1 = Schema.create(hash_key=('foo', 'N')) |
| 96 s2 = Schema.create(hash_key=('bar', 'N')) |
| 97 s3 = Schema.create(hash_key=('foo', 'S')) |
| 98 self.assertNotEqual(s1, s2) |
| 99 self.assertNotEqual(s1, s3) |
| 100 |
| 101 def test_equal_with_hash_and_range(self): |
| 102 s1 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'S')) |
| 103 s2 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'S')) |
| 104 self.assertEqual(s1, s2) |
| 105 |
| 106 def test_schema_with_hash_and_range_not_equal(self): |
| 107 s1 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'S')) |
| 108 s2 = Schema.create(hash_key=('foo', 'N'), range_key=('bar', 'N')) |
| 109 s3 = Schema.create(hash_key=('foo', 'S'), range_key=('baz', 'N')) |
| 110 s4 = Schema.create(hash_key=('bar', 'N'), range_key=('baz', 'N')) |
| 111 self.assertNotEqual(s1, s2) |
| 112 self.assertNotEqual(s1, s3) |
| 113 self.assertNotEqual(s1, s4) |
| 114 self.assertNotEqual(s2, s4) |
| 115 self.assertNotEqual(s3, s4) |
| 116 |
| 117 |
| 118 if __name__ == '__main__': |
| 119 unittest.main() |
OLD | NEW |