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

Fix unclosed resource warnings in fetch backend tests

Browse Source
This commit is contained in:
Kovid Goyal 2024-08-26 16:01:53 +05:30
parent d8d63646f4
commit 60555c1389
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/calibre/scraper/qt.py
View File

@ -102,6 +102,12 @@ class FakeResponse:
def close(self): def close(self):
self._data.close() self._data.close()
def __enter__(self):
return self
def __exit__(self, *a):
self._data.close()
def shutdown_browser(bref): def shutdown_browser(bref):
br = bref() br = bref()

8
src/calibre/scraper/test_fetch_backend.py
View File

@ -109,10 +109,10 @@ class TestFetchBackend(unittest.TestCase):
req = Request(url, headers=headers) req = Request(url, headers=headers)
else: else:
req = url req = url
res = br.open(req, data=data, timeout=timeout) with br.open(req, data=data, timeout=timeout) as res:
raw = res.read() raw = res.read()
ans = json.loads(raw) ans = json.loads(raw)
ans['final_url'] = res.geturl() ans['final_url'] = res.geturl()
return ans return ans
def test_with_timeout(no_response=True): def test_with_timeout(no_response=True):