| Index: third_party/gsutil/boto/tests/integration/sdb/test_connection.py
|
| diff --git a/third_party/gsutil/boto/tests/integration/sdb/test_connection.py b/third_party/gsutil/boto/tests/integration/sdb/test_connection.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..72a26cf82b6da57e3bef7bf1c580b300ed4a90e5
|
| --- /dev/null
|
| +++ b/third_party/gsutil/boto/tests/integration/sdb/test_connection.py
|
| @@ -0,0 +1,119 @@
|
| +# Copyright (c) 2006-2010 Mitch Garnaat http://garnaat.org/
|
| +# Copyright (c) 2010, Eucalyptus Systems, Inc.
|
| +# 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.
|
| +
|
| +"""
|
| +Some unit tests for the SDBConnection
|
| +"""
|
| +
|
| +import unittest
|
| +import time
|
| +from boto.sdb.connection import SDBConnection
|
| +from boto.exception import SDBResponseError
|
| +
|
| +class SDBConnectionTest (unittest.TestCase):
|
| + sdb = True
|
| +
|
| + def test_1_basic(self):
|
| + print '--- running SDBConnection tests ---'
|
| + c = SDBConnection()
|
| + rs = c.get_all_domains()
|
| + num_domains = len(rs)
|
| +
|
| + # try illegal name
|
| + try:
|
| + domain = c.create_domain('bad:domain:name')
|
| + except SDBResponseError:
|
| + pass
|
| +
|
| + # now create one that should work and should be unique (i.e. a new one)
|
| + domain_name = 'test%d' % int(time.time())
|
| + domain = c.create_domain(domain_name)
|
| + rs = c.get_all_domains()
|
| + assert len(rs) == num_domains + 1
|
| +
|
| + # now let's a couple of items and attributes
|
| + item_1 = 'item1'
|
| + same_value = 'same_value'
|
| + attrs_1 = {'name1' : same_value, 'name2' : 'diff_value_1'}
|
| + domain.put_attributes(item_1, attrs_1)
|
| + item_2 = 'item2'
|
| + attrs_2 = {'name1' : same_value, 'name2' : 'diff_value_2'}
|
| + domain.put_attributes(item_2, attrs_2)
|
| +
|
| + # try to get the attributes and see if they match
|
| + item = domain.get_attributes(item_1, consistent_read=True)
|
| + assert len(item.keys()) == len(attrs_1.keys())
|
| + assert item['name1'] == attrs_1['name1']
|
| + assert item['name2'] == attrs_1['name2']
|
| +
|
| + # try a search or two
|
| + query = 'select * from %s where name1="%s"' % (domain_name, same_value)
|
| + rs = domain.select(query, consistent_read=True)
|
| + n = 0
|
| + for item in rs:
|
| + n += 1
|
| + assert n == 2
|
| + query = 'select * from %s where name2="diff_value_2"' % domain_name
|
| + rs = domain.select(query, consistent_read=True)
|
| + n = 0
|
| + for item in rs:
|
| + n += 1
|
| + assert n == 1
|
| +
|
| + # delete all attributes associated with item_1
|
| + stat = domain.delete_attributes(item_1)
|
| + assert stat
|
| +
|
| + # now try a batch put operation on the domain
|
| + item3 = {'name3_1' : 'value3_1',
|
| + 'name3_2' : 'value3_2',
|
| + 'name3_3' : ['value3_3_1', 'value3_3_2']}
|
| +
|
| + item4 = {'name4_1' : 'value4_1',
|
| + 'name4_2' : ['value4_2_1', 'value4_2_2'],
|
| + 'name4_3' : 'value4_3'}
|
| + items = {'item3' : item3, 'item4' : item4}
|
| + domain.batch_put_attributes(items)
|
| +
|
| + item = domain.get_attributes('item3', consistent_read=True)
|
| + assert item['name3_2'] == 'value3_2'
|
| +
|
| + # now try a batch delete operation (variation #1)
|
| + items = {'item3' : item3}
|
| + stat = domain.batch_delete_attributes(items)
|
| +
|
| + item = domain.get_attributes('item3', consistent_read=True)
|
| + assert not item
|
| +
|
| + # now try a batch delete operation (variation #2)
|
| + stat = domain.batch_delete_attributes({'item4' : None})
|
| +
|
| + item = domain.get_attributes('item4', consistent_read=True)
|
| + assert not item
|
| +
|
| + # now delete the domain
|
| + stat = c.delete_domain(domain)
|
| + assert stat
|
| +
|
| + print '--- tests completed ---'
|
| +
|
|
|