mirror of
https://github.com/sissbruecker/linkding
synced 2024-11-23 03:43:10 +00:00
296 lines
11 KiB
Python
296 lines
11 KiB
Python
import random
|
|
from unittest.mock import patch, Mock
|
|
|
|
import requests
|
|
from django.test import TestCase, override_settings
|
|
from django.urls import reverse
|
|
from requests import RequestException
|
|
|
|
from bookmarks.models import UserProfile
|
|
from bookmarks.services import tasks
|
|
from bookmarks.tests.helpers import BookmarkFactoryMixin
|
|
from bookmarks.views.settings import app_version, get_version_info
|
|
|
|
|
|
class SettingsGeneralViewTestCase(TestCase, BookmarkFactoryMixin):
|
|
|
|
def setUp(self) -> None:
|
|
user = self.get_or_create_test_user()
|
|
self.client.force_login(user)
|
|
|
|
def create_profile_form_data(self, overrides=None):
|
|
if not overrides:
|
|
overrides = {}
|
|
form_data = {
|
|
"theme": UserProfile.THEME_AUTO,
|
|
"bookmark_date_display": UserProfile.BOOKMARK_DATE_DISPLAY_RELATIVE,
|
|
"bookmark_link_target": UserProfile.BOOKMARK_LINK_TARGET_BLANK,
|
|
"web_archive_integration": UserProfile.WEB_ARCHIVE_INTEGRATION_DISABLED,
|
|
"enable_sharing": False,
|
|
"enable_public_sharing": False,
|
|
"enable_favicons": False,
|
|
"tag_search": UserProfile.TAG_SEARCH_STRICT,
|
|
"display_url": False,
|
|
"permanent_notes": False,
|
|
"custom_css": "",
|
|
}
|
|
|
|
return {**form_data, **overrides}
|
|
|
|
def test_should_render_successfully(self):
|
|
response = self.client.get(reverse("bookmarks:settings.general"))
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_should_check_authentication(self):
|
|
self.client.logout()
|
|
response = self.client.get(reverse("bookmarks:settings.general"), follow=True)
|
|
|
|
self.assertRedirects(
|
|
response,
|
|
reverse("login") + "?next=" + reverse("bookmarks:settings.general"),
|
|
)
|
|
|
|
def test_update_profile(self):
|
|
form_data = {
|
|
"update_profile": "",
|
|
"theme": UserProfile.THEME_DARK,
|
|
"bookmark_date_display": UserProfile.BOOKMARK_DATE_DISPLAY_HIDDEN,
|
|
"bookmark_link_target": UserProfile.BOOKMARK_LINK_TARGET_SELF,
|
|
"web_archive_integration": UserProfile.WEB_ARCHIVE_INTEGRATION_ENABLED,
|
|
"enable_sharing": True,
|
|
"enable_public_sharing": True,
|
|
"enable_favicons": True,
|
|
"tag_search": UserProfile.TAG_SEARCH_LAX,
|
|
"display_url": True,
|
|
"permanent_notes": True,
|
|
"custom_css": "body { background-color: #000; }",
|
|
}
|
|
response = self.client.post(reverse("bookmarks:settings.general"), form_data)
|
|
html = response.content.decode()
|
|
|
|
self.user.profile.refresh_from_db()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(self.user.profile.theme, form_data["theme"])
|
|
self.assertEqual(
|
|
self.user.profile.bookmark_date_display, form_data["bookmark_date_display"]
|
|
)
|
|
self.assertEqual(
|
|
self.user.profile.bookmark_link_target, form_data["bookmark_link_target"]
|
|
)
|
|
self.assertEqual(
|
|
self.user.profile.web_archive_integration,
|
|
form_data["web_archive_integration"],
|
|
)
|
|
self.assertEqual(self.user.profile.enable_sharing, form_data["enable_sharing"])
|
|
self.assertEqual(
|
|
self.user.profile.enable_public_sharing, form_data["enable_public_sharing"]
|
|
)
|
|
self.assertEqual(
|
|
self.user.profile.enable_favicons, form_data["enable_favicons"]
|
|
)
|
|
self.assertEqual(self.user.profile.tag_search, form_data["tag_search"])
|
|
self.assertEqual(self.user.profile.display_url, form_data["display_url"])
|
|
self.assertEqual(
|
|
self.user.profile.permanent_notes, form_data["permanent_notes"]
|
|
)
|
|
self.assertEqual(self.user.profile.custom_css, form_data["custom_css"])
|
|
self.assertInHTML(
|
|
"""
|
|
<p class="form-input-hint">Profile updated</p>
|
|
""",
|
|
html,
|
|
)
|
|
|
|
def test_update_profile_should_not_be_called_without_respective_form_action(self):
|
|
form_data = {
|
|
"theme": UserProfile.THEME_DARK,
|
|
}
|
|
response = self.client.post(reverse("bookmarks:settings.general"), form_data)
|
|
html = response.content.decode()
|
|
|
|
self.user.profile.refresh_from_db()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(self.user.profile.theme, UserProfile.THEME_AUTO)
|
|
self.assertInHTML(
|
|
"""
|
|
<p class="form-input-hint">Profile updated</p>
|
|
""",
|
|
html,
|
|
count=0,
|
|
)
|
|
|
|
def test_enable_favicons_should_schedule_icon_update(self):
|
|
with patch.object(
|
|
tasks, "schedule_bookmarks_without_favicons"
|
|
) as mock_schedule_bookmarks_without_favicons:
|
|
# Enabling favicons schedules update
|
|
form_data = self.create_profile_form_data(
|
|
{
|
|
"update_profile": "",
|
|
"enable_favicons": True,
|
|
}
|
|
)
|
|
self.client.post(reverse("bookmarks:settings.general"), form_data)
|
|
|
|
mock_schedule_bookmarks_without_favicons.assert_called_once_with(self.user)
|
|
|
|
# No update scheduled if favicons are already enabled
|
|
mock_schedule_bookmarks_without_favicons.reset_mock()
|
|
|
|
self.client.post(reverse("bookmarks:settings.general"), form_data)
|
|
|
|
mock_schedule_bookmarks_without_favicons.assert_not_called()
|
|
|
|
# No update scheduled when disabling favicons
|
|
form_data = self.create_profile_form_data(
|
|
{
|
|
"enable_favicons": False,
|
|
}
|
|
)
|
|
|
|
self.client.post(reverse("bookmarks:settings.general"), form_data)
|
|
|
|
mock_schedule_bookmarks_without_favicons.assert_not_called()
|
|
|
|
def test_refresh_favicons(self):
|
|
with patch.object(
|
|
tasks, "schedule_refresh_favicons"
|
|
) as mock_schedule_refresh_favicons:
|
|
form_data = {
|
|
"refresh_favicons": "",
|
|
}
|
|
response = self.client.post(
|
|
reverse("bookmarks:settings.general"), form_data
|
|
)
|
|
html = response.content.decode()
|
|
|
|
mock_schedule_refresh_favicons.assert_called_once()
|
|
self.assertInHTML(
|
|
"""
|
|
<p class="form-input-hint">
|
|
Scheduled favicon update. This may take a while...
|
|
</p>
|
|
""",
|
|
html,
|
|
)
|
|
|
|
def test_refresh_favicons_should_not_be_called_without_respective_form_action(self):
|
|
with patch.object(
|
|
tasks, "schedule_refresh_favicons"
|
|
) as mock_schedule_refresh_favicons:
|
|
form_data = {}
|
|
response = self.client.post(
|
|
reverse("bookmarks:settings.general"), form_data
|
|
)
|
|
html = response.content.decode()
|
|
|
|
mock_schedule_refresh_favicons.assert_not_called()
|
|
self.assertInHTML(
|
|
"""
|
|
<p class="form-input-hint">
|
|
Scheduled favicon update. This may take a while...
|
|
</p>
|
|
""",
|
|
html,
|
|
count=0,
|
|
)
|
|
|
|
def test_refresh_favicons_should_be_visible_when_favicons_enabled_in_profile(self):
|
|
profile = self.get_or_create_test_user().profile
|
|
profile.enable_favicons = True
|
|
profile.save()
|
|
|
|
response = self.client.get(reverse("bookmarks:settings.general"))
|
|
html = response.content.decode()
|
|
|
|
self.assertInHTML(
|
|
"""
|
|
<button class="btn mt-2" name="refresh_favicons">Refresh Favicons</button>
|
|
""",
|
|
html,
|
|
count=1,
|
|
)
|
|
|
|
def test_refresh_favicons_should_not_be_visible_when_favicons_disabled_in_profile(
|
|
self,
|
|
):
|
|
profile = self.get_or_create_test_user().profile
|
|
profile.enable_favicons = False
|
|
profile.save()
|
|
|
|
response = self.client.get(reverse("bookmarks:settings.general"))
|
|
html = response.content.decode()
|
|
|
|
self.assertInHTML(
|
|
"""
|
|
<button class="btn mt-2" name="refresh_favicons">Refresh Favicons</button>
|
|
""",
|
|
html,
|
|
count=0,
|
|
)
|
|
|
|
@override_settings(LD_ENABLE_REFRESH_FAVICONS=False)
|
|
def test_refresh_favicons_should_not_be_visible_when_disabled(self):
|
|
profile = self.get_or_create_test_user().profile
|
|
profile.enable_favicons = True
|
|
profile.save()
|
|
|
|
response = self.client.get(reverse("bookmarks:settings.general"))
|
|
html = response.content.decode()
|
|
|
|
self.assertInHTML(
|
|
"""
|
|
<button class="btn mt-2" name="refresh_favicons">Refresh Favicons</button>
|
|
""",
|
|
html,
|
|
count=0,
|
|
)
|
|
|
|
def test_about_shows_version_info(self):
|
|
response = self.client.get(reverse("bookmarks:settings.general"))
|
|
html = response.content.decode()
|
|
|
|
self.assertInHTML(
|
|
f"""
|
|
<tr>
|
|
<td>Version</td>
|
|
<td>{get_version_info(random.random())}</td>
|
|
</tr>
|
|
""",
|
|
html,
|
|
)
|
|
|
|
def test_get_version_info_just_displays_latest_when_versions_are_equal(self):
|
|
latest_version_response_mock = Mock(
|
|
status_code=200, json=lambda: {"name": f"v{app_version}"}
|
|
)
|
|
with patch.object(requests, "get", return_value=latest_version_response_mock):
|
|
version_info = get_version_info(random.random())
|
|
self.assertEqual(version_info, f"{app_version} (latest)")
|
|
|
|
def test_get_version_info_shows_latest_version_when_versions_are_not_equal(self):
|
|
latest_version_response_mock = Mock(
|
|
status_code=200, json=lambda: {"name": f"v123.0.1"}
|
|
)
|
|
with patch.object(requests, "get", return_value=latest_version_response_mock):
|
|
version_info = get_version_info(random.random())
|
|
self.assertEqual(version_info, f"{app_version} (latest: 123.0.1)")
|
|
|
|
def test_get_version_info_silently_ignores_request_errors(self):
|
|
with patch.object(requests, "get", side_effect=RequestException()):
|
|
version_info = get_version_info(random.random())
|
|
self.assertEqual(version_info, f"{app_version}")
|
|
|
|
def test_get_version_info_handles_invalid_response(self):
|
|
latest_version_response_mock = Mock(status_code=403, json=lambda: {})
|
|
with patch.object(requests, "get", return_value=latest_version_response_mock):
|
|
version_info = get_version_info(random.random())
|
|
self.assertEqual(version_info, app_version)
|
|
|
|
latest_version_response_mock = Mock(status_code=200, json=lambda: {})
|
|
with patch.object(requests, "get", return_value=latest_version_response_mock):
|
|
version_info = get_version_info(random.random())
|
|
self.assertEqual(version_info, app_version)
|