Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(71)

Side by Side Diff: third_party/gsutil/boto/tests/unit/dynamodb/test_layer2.py

Issue 12317103: Added gsutil to depot tools (Closed) Base URL: https://chromium.googlesource.com/chromium/tools/depot_tools.git@master
Patch Set: added readme Created 7 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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()
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698