Coverage

Author: rogerbinns

apsw/tests/__main__.py

@@ -3068,6 +3068,15 @@ def phcontinue():
         self.db.execute("select 3").get
         self.assertEqual(called, [False, True])
 
+
+        # bad id
+        class bad_eq:
+            def __eq__(self, other):
+                1 / 0
+
+        self.db.set_progress_handler(phabort, id=bad_eq())
+        self.assertRaises(ZeroDivisionError, self.db.set_progress_handler, phabort, id=3)
+
     def testChanges(self):
         "Verify reporting of changes"
         c = self.db.cursor()
@@ -3375,6 +3384,13 @@ def hook(i):
         self.assertRaises(apsw.ConstraintError, self.db.execute, "insert into foo values(99)")
         self.assertEqual(seen, expected | {11})
 
+        # bad id
+        class bad_eq:
+            def __eq__(self, other):
+                1 / 0
+
+        self.assertRaises(ZeroDivisionError, self.db.set_commit_hook, ch, id=bad_eq())
+
     def testRollbackHook(self):
         "Verify rollback hooks"
         c = self.db.cursor()
@@ -3454,6 +3470,14 @@ def hook3(x):
 
         self.db.execute("begin ; insert into foo values(12); rollback")
 
+        # bad id
+        class bad_eq:
+            def __eq__(self, other):
+                1 / 0
+
+        self.db.set_rollback_hook(hook2, id=bad_eq())
+        self.assertRaises(ZeroDivisionError, self.db.set_rollback_hook, hook3, id=3)
+
     def testUpdateHook(self):
         "Verify update hooks"
         c = self.db.cursor()
@@ -6424,9 +6448,13 @@ def hook(myid, update):
             else:
                 raise Exception("unexpected")
 
+        self.db.preupdate_hook(lambda: 1/0)
+
         for id in test_ids:
             self.db.preupdate_hook(functools.partial(hook, id), id=id)
 
+        self.db.preupdate_hook(None)
+
         gc.get_referents(self.db)
 
         self.db.execute("create table foo(x,y,z)")