1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
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()
|
