44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
ALLOWED_URL_PREFIXES = ("https://", "mailto:", "tel:")
|
||
|
|
URL_KEYS = {"url", "imprint", "imprint_url"}
|
||
|
|
|
||
|
|
|
||
|
|
class TestSampleConfigUrls(unittest.TestCase):
|
||
|
|
def setUp(self) -> None:
|
||
|
|
repo_root = Path(__file__).resolve().parents[2]
|
||
|
|
sample_config_path = repo_root / "app" / "config.sample.yaml"
|
||
|
|
with sample_config_path.open("r", encoding="utf-8") as handle:
|
||
|
|
self.sample_config = yaml.safe_load(handle)
|
||
|
|
|
||
|
|
def _iter_urls(self, data, path="root"):
|
||
|
|
if isinstance(data, dict):
|
||
|
|
for key, value in data.items():
|
||
|
|
next_path = f"{path}.{key}"
|
||
|
|
if key in URL_KEYS and isinstance(value, str):
|
||
|
|
yield next_path, value
|
||
|
|
yield from self._iter_urls(value, next_path)
|
||
|
|
elif isinstance(data, list):
|
||
|
|
for index, item in enumerate(data):
|
||
|
|
yield from self._iter_urls(item, f"{path}[{index}]")
|
||
|
|
|
||
|
|
def test_sample_config_urls_use_safe_schemes(self):
|
||
|
|
invalid_urls = [
|
||
|
|
f"{path} -> {url}"
|
||
|
|
for path, url in self._iter_urls(self.sample_config)
|
||
|
|
if not url.startswith(ALLOWED_URL_PREFIXES)
|
||
|
|
]
|
||
|
|
|
||
|
|
self.assertFalse(
|
||
|
|
invalid_urls,
|
||
|
|
"The sample config contains URLs with unsupported schemes:\n"
|
||
|
|
+ "\n".join(f"- {entry}" for entry in invalid_urls),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|