datastore fixes

1 files changed, 107 insertions, 0 deletions

diff --git a/tests/test-datastore.py b/tests/test-datastore.py
new file mode 100755
index 0000000..f915a60
--- /dev/null
+++ b/tests/test-datastore.py
@@ -0,0 +1,107 @@
+#!/usr/bin/python
+# Copyright (C) 2006, Red Hat, Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+import unittest
+from sugar.datastore import datastore
+import dbus
+
+class NotFoundError(dbus.DBusException): pass
+
+_ds = datastore.get_instance()
+
+class DataStoreTestCase(unittest.TestCase):
+    _TEST_DATA = "adsfkjadsfadskjasdkjf"
+    _TEST_PROPS = {'foo': 1, 'bar': 'baz'}
+    def _create_test_object(self):
+        obj = _ds.create(self._TEST_DATA, self._TEST_PROPS)
+        self.assert_(obj)
+        return obj
+
+    def testObjectCreate(self):
+        obj = self._create_test_object()
+        self.assert_(obj.uid())
+        _ds.delete(obj)
+
+    def testObjectGet(self):
+        # create a new object
+        obj = self._create_test_object()
+        self.assert_(obj.uid())
+        obj2 = _ds.get(obj.uid())
+        self.assert_(obj2)
+        _ds.delete(obj)
+
+    def testObjectDelete(self):
+        obj = self._create_test_object()
+        uid = obj.uid()
+        _ds.delete(obj)
+        try:
+            _ds.get(uid)
+        except dbus.DBusException, e:
+            if str(e).find("NotFoundError:") < 0:
+                self.fail("Expected a NotFoundError")
+        else:
+            self.fail("Expected a NotFoundError.")
+
+    def testObjectFind(self):
+        obj = self._create_test_object()
+        found = _ds.find(self._TEST_PROPS)
+        self.assert_(obj in found)
+        _ds.delete(obj)
+
+    def testObjectGetData(self):
+        obj = self._create_test_object()
+        data = obj.get_data()
+        self.assert_(data == self._TEST_DATA)
+        _ds.delete(obj)
+
+    _OTHER_DATA = "532532532532532;lkjkjkjfsakjfakjfdsakj"
+    def testObjectSetData(self):
+        obj = self._create_test_object()
+        data = obj.get_data()
+        self.assert_(data == self._TEST_DATA)
+        obj.set_data(self._OTHER_DATA)
+        data = obj.get_data()
+        self.assert_(data == self._OTHER_DATA)
+        _ds.delete(obj)
+
+    def testObjectGetProperties(self):
+        obj = self._create_test_object()
+        props = obj.get_properties()
+        for (key, value) in props.items():
+            if key == 'uid':
+                continue
+            self.assert_(key in self._TEST_PROPS)
+            self.assert_(str(self._TEST_PROPS[key]) == str(value))
+        for (key, value) in self._TEST_PROPS.items():
+            self.assert_(key in props)
+            self.assert_(str(props[key]) == str(value))
+        _ds.delete(obj)
+
+def main():
+    dsTestSuite = unittest.TestSuite()
+    dsTestSuite.addTest(DataStoreTestCase('testObjectCreate'))
+    dsTestSuite.addTest(DataStoreTestCase('testObjectGet'))
+    dsTestSuite.addTest(DataStoreTestCase('testObjectDelete'))
+    dsTestSuite.addTest(DataStoreTestCase('testObjectFind'))
+    dsTestSuite.addTest(DataStoreTestCase('testObjectGetData'))
+    dsTestSuite.addTest(DataStoreTestCase('testObjectSetData'))
+    dsTestSuite.addTest(DataStoreTestCase('testObjectGetProperties'))
+    unittest.TextTestRunner(verbosity=2).run(dsTestSuite)
+
+
+if __name__ == "__main__":
+    main()