SQLite delitem implementation

darcs-hash:20060201075358-86b55-07095573f5b30ad78829bdc933d7561c8a088294.gz

jabber/storage.py

@@ -276,4 +276,12 @@ class SQLiteStorage(Storage):
         
     def __delitem__(self, pk_tuple):
         Storage.__delitem__(self, pk_tuple)
-        # TODO
+        cursor = self.__connection.cursor()
+        cursor.execute("""
+        delete from account where jid = ? and name = ?
+        """,
+                       (pk_tuple[0],
+                        pk_tuple[1]))
+        self.__connection.commit()
+        cursor.close()
+

tests/test_storage.py

@@ -35,14 +35,6 @@ class Storage_TestCase(unittest.TestCase):
         self.assertEquals(self._storage.spool_dir, spool_dir)
         os.removedirs(spool_dir)
 
-# TODO
-class SQLiteStorage_TestCase(unittest.TestCase):
-    def setUp(self):
-        self._storage = SQLiteStorage()
-
-    def tearDown(self):
-        pass
-
 class DBMStorage_TestCase(unittest.TestCase):
     def setUp(self):
         spool_dir = "./spool/test"
@@ -209,3 +201,14 @@ class SQLiteStorage_TestCase(DBMStorage_TestCase):
                           self._account1)
         self.assertEquals(loaded_storage[("test@localhost", "account2")],
                           self._account2)
+
+    def test_del_sync_get(self):
+        self._storage[("test@localhost", "account1")] = self._account1
+        self._storage[("test@localhost", "account2")] = self._account2
+        del self._storage[("test@localhost", "account2")]
+        loaded_storage = SQLiteStorage(nb_pk_fields = 2, spool_dir = "./spool/test")
+        self.assertEquals(len(loaded_storage.keys()),
+                          1)
+        self.assertEquals(loaded_storage[("test@localhost", "account1")],
+                          self._account1)
+