Mirrors/linkding
2
0
Fork 0
mirror of https://github.com/sissbruecker/linkding synced 2024-11-12 23:07:06 +00:00
Code Issues Projects Releases Packages Wiki Activity

Use temp dir for favicon loader tests

Browse source
This commit is contained in:
Sascha Ißbrücker 2024-04-20 19:45:57 +02:00
parent 023a213ba6
commit 9cf5eb5ec0
1 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
bookmarks/tests/test_favicon_loader.py
View file

@ -1,8 +1,9 @@
import io import io
import os.path import os.path
import time import time
import tempfile
from pathlib import Path from pathlib import Path
from unittest import mock, skip from unittest import mock
from django.conf import settings from django.conf import settings
from django.test import TestCase, override_settings from django.test import TestCase, override_settings
@ -29,17 +30,21 @@ class MockStreamingResponse:
class FaviconLoaderTestCase(TestCase): class FaviconLoaderTestCase(TestCase):
def setUp(self) -> None: def setUp(self) -> None:
self.ensure_favicon_folder() self.temp_favicon_folder = tempfile.TemporaryDirectory()
self.clear_favicon_folder() self.favicon_folder_override = self.settings(
LD_FAVICON_FOLDER=self.temp_favicon_folder.name
)
self.favicon_folder_override.enable()
def tearDown(self) -> None:
self.temp_favicon_folder.cleanup()
self.favicon_folder_override.disable()
def create_mock_response(self, icon_data=mock_icon_data, content_type="image/png"): def create_mock_response(self, icon_data=mock_icon_data, content_type="image/png"):
mock_response = mock.Mock() mock_response = mock.Mock()
mock_response.raw = io.BytesIO(icon_data) mock_response.raw = io.BytesIO(icon_data)
return MockStreamingResponse(icon_data, content_type) return MockStreamingResponse(icon_data, content_type)
def ensure_favicon_folder(self):
Path(settings.LD_FAVICON_FOLDER).mkdir(parents=True, exist_ok=True)
def clear_favicon_folder(self): def clear_favicon_folder(self):
folder = Path(settings.LD_FAVICON_FOLDER) folder = Path(settings.LD_FAVICON_FOLDER)
for file in folder.iterdir(): for file in folder.iterdir():
@ -177,7 +182,6 @@ class FaviconLoaderTestCase(TestCase):
self.assertTrue(self.icon_exists("https_example_com.png")) self.assertTrue(self.icon_exists("https_example_com.png"))
self.clear_favicon_folder() self.clear_favicon_folder()
self.ensure_favicon_folder()
with mock.patch("requests.get") as mock_get: with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response( mock_get.return_value = self.create_mock_response(