def test_commit(self): self.connection_mock.commit = MagicMock() db = Database('my_db') db.conn = self.connection_mock db.commit() self.connection_mock.commit.assert_called_once()