googlecloud 0.0.4 → 0.0.6

data/packages/gsutil/boto/tests/__init__.py

@@ -0,0 +1,20 @@
+# Copyright (c) 2006-2011 Mitch Garnaat http://garnaat.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.

data/packages/gsutil/boto/tests/db/test_lists.py

@@ -0,0 +1,96 @@
+# 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
+from boto.sdb.db.model import Model
+import time
+
+class SimpleListModel(Model):
+    """Test the List Property"""
+    nums = ListProperty(int)
+    strs = ListProperty(str)
+
+class TestLists(object):
+    """Test the List property"""
+
+    def setup_class(cls):
+        """Setup this class"""
+        cls.objs = []
+
+    def teardown_class(cls):
+        """Remove our objects"""
+        for o in cls.objs:
+            try:
+                o.delete()
+            except:
+                pass
+
+    def test_list_order(self):
+        """Testing the order of lists"""
+        t = SimpleListModel()
+        t.nums = [5, 4, 1, 3, 2]
+        t.strs = ["B", "C", "A", "D", "Foo"]
+        t.put()
+        self.objs.append(t)
+        time.sleep(3)
+        t = SimpleListModel.get_by_id(t.id)
+        assert(t.nums == [5, 4, 1, 3, 2])
+        assert(t.strs == ["B", "C", "A", "D", "Foo"])
+
+    def test_old_compat(self):
+        """Testing to make sure the old method of encoding lists will still return results"""
+        t = SimpleListModel()
+        t.put()
+        self.objs.append(t)
+        time.sleep(3)
+        item = t._get_raw_item()
+        item['strs'] = ["A", "B", "C"]
+        item.save()
+        time.sleep(3)
+        t = SimpleListModel.get_by_id(t.id)
+        i1 = sorted(item['strs'])
+        i2 = t.strs
+        i2.sort()
+        assert(i1 == i2)
+
+    def test_query_equals(self):
+        """We noticed a slight problem with querying, since the query uses the same encoder,
+        it was asserting that the value was at the same position in the list, not just "in" the list"""
+        t = SimpleListModel()
+        t.strs = ["Bizzle", "Bar"]
+        t.put()
+        self.objs.append(t)
+        time.sleep(3)
+        assert(SimpleListModel.find(strs="Bizzle").count() == 1)
+        assert(SimpleListModel.find(strs="Bar").count() == 1)
+        assert(SimpleListModel.find(strs=["Bar", "Bizzle"]).count() == 1)
+
+    def test_query_not_equals(self):
+        """Test a not equal filter"""
+        t = SimpleListModel()
+        t.strs = ["Fizzle"]
+        t.put()
+        self.objs.append(t)
+        time.sleep(3)
+        print SimpleListModel.all().filter("strs !=", "Fizzle").get_query()
+        for tt in SimpleListModel.all().filter("strs !=", "Fizzle"):
+            print tt.strs
+            assert("Fizzle" not in tt.strs)

data/packages/gsutil/boto/tests/db/test_password.py

@@ -0,0 +1,128 @@
+# Copyright (c) 2010 Robert Mela
+#
+# 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.
+ 
+
+import unittest
+import logging
+import time
+
+log= logging.getLogger('password_property_test')
+log.setLevel(logging.DEBUG)
+
+class PasswordPropertyTest(unittest.TestCase):
+    """Test the PasswordProperty"""
+
+    def tearDown(self):
+        cls=self.test_model()
+        for obj in cls.all(): obj.delete()
+
+    def hmac_hashfunc(self):
+        import hmac
+        def hashfunc(msg):
+            return hmac.new('mysecret', msg)
+        return hashfunc
+
+    def test_model(self,hashfunc=None):
+        from boto.utils import Password
+        from boto.sdb.db.model import Model
+        from boto.sdb.db.property import PasswordProperty
+        import hashlib
+        class MyModel(Model):
+            password=PasswordProperty(hashfunc=hashfunc)
+        return MyModel
+
+    def test_custom_password_class(self):
+        from boto.utils import Password
+        from boto.sdb.db.model import Model
+        from boto.sdb.db.property import PasswordProperty
+        import hmac, hashlib
+
+
+        myhashfunc = hashlib.md5
+	## Define a new Password class
+        class MyPassword(Password):
+            hashfunc = myhashfunc #hashlib.md5 #lambda cls,msg: hmac.new('mysecret',msg)
+
+	## Define a custom password property using the new Password class
+
+        class MyPasswordProperty(PasswordProperty):
+            data_type=MyPassword
+            type_name=MyPassword.__name__
+
+	## Define a model using the new password property
+
+        class MyModel(Model):
+            password=MyPasswordProperty()#hashfunc=hashlib.md5)
+
+        obj = MyModel()
+        obj.password = 'bar'
+        expected = myhashfunc('bar').hexdigest() #hmac.new('mysecret','bar').hexdigest()
+        log.debug("\npassword=%s\nexpected=%s" % (obj.password, expected))
+        self.assertTrue(obj.password == 'bar' )
+        obj.save()
+        id= obj.id
+        time.sleep(5)
+        obj = MyModel.get_by_id(id)
+        self.assertEquals(obj.password, 'bar')
+        self.assertEquals(str(obj.password), expected)
+                          #hmac.new('mysecret','bar').hexdigest())
+ 
+        
+    def test_aaa_default_password_property(self):
+        cls = self.test_model()
+        obj = cls(id='passwordtest')
+        obj.password = 'foo'
+        self.assertEquals('foo', obj.password)
+        obj.save()
+        time.sleep(5)
+        obj = cls.get_by_id('passwordtest')
+        self.assertEquals('foo', obj.password)
+
+    def test_password_constructor_hashfunc(self):
+        import hmac
+        myhashfunc=lambda msg: hmac.new('mysecret', msg)
+        cls = self.test_model(hashfunc=myhashfunc)
+        obj = cls()
+        obj.password='hello'
+        expected = myhashfunc('hello').hexdigest()
+        self.assertEquals(obj.password, 'hello')
+        self.assertEquals(str(obj.password), expected)
+        obj.save()
+        id = obj.id
+        time.sleep(5)
+        obj = cls.get_by_id(id)
+        log.debug("\npassword=%s" % obj.password)
+        self.assertTrue(obj.password == 'hello')
+
+       
+ 
+if __name__ == '__main__':
+    import sys, os
+    curdir = os.path.dirname( os.path.abspath(__file__) )
+    srcroot = curdir + "/../.."
+    sys.path = [ srcroot ] + sys.path
+    logging.basicConfig()
+    log.setLevel(logging.INFO)
+    suite = unittest.TestLoader().loadTestsFromTestCase(PasswordPropertyTest)
+    unittest.TextTestRunner(verbosity=2).run(suite)
+
+    import boto
+ 

data/packages/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)

data/packages/gsutil/boto/tests/db/test_sequence.py

@@ -0,0 +1,109 @@
+# 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.
+ 
+
+class TestDBHandler(object):
+    """Test the DBHandler"""
+
+    def setup_class(cls):
+        """Setup this class"""
+        cls.sequences = []
+
+    def teardown_class(cls):
+        """Remove our sequences"""
+        for s in cls.sequences:
+            try:
+                s.delete()
+            except:
+                pass
+
+    def test_sequence_generator_no_rollover(self):
+        """Test the sequence generator without rollover"""
+        from boto.sdb.db.sequence import SequenceGenerator
+        gen = SequenceGenerator("ABC")
+        assert(gen("") == "A")
+        assert(gen("A") == "B")
+        assert(gen("B") == "C")
+        assert(gen("C") == "AA")
+        assert(gen("AC") == "BA")
+
+    def test_sequence_generator_with_rollover(self):
+        """Test the sequence generator with rollover"""
+        from boto.sdb.db.sequence import SequenceGenerator
+        gen = SequenceGenerator("ABC", rollover=True)
+        assert(gen("") == "A")
+        assert(gen("A") == "B")
+        assert(gen("B") == "C")
+        assert(gen("C") == "A")
+
+    def test_sequence_simple_int(self):
+        """Test a simple counter sequence"""
+        from boto.sdb.db.sequence import Sequence
+        s = Sequence()
+        self.sequences.append(s)
+        assert(s.val == 0)
+        assert(s.next() == 1)
+        assert(s.next() == 2)
+        s2 = Sequence(s.id)
+        assert(s2.val == 2)
+        assert(s.next() == 3)
+        assert(s.val == 3)
+        assert(s2.val == 3)
+
+    def test_sequence_simple_string(self):
+        from boto.sdb.db.sequence import Sequence, increment_string
+        s = Sequence(fnc=increment_string)
+        self.sequences.append(s)
+        assert(s.val == "A")
+        assert(s.next() == "B")
+
+    def test_fib(self):
+        """Test the fibonacci sequence generator"""
+        from boto.sdb.db.sequence import fib
+        # Just check the first few numbers in the sequence
+        lv = 0
+        for v in [1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144]:
+            assert(fib(v, lv) == lv+v)
+            lv = fib(v, lv)
+
+    def test_sequence_fib(self):
+        """Test the fibonacci sequence"""
+        from boto.sdb.db.sequence import Sequence, fib
+        s = Sequence(fnc=fib)
+        s2 = Sequence(s.id)
+        self.sequences.append(s)
+        assert(s.val == 1)
+        # Just check the first few numbers in the sequence
+        for v in [1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144]:
+            assert(s.next() == v)
+            assert(s.val == v)
+            assert(s2.val == v) # it shouldn't matter which reference we use since it's garunteed to be consistent
+
+    def test_sequence_string(self):
+        """Test the String incrementation sequence"""
+        from boto.sdb.db.sequence import Sequence, increment_string
+        s = Sequence(fnc=increment_string)
+        self.sequences.append(s)
+        assert(s.val == "A")
+        assert(s.next() == "B")
+        s.val = "Z"
+        assert(s.val == "Z")
+        assert(s.next() == "AA")