| Index: third_party/gsutil/boto/tests/db/test_query.py
|
| diff --git a/third_party/gsutil/boto/tests/db/test_query.py b/third_party/gsutil/boto/tests/db/test_query.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..047bf87313b0a09cf5d154fe926e3638b10893b0
|
| --- /dev/null
|
| +++ b/third_party/gsutil/boto/tests/db/test_query.py
|
| @@ -0,0 +1,152 @@
|
| +# Copyright (c) 2010 Chris Moyer http://coredumped.org/
|
| +#
|
| +# 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 boto.sdb.db.property import ListProperty, StringProperty, ReferenceProperty, IntegerProperty
|
| +from boto.sdb.db.model import Model
|
| +import time
|
| +
|
| +class SimpleModel(Model):
|
| + """Simple Test Model"""
|
| + name = StringProperty()
|
| + strs = ListProperty(str)
|
| + num = IntegerProperty()
|
| +
|
| +class SubModel(SimpleModel):
|
| + """Simple Subclassed Model"""
|
| + ref = ReferenceProperty(SimpleModel, collection_name="reverse_ref")
|
| +
|
| +
|
| +class TestQuerying(object):
|
| + """Test different querying capabilities"""
|
| +
|
| + def setup_class(cls):
|
| + """Setup this class"""
|
| + cls.objs = []
|
| +
|
| + o = SimpleModel()
|
| + o.name = "Simple Object"
|
| + o.strs = ["B", "A", "C", "Foo"]
|
| + o.num = 1
|
| + o.put()
|
| + cls.objs.append(o)
|
| +
|
| + o2 = SimpleModel()
|
| + o2.name = "Referenced Object"
|
| + o2.num = 2
|
| + o2.put()
|
| + cls.objs.append(o2)
|
| +
|
| + o3 = SubModel()
|
| + o3.name = "Sub Object"
|
| + o3.num = 3
|
| + o3.ref = o2
|
| + o3.put()
|
| + cls.objs.append(o3)
|
| +
|
| + time.sleep(3)
|
| +
|
| +
|
| +
|
| + def teardown_class(cls):
|
| + """Remove our objects"""
|
| + for o in cls.objs:
|
| + try:
|
| + o.delete()
|
| + except:
|
| + pass
|
| +
|
| + def test_find(self):
|
| + """Test using the "Find" method"""
|
| + assert(SimpleModel.find(name="Simple Object").next().id == self.objs[0].id)
|
| + assert(SimpleModel.find(name="Referenced Object").next().id == self.objs[1].id)
|
| + assert(SimpleModel.find(name="Sub Object").next().id == self.objs[2].id)
|
| +
|
| + def test_like_filter(self):
|
| + """Test a "like" filter"""
|
| + query = SimpleModel.all()
|
| + query.filter("name like", "% Object")
|
| + assert(query.count() == 3)
|
| +
|
| + query = SimpleModel.all()
|
| + query.filter("name not like", "% Object")
|
| + assert(query.count() == 0)
|
| +
|
| + def test_equals_filter(self):
|
| + """Test an "=" and "!=" filter"""
|
| + query = SimpleModel.all()
|
| + query.filter("name =", "Simple Object")
|
| + assert(query.count() == 1)
|
| +
|
| + query = SimpleModel.all()
|
| + query.filter("name !=", "Simple Object")
|
| + assert(query.count() == 2)
|
| +
|
| + def test_or_filter(self):
|
| + """Test a filter function as an "or" """
|
| + query = SimpleModel.all()
|
| + query.filter("name =", ["Simple Object", "Sub Object"])
|
| + assert(query.count() == 2)
|
| +
|
| + def test_and_filter(self):
|
| + """Test Multiple filters which are an "and" """
|
| + query = SimpleModel.all()
|
| + query.filter("name like", "% Object")
|
| + query.filter("name like", "Simple %")
|
| + assert(query.count() == 1)
|
| +
|
| + def test_none_filter(self):
|
| + """Test filtering for a value that's not set"""
|
| + query = SimpleModel.all()
|
| + query.filter("ref =", None)
|
| + assert(query.count() == 2)
|
| +
|
| + def test_greater_filter(self):
|
| + """Test filtering Using >, >="""
|
| + query = SimpleModel.all()
|
| + query.filter("num >", 1)
|
| + assert(query.count() == 2)
|
| +
|
| + query = SimpleModel.all()
|
| + query.filter("num >=", 1)
|
| + assert(query.count() == 3)
|
| +
|
| + def test_less_filter(self):
|
| + """Test filtering Using <, <="""
|
| + query = SimpleModel.all()
|
| + query.filter("num <", 3)
|
| + assert(query.count() == 2)
|
| +
|
| + query = SimpleModel.all()
|
| + query.filter("num <=", 3)
|
| + assert(query.count() == 3)
|
| +
|
| + def test_query_on_list(self):
|
| + """Test querying on a list"""
|
| + assert(SimpleModel.find(strs="A").next().id == self.objs[0].id)
|
| + assert(SimpleModel.find(strs="B").next().id == self.objs[0].id)
|
| + assert(SimpleModel.find(strs="C").next().id == self.objs[0].id)
|
| +
|
| + def test_like(self):
|
| + """Test with a "like" expression"""
|
| + query = SimpleModel.all()
|
| + query.filter("strs like", "%oo%")
|
| + print query.get_query()
|
| + assert(query.count() == 1)
|
|
|