Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Ensure cache object is closed at end of tests

Browse Source
This commit is contained in:
Kovid Goyal 2022-04-27 19:29:01 +05:30
parent d52bb3646a
commit be83f8bfa6
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/calibre/db/tests/fts_api.py
View File

@ -27,8 +27,10 @@ class FTSAPITest(BaseTest):
super().setUp() super().setUp()
from calibre_extensions.sqlite_extension import set_ui_language from calibre_extensions.sqlite_extension import set_ui_language
set_ui_language('en') set_ui_language('en')
self.libraries_to_close = []
def tearDown(self): def tearDown(self):
[c.close() for c in self.libraries_to_close]
super().tearDown() super().tearDown()
from calibre_extensions.sqlite_extension import set_ui_language from calibre_extensions.sqlite_extension import set_ui_language
set_ui_language('en') set_ui_language('en')
@ -38,7 +40,9 @@ class FTSAPITest(BaseTest):
shutil.rmtree(self.library_path) shutil.rmtree(self.library_path)
os.makedirs(self.library_path) os.makedirs(self.library_path)
self.create_db(self.library_path) self.create_db(self.library_path)
return self.init_cache() ans = self.init_cache()
self.libraries_to_close.append(ans)
return ans
def wait_for_fts_to_finish(self, fts, timeout=10): def wait_for_fts_to_finish(self, fts, timeout=10):
if fts.pool.initialized: if fts.pool.initialized:
@ -134,6 +138,7 @@ class FTSAPITest(BaseTest):
'some other long text that will [also] help with the testing of search'}) 'some other long text that will [also] help with the testing of search'})
self.ae({x['text'] for x in cache.fts_search('also', highlight_start='[', highlight_end=']', snippet_size=3)}, { self.ae({x['text'] for x in cache.fts_search('also', highlight_start='[', highlight_end=']', snippet_size=3)}, {
'…will [also] help…'}) '…will [also] help…'})
cache.close()
def test_fts_triggers(self): def test_fts_triggers(self):
cache = self.init_cache() cache = self.init_cache()