diff --git a/emmental/db/table.py b/emmental/db/table.py index 866306c..372d8f9 100644 --- a/emmental/db/table.py +++ b/emmental/db/table.py @@ -188,6 +188,7 @@ class Table(Gtk.FilterListModel): def delete(self, row: Row) -> bool: """Delete a Row from the Table.""" if row in self and self.do_sql_delete(row).rowcount == 1: + self.sql.commit() self.store.remove(row) del self.rows[row.primary_key] return True diff --git a/tests/db/test_table.py b/tests/db/test_table.py index 8b2d4e2..be83d52 100644 --- a/tests/db/test_table.py +++ b/tests/db/test_table.py @@ -308,9 +308,12 @@ class TestTableFunctions(tests.util.TestCase): def test_delete(self): """Test deleting rows.""" row = self.table.create(number=1) - self.assertTrue(row.delete()) - self.assertEqual(len(self.table), 0) - self.assertDictEqual(self.table.rows, dict()) + + with unittest.mock.patch.object(self.sql, "commit") as mock_commit: + self.assertTrue(row.delete()) + self.assertEqual(len(self.table), 0) + self.assertDictEqual(self.table.rows, dict()) + mock_commit.assert_called() self.assertFalse(row.delete())