marcel/linkding
1
0
Fork 0
mirror of https://github.com/sissbruecker/linkding.git synced 2025-08-07 18:58:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add black code formatter

Browse Source
This commit is contained in:
Sascha Ißbrücker
2024-01-27 11:29:16 +01:00
parent 6775633be5
commit 98b9a9c1a0
128 changed files with 7181 additions and 4264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

112
bookmarks/tests/test_favicon_loader.py
View File

@@ -9,13 +9,13 @@ from django.test import TestCase, override_settings
from bookmarks.services import favicon_loader
mock_icon_data = b'mock_icon'
mock_icon_data = b"mock_icon"
class MockStreamingResponse:
def __init__(self, data=mock_icon_data, content_type='image/png'):
def __init__(self, data=mock_icon_data, content_type="image/png"):
self.chunks = [data]
self.headers = {'Content-Type': content_type}
self.headers = {"Content-Type": content_type}
def iter_content(self, **kwargs):
return self.chunks
@@ -32,7 +32,7 @@ class FaviconLoaderTestCase(TestCase):
self.ensure_favicon_folder()
self.clear_favicon_folder()
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.raw = io.BytesIO(icon_data)
return MockStreamingResponse(icon_data, content_type)
@@ -59,18 +59,20 @@ class FaviconLoaderTestCase(TestCase):
return len(files)
def test_load_favicon(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon("https://example.com")
# should create icon file
self.assertTrue(self.icon_exists('https_example_com.png'))
self.assertTrue(self.icon_exists("https_example_com.png"))
# should store image data
self.assertEqual(mock_icon_data, self.get_icon_data('https_example_com.png'))
self.assertEqual(
mock_icon_data, self.get_icon_data("https_example_com.png")
)
def test_load_favicon_creates_folder_if_not_exists(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
folder = Path(settings.LD_FAVICON_FOLDER)
@@ -78,99 +80,109 @@ class FaviconLoaderTestCase(TestCase):
self.assertFalse(folder.exists())
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon("https://example.com")
self.assertTrue(folder.exists())
def test_load_favicon_creates_single_icon_for_same_base_url(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon('https://example.com?foo=bar')
favicon_loader.load_favicon('https://example.com/foo')
favicon_loader.load_favicon("https://example.com")
favicon_loader.load_favicon("https://example.com?foo=bar")
favicon_loader.load_favicon("https://example.com/foo")
self.assertEqual(1, self.count_icons())
self.assertTrue(self.icon_exists('https_example_com.png'))
self.assertTrue(self.icon_exists("https_example_com.png"))
def test_load_favicon_creates_multiple_icons_for_different_base_url(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon('https://sub.example.com')
favicon_loader.load_favicon('https://other-domain.com')
favicon_loader.load_favicon("https://example.com")
favicon_loader.load_favicon("https://sub.example.com")
favicon_loader.load_favicon("https://other-domain.com")
self.assertEqual(3, self.count_icons())
self.assertTrue(self.icon_exists('https_example_com.png'))
self.assertTrue(self.icon_exists('https_sub_example_com.png'))
self.assertTrue(self.icon_exists('https_other_domain_com.png'))
self.assertTrue(self.icon_exists("https_example_com.png"))
self.assertTrue(self.icon_exists("https_sub_example_com.png"))
self.assertTrue(self.icon_exists("https_other_domain_com.png"))
def test_load_favicon_caches_icons(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_file = favicon_loader.load_favicon('https://example.com')
favicon_file = favicon_loader.load_favicon("https://example.com")
mock_get.assert_called()
self.assertEqual(favicon_file, 'https_example_com.png')
self.assertEqual(favicon_file, "https_example_com.png")
mock_get.reset_mock()
updated_favicon_file = favicon_loader.load_favicon('https://example.com')
updated_favicon_file = favicon_loader.load_favicon("https://example.com")
mock_get.assert_not_called()
self.assertEqual(favicon_file, updated_favicon_file)
def test_load_favicon_updates_stale_icon(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon("https://example.com")
icon_path = self.get_icon_path('https_example_com.png')
icon_path = self.get_icon_path("https_example_com.png")
updated_mock_icon_data = b'updated_mock_icon'
mock_get.return_value = self.create_mock_response(icon_data=updated_mock_icon_data)
updated_mock_icon_data = b"updated_mock_icon"
mock_get.return_value = self.create_mock_response(
icon_data=updated_mock_icon_data
)
mock_get.reset_mock()
# change icon modification date so it is not stale yet
nearly_one_day_ago = time.time() - 60 * 60 * 23
os.utime(icon_path.absolute(), (nearly_one_day_ago, nearly_one_day_ago))
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon("https://example.com")
mock_get.assert_not_called()
# change icon modification date so it is considered stale
one_day_ago = time.time() - 60 * 60 * 24
os.utime(icon_path.absolute(), (one_day_ago, one_day_ago))
favicon_loader.load_favicon('https://example.com')
favicon_loader.load_favicon("https://example.com")
mock_get.assert_called()
self.assertEqual(updated_mock_icon_data, self.get_icon_data('https_example_com.png'))
self.assertEqual(
updated_mock_icon_data, self.get_icon_data("https_example_com.png")
)
@override_settings(LD_FAVICON_PROVIDER='https://custom.icons.com/?url={url}')
@override_settings(LD_FAVICON_PROVIDER="https://custom.icons.com/?url={url}")
def test_custom_provider_with_url_param(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_loader.load_favicon('https://example.com/foo?bar=baz')
mock_get.assert_called_with('https://custom.icons.com/?url=https://example.com', stream=True)
favicon_loader.load_favicon("https://example.com/foo?bar=baz")
mock_get.assert_called_with(
"https://custom.icons.com/?url=https://example.com", stream=True
)
@override_settings(LD_FAVICON_PROVIDER='https://custom.icons.com/?url={domain}')
@override_settings(LD_FAVICON_PROVIDER="https://custom.icons.com/?url={domain}")
def test_custom_provider_with_domain_param(self):
with mock.patch('requests.get') as mock_get:
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response()
favicon_loader.load_favicon('https://example.com/foo?bar=baz')
mock_get.assert_called_with('https://custom.icons.com/?url=example.com', stream=True)
favicon_loader.load_favicon("https://example.com/foo?bar=baz")
mock_get.assert_called_with(
"https://custom.icons.com/?url=example.com", stream=True
)
def test_guess_file_extension(self):
with mock.patch('requests.get') as mock_get:
mock_get.return_value = self.create_mock_response(content_type='image/png')
favicon_loader.load_favicon('https://example.com')
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response(content_type="image/png")
favicon_loader.load_favicon("https://example.com")
self.assertTrue(self.icon_exists('https_example_com.png'))
self.assertTrue(self.icon_exists("https_example_com.png"))
self.clear_favicon_folder()
self.ensure_favicon_folder()
with mock.patch('requests.get') as mock_get:
mock_get.return_value = self.create_mock_response(content_type='image/x-icon')
favicon_loader.load_favicon('https://example.com')
with mock.patch("requests.get") as mock_get:
mock_get.return_value = self.create_mock_response(
content_type="image/x-icon"
)
favicon_loader.load_favicon("https://example.com")
self.assertTrue(self.icon_exists('https_example_com.ico'))
self.assertTrue(self.icon_exists("https_example_com.ico"))