From d9c0da9ec3eb84a73d78ffdefcc92f1f082c3ced Mon Sep 17 00:00:00 2001 From: Kovid Goyal Date: Sat, 19 Jun 2021 13:13:03 +0530 Subject: [PATCH] ... --- src/calibre/db/sqlite_extension.cpp | 1 + src/calibre/db/tests/fts.py | 10 ++++++++++ 2 files changed, 11 insertions(+) diff --git a/src/calibre/db/sqlite_extension.cpp b/src/calibre/db/sqlite_extension.cpp index daacd3b86a..6b083abdd7 100644 --- a/src/calibre/db/sqlite_extension.cpp +++ b/src/calibre/db/sqlite_extension.cpp @@ -421,6 +421,7 @@ tokenize(PyObject *self, PyObject *args) { if (!remove_diacritics) targs[1] = "0"; Tokenizer t(targs, sizeof(targs)/sizeof(targs[0])); pyobject_raii ans(PyList_New(0)); + if (!ans) return NULL; t.tokenize(ans.ptr(), flags, text, text_length, py_callback); return ans.detach(); } diff --git a/src/calibre/db/tests/fts.py b/src/calibre/db/tests/fts.py index 65687fc647..e1874444c5 100644 --- a/src/calibre/db/tests/fts.py +++ b/src/calibre/db/tests/fts.py @@ -150,3 +150,13 @@ class FTSTest(BaseTest): self.ae(conn.search('moose AND one'), []) # }}} + + +def find_tests(): + import unittest + return unittest.defaultTestLoader.loadTestsFromTestCase(FTSTest) + + +def run_tests(): + from calibre.utils.run_tests import run_tests + run_tests(find_tests)