mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-04-15 09:51:21 +00:00
46f13b2be9
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org> Reviewed-by: James Valleroy <jvalleroy@mailbox.org>
116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""
|
|
Tests for common FreedomBox views.
|
|
"""
|
|
|
|
import json
|
|
|
|
import pytest
|
|
from django.http import JsonResponse
|
|
from django.urls import resolve
|
|
|
|
from plinth.views import get_breadcrumbs, is_safe_url, json_exception
|
|
|
|
|
|
def test_get_breadcrumbs(rf, test_menu):
|
|
"""Test that computing breadcrumbs works."""
|
|
|
|
def _crumb(name: str, is_active: bool = False, url_name: str | None = None,
|
|
is_active_section: bool = False):
|
|
crumb = {'name': name, 'is_active': is_active, 'url_name': url_name}
|
|
if is_active_section:
|
|
crumb['is_active_section'] = True
|
|
|
|
return crumb
|
|
|
|
def _get(path: str):
|
|
request = rf.get(path)
|
|
request.resolver_match = resolve(path)
|
|
return get_breadcrumbs(request)
|
|
|
|
def _compare(dict1: dict[str, dict[str, str | bool]],
|
|
dict2: dict[str, dict[str, str | bool]]):
|
|
"""Compare dictionaries with order."""
|
|
assert list(dict1.items()) == list(dict2.items())
|
|
|
|
_compare(_get('/'), {'/': _crumb('Home', True, 'index', True)})
|
|
_compare(
|
|
_get('/apps/'), {
|
|
'/apps/': _crumb('Apps', True, 'apps', True),
|
|
'/': _crumb('Home', False, 'index'),
|
|
})
|
|
_compare(
|
|
_get('/apps/testapp/'), {
|
|
'/apps/testapp/': _crumb('Test App', True, 'testapp:index'),
|
|
'/apps/': _crumb('Apps', False, 'apps', True),
|
|
'/': _crumb('Home', False, 'index'),
|
|
})
|
|
_compare(
|
|
_get('/apps/testapp/create/'), {
|
|
'/apps/testapp/create/': _crumb('Here', True, 'testapp:create'),
|
|
'/apps/testapp/': _crumb('Test App', False, 'testapp:index'),
|
|
'/apps/': _crumb('Apps', False, 'apps', True),
|
|
'/': _crumb('Home', False, 'index'),
|
|
})
|
|
_compare(
|
|
_get('/test/1/2/3/'), {
|
|
'/test/1/2/3/': _crumb('Here', True, 'test', True),
|
|
'/': _crumb('Home', False, 'index'),
|
|
})
|
|
|
|
|
|
@pytest.mark.parametrize('url', [
|
|
'/freedombox/login/',
|
|
'/',
|
|
'safe',
|
|
])
|
|
def test_is_safe_url_valid_url(url):
|
|
"""Test valid URLs for safe URL checks."""
|
|
assert is_safe_url(url)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'url',
|
|
[
|
|
'',
|
|
None,
|
|
'\\freedombox',
|
|
'///freedombox',
|
|
'https://example.com/freedombox/login/',
|
|
'https:///example.com',
|
|
'https:///freedombox/login',
|
|
'ftp://example.com',
|
|
'https://[aabb::ccdd', # Invalid IPv6
|
|
])
|
|
def test_is_safe_url_invalid_url(url):
|
|
"""Test invalid URLs for safe URL checks."""
|
|
assert not is_safe_url(url)
|
|
|
|
|
|
def test_json_exception(rf):
|
|
"""Test handling exceptions in JSON views."""
|
|
|
|
@json_exception
|
|
def error_view(request):
|
|
raise Exception('exception-message')
|
|
|
|
assert error_view.__name__ == 'error_view'
|
|
|
|
request = rf.get('')
|
|
response = error_view(request)
|
|
assert response.status_code == 500
|
|
json_content = json.loads(response.content)
|
|
assert not json_content['result']
|
|
assert json_content['error_name'] == 'Exception'
|
|
assert json_content['error_string'] == 'exception-message'
|
|
|
|
def success_view(request):
|
|
return JsonResponse({'result': True, 'error_string': None})
|
|
|
|
request = rf.get('')
|
|
response = success_view(request)
|
|
assert response.status_code == 200
|
|
json_content = json.loads(response.content)
|
|
assert json_content['result']
|
|
assert json_content['error_string'] is None
|