django-components/tests/test_finders.py
Juro Oravec ad402fc619
refactor: fix compat with Django 5.2 Finder.find() (#1121)
* refactor: fix compat with Django 5.2 Finder.find()

* refactor: fix tests and linters
2025-04-12 09:35:33 +02:00

254 lines
9.8 KiB
Python

import re
from pathlib import Path
from django.contrib.staticfiles import finders
from django.contrib.staticfiles.management.commands.collectstatic import Command
from django.test import SimpleTestCase
from django_components.testing import djc_test
from .testutils import setup_test_config
setup_test_config({"autodiscover": False})
# This subclass allows us to call the `collectstatic` command from within Python.
# We call the `collect` method, which returns info about what files were collected.
#
# The methods below are overriden to ensure we don't make any filesystem changes
# (copy/delete), as the original command copies files. Thus we can safely test that
# our app works as intended.
class MockCollectstaticCommand(Command):
# NOTE: We do not expect this to be called
def clear_dir(self, path):
raise NotImplementedError()
# NOTE: We do not expect this to be called
def link_file(self, path, prefixed_path, source_storage):
raise NotImplementedError()
def copy_file(self, path, prefixed_path, source_storage):
# Skip this file if it was already copied earlier
if prefixed_path in self.copied_files:
return self.log("Skipping '%s' (already copied earlier)" % path)
# Delete the target file if needed or break
if not self.delete_file(path, prefixed_path, source_storage):
return
# The full path of the source file
source_path = source_storage.path(path)
# Finally start copying
if self.dry_run:
self.log("Pretending to copy '%s'" % source_path, level=1)
else:
self.log("Copying '%s'" % source_path, level=2)
# ############# OUR CHANGE ##############
# with source_storage.open(path) as source_file:
# self.storage.save(prefixed_path, source_file)
# ############# OUR CHANGE ##############
self.copied_files.append(prefixed_path)
def do_collect():
cmd = MockCollectstaticCommand()
cmd.set_options(
interactive=False,
verbosity=1,
link=False,
clear=False,
dry_run=False,
ignore_patterns=[],
use_default_ignore_patterns=True,
post_process=True,
)
collected = cmd.collect()
# Convert collected paths from string to Path, so we can run tests on both Unix and Windows
collected = {key: [Path(item) for item in items] for key, items in collected.items()}
return collected
common_settings = {
"STATIC_URL": "static/",
"STATIC_ROOT": "staticfiles",
"ROOT_URLCONF": __name__,
"INSTALLED_APPS": ("django_components", "django.contrib.staticfiles"),
}
COMPONENTS = {
"dirs": [Path(__file__).resolve().parent / "components"],
}
urlpatterns: list = []
class StaticFilesFinderTests(SimpleTestCase):
@djc_test(
django_settings={
**common_settings,
"STATICFILES_FINDERS": [
# Default finders
"django.contrib.staticfiles.finders.FileSystemFinder",
"django.contrib.staticfiles.finders.AppDirectoriesFinder",
],
},
components_settings=COMPONENTS,
)
def test_python_and_html_included(self):
collected = do_collect()
# Check that the component files are NOT loaded when our finder is NOT added
self.assertNotIn(Path("staticfiles/staticfiles.css"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.js"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.html"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.py"), collected["modified"])
self.assertListEqual(collected["unmodified"], [])
self.assertListEqual(collected["post_processed"], [])
@djc_test(
django_settings={
**common_settings,
"STATICFILES_FINDERS": [
# Default finders
"django.contrib.staticfiles.finders.FileSystemFinder",
"django.contrib.staticfiles.finders.AppDirectoriesFinder",
# Django components
"django_components.finders.ComponentsFileSystemFinder",
],
},
components_settings=COMPONENTS,
)
def test_python_and_html_omitted(self):
collected = do_collect()
# Check that our staticfiles_finder finds the files and OMITS .py and .html files
self.assertIn(Path("staticfiles/staticfiles.css"), collected["modified"])
self.assertIn(Path("staticfiles/staticfiles.js"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.html"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.py"), collected["modified"])
self.assertListEqual(collected["unmodified"], [])
self.assertListEqual(collected["post_processed"], [])
@djc_test(
django_settings={
**common_settings,
"STATICFILES_FINDERS": [
# Default finders
"django.contrib.staticfiles.finders.FileSystemFinder",
"django.contrib.staticfiles.finders.AppDirectoriesFinder",
# Django components
"django_components.finders.ComponentsFileSystemFinder",
],
},
components_settings={
**COMPONENTS,
"static_files_allowed": [
".js",
],
"static_files_forbidden": [],
},
)
def test_set_static_files_allowed(self):
collected = do_collect()
# Check that our staticfiles_finder finds the files and OMITS .py and .html files
self.assertNotIn(Path("staticfiles/staticfiles.css"), collected["modified"])
self.assertIn(Path("staticfiles/staticfiles.js"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.html"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.py"), collected["modified"])
self.assertListEqual(collected["unmodified"], [])
self.assertListEqual(collected["post_processed"], [])
@djc_test(
django_settings={
**common_settings,
"STATICFILES_FINDERS": [
# Default finders
"django.contrib.staticfiles.finders.FileSystemFinder",
"django.contrib.staticfiles.finders.AppDirectoriesFinder",
# Django components
"django_components.finders.ComponentsFileSystemFinder",
],
},
components_settings={
**COMPONENTS,
"static_files_allowed": [
re.compile(r".*"),
],
"static_files_forbidden": [
re.compile(r"\.(?:js)$"),
],
},
)
def test_set_forbidden_files(self):
collected = do_collect()
# Check that our staticfiles_finder finds the files and OMITS .py and .html files
self.assertIn(Path("staticfiles/staticfiles.css"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.js"), collected["modified"])
self.assertIn(Path("staticfiles/staticfiles.html"), collected["modified"])
self.assertIn(Path("staticfiles/staticfiles.py"), collected["modified"])
self.assertListEqual(collected["unmodified"], [])
self.assertListEqual(collected["post_processed"], [])
@djc_test(
django_settings={
**common_settings,
"STATICFILES_FINDERS": [
# Default finders
"django.contrib.staticfiles.finders.FileSystemFinder",
"django.contrib.staticfiles.finders.AppDirectoriesFinder",
# Django components
"django_components.finders.ComponentsFileSystemFinder",
],
},
components_settings={
**COMPONENTS,
"static_files_allowed": [
".js",
".css",
],
"static_files_forbidden": [
".js",
],
},
)
def test_set_both_allowed_and_forbidden_files(self):
collected = do_collect()
# Check that our staticfiles_finder finds the files and OMITS .py and .html files
self.assertIn(Path("staticfiles/staticfiles.css"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.js"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.html"), collected["modified"])
self.assertNotIn(Path("staticfiles/staticfiles.py"), collected["modified"])
self.assertListEqual(collected["unmodified"], [])
self.assertListEqual(collected["post_processed"], [])
# Handle deprecated `all` parameter:
# - In Django 5.2, the `all` parameter was deprecated in favour of `find_all`.
# - Between Django 5.2 (inclusive) and 6.1 (exclusive), the `all` parameter was still
# supported, but an error was raised if both were provided.
# - In Django 6.1, the `all` parameter was removed.
#
# See https://github.com/django/django/blob/5.2/django/contrib/staticfiles/finders.py#L58C9-L58C37
# And https://github.com/django-components/django-components/issues/1119
@djc_test(
django_settings={
**common_settings,
"STATICFILES_FINDERS": [
# Default finders
"django.contrib.staticfiles.finders.FileSystemFinder",
"django.contrib.staticfiles.finders.AppDirectoriesFinder",
# Django components
"django_components.finders.ComponentsFileSystemFinder",
],
},
components_settings=COMPONENTS,
)
def test_find_compat(self):
# NOTE: This would raise an error in Django 5.2 without a fix
result = finders.find("staticfiles/staticfiles.css")
assert Path(result) == Path("./tests/components/staticfiles/staticfiles.css").resolve()