From 09184d6e8b8de8d9254e2f5a1a2b9a6a54ef33d6 Mon Sep 17 00:00:00 2001
From: Anentropic <ego@anentropic.com>
Date: Fri, 31 May 2024 23:29:09 +0100
Subject: [PATCH] support Jinja2 template backend, providing extension and tags

---
 README.md                                     |  43 +++
 django_vite/templatetags/base.py              | 180 +++++++++++
 django_vite/templatetags/django_vite.py       | 209 +------------
 django_vite/templatetags/jinja.py             |  48 +++
 tests/conftest.py                             |  11 +-
 tests/tests/jinja/test_vite_asset.py          | 287 ++++++++++++++++++
 tests/tests/jinja/test_vite_asset_url.py      |  41 +++
 tests/tests/jinja/test_vite_hmr_client.py     |  80 +++++
 tests/tests/jinja/test_vite_legacy_asset.py   |  72 +++++
 .../tests/jinja/test_vite_legacy_polyfills.py | 156 ++++++++++
 tests/tests/jinja/test_vite_preload_asset.py  |  43 +++
 tests/tests/jinja/test_vite_react_refresh.py  | 134 ++++++++
 tests/tests/test_asset_loader.py              |   6 +-
 13 files changed, 1114 insertions(+), 196 deletions(-)
 create mode 100644 django_vite/templatetags/base.py
 create mode 100644 django_vite/templatetags/jinja.py
 create mode 100644 tests/tests/jinja/test_vite_asset.py
 create mode 100644 tests/tests/jinja/test_vite_asset_url.py
 create mode 100644 tests/tests/jinja/test_vite_hmr_client.py
 create mode 100644 tests/tests/jinja/test_vite_legacy_asset.py
 create mode 100644 tests/tests/jinja/test_vite_legacy_polyfills.py
 create mode 100644 tests/tests/jinja/test_vite_preload_asset.py
 create mode 100644 tests/tests/jinja/test_vite_react_refresh.py

diff --git a/README.md b/README.md
index 664f968..fc9810a 100644
--- a/README.md
+++ b/README.md
@@ -216,6 +216,49 @@ If you want to overrides default attributes just add them like new attributes :
 
 Although it's recommended to keep the default `type="module"` attribute as ViteJS build scripts as ES6 modules.
 
+### Jinja2 template backend
+
+If you are using Django with the Jinja2 template backend then alternative versions of the templatetags are provided, via an Extension.
+
+If you are using [`django-jinja` library](https://niwi.nz/django-jinja/latest/#_add_additional_extensions) then you can:
+
+```python
+from django_jinja.builtins import DEFAULT_EXTENSIONS
+
+"OPTIONS": {
+    "extensions": DEFAULT_EXTENSIONS + [
+        # Your extensions here...
+        "django_vite.templatetags.jinja.DjangoViteExtension"
+    ]
+}
+```
+
+Or for a [vanilla Django + Jinja2 setup](https://docs.djangoproject.com/fr/4.2/topics/templates/#django.template.backends.jinja2.Jinja2) you can do something like:
+
+```python
+from django.templatetags.static import static
+from django.urls import reverse
+
+from jinja2 import Environment
+from django_vite.templatetags.jinja import DjangoViteExtension
+
+
+def environment(**options):
+    options.setdefault("extensions", []).append(DjangoViteExtension)
+    env = Environment(**options)
+    return env
+```
+
+Then usage in your templates is similar, but adjusted as is normal for Jinja2 rather than Django template language.
+
+- Omit the `{% load django_vite %}` which is not needed.
+- Replace tag syntax with function calls, e.g.
+  - `{% vite_react_refresh %}` becomes:
+    `{{ vite_react_refresh() }}`
+  - `{% vite_asset '<path to your asset>' foo="bar" %}` becomes:
+    `{{ vite_asset('<path to your asset>', foo="bar") }}`
+    etc
+
 ## Vite Legacy Plugin
 
 If you want to consider legacy browsers that don't support ES6 modules loading
diff --git a/django_vite/templatetags/base.py b/django_vite/templatetags/base.py
new file mode 100644
index 0000000..e94fdc4
--- /dev/null
+++ b/django_vite/templatetags/base.py
@@ -0,0 +1,180 @@
+from typing import Dict
+
+from django_vite.core.asset_loader import DjangoViteAssetLoader, DEFAULT_APP_NAME
+
+
+def vite_hmr_client(app: str = DEFAULT_APP_NAME, **kwargs: Dict[str, str]) -> str:
+    """
+    Generates the script tag for the Vite WS client for HMR.
+    Only used in development, in production this method returns
+    an empty string.
+
+    Arguments:
+        config {str} -- Configuration to use.
+
+    Returns:
+        str -- The script tag or an empty string.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+    """
+
+    return DjangoViteAssetLoader.instance().generate_vite_ws_client(app, **kwargs)
+
+
+def vite_asset(
+    path: str,
+    app: str = DEFAULT_APP_NAME,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates a <script> tag for this JS/TS asset, a <link> tag for
+    all of its CSS dependencies, and a <link rel="modulepreload">
+    for all js dependencies, as listed in the manifest file
+    In development Vite loads all by itself.
+
+    Arguments:
+        path {str} -- Path to a Vite JS/TS asset to include.
+        config {str} -- Configuration to use.
+
+    Returns:
+        str -- All tags to import this file in your HTML page.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If cannot find the file path in the
+            manifest (only in production).
+
+    Returns:
+        str -- The <script> tag and all <link> tags to import this
+            asset in your page.
+    """
+    assert path is not None
+    return DjangoViteAssetLoader.instance().generate_vite_asset(path, app, **kwargs)
+
+
+def vite_preload_asset(path: str, app: str = DEFAULT_APP_NAME) -> str:
+    """
+    Generates preloadmodule tag for this JS/TS asset and preloads
+    all of its CSS and JS dependencies by reading the manifest
+    file (for production only).
+    In development does nothing.
+
+    Arguments:
+        path {str} -- Path to a Vite JS/TS asset to include.
+
+    Returns:
+        str -- All tags to import this file in your HTML page.
+
+    Raises:
+        RuntimeError: If cannot find the file path in the
+            manifest (only in production).
+
+    """
+    assert path is not None
+    return DjangoViteAssetLoader.instance().preload_vite_asset(path, app)
+
+
+def vite_asset_url(path: str, app: str = DEFAULT_APP_NAME) -> str:
+    """
+    Generates only the URL of an asset managed by ViteJS.
+    Warning, this function does not generate URLs for dependant assets.
+
+    Arguments:
+        path {str} -- Path to a Vite asset.
+        config {str} -- Configuration to use.
+
+    Raises:
+        RuntimeError: If cannot find the asset path in the
+            manifest (only in production).
+
+    Returns:
+        str -- The URL of this asset.
+    """
+    assert path is not None
+    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, app)
+
+
+def vite_legacy_polyfills(app: str = DEFAULT_APP_NAME, **kwargs: Dict[str, str]) -> str:
+    """
+    Generates a <script> tag to the polyfills generated
+    by '@vitejs/plugin-legacy' if used.
+    This tag must be included at end of the <body> before including
+    other legacy scripts.
+
+    Arguments:
+        app {str} -- Configuration to use.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If polyfills path not found inside
+            the 'manifest.json' (only in production).
+
+    Returns:
+        str -- The script tag to the polyfills.
+    """
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
+        app, **kwargs
+    )
+
+
+def vite_legacy_asset(
+    path: str,
+    app: str = DEFAULT_APP_NAME,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates a <script> tag for legacy assets JS/TS
+    generated by '@vitejs/plugin-legacy'
+    (in production only, in development do nothing).
+
+    Arguments:
+        path {str} -- Path to a Vite asset to include
+            (must contains '-legacy' in its name).
+        app {str} -- Configuration to use.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If cannot find the asset path in
+            the manifest (only in production).
+
+    Returns:
+        str -- The script tag of this legacy asset.
+    """
+
+    assert path is not None
+
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
+        path, app, **kwargs
+    )
+
+
+def vite_react_refresh(
+    app: str = DEFAULT_APP_NAME,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates the script for the Vite React Refresh for HMR.
+    Only used in development, in production this method returns
+    an empty string.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Returns:
+        str -- The script or an empty string.
+    """
+    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(
+        app, **kwargs
+    )
diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 33aab81..d6af839 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -1,199 +1,22 @@
-from typing import Dict
-
 from django import template
 from django.utils.safestring import mark_safe
 
-from django_vite.core.asset_loader import DjangoViteAssetLoader, DEFAULT_APP_NAME
-
+from django_vite.templatetags.base import (
+    vite_asset as base_vite_asset,
+    vite_asset_url as base_vite_asset_url,
+    vite_hmr_client as base_vite_hmr_client,
+    vite_legacy_asset as base_vite_legacy_asset,
+    vite_legacy_polyfills as base_vite_legacy_polyfills,
+    vite_preload_asset as base_vite_preload_asset,
+    vite_react_refresh as base_vite_react_refresh,
+)
 
 register = template.Library()
 
-
-@register.simple_tag
-@mark_safe
-def vite_hmr_client(app: str = DEFAULT_APP_NAME, **kwargs: Dict[str, str]) -> str:
-    """
-    Generates the script tag for the Vite WS client for HMR.
-    Only used in development, in production this method returns
-    an empty string.
-
-    Arguments:
-        config {str} -- Configuration to use.
-
-    Returns:
-        str -- The script tag or an empty string.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-    """
-
-    return DjangoViteAssetLoader.instance().generate_vite_ws_client(app, **kwargs)
-
-
-@register.simple_tag
-@mark_safe
-def vite_asset(
-    path: str,
-    app: str = DEFAULT_APP_NAME,
-    **kwargs: Dict[str, str],
-) -> str:
-    """
-    Generates a <script> tag for this JS/TS asset, a <link> tag for
-    all of its CSS dependencies, and a <link rel="modulepreload">
-    for all js dependencies, as listed in the manifest file
-    In development Vite loads all by itself.
-
-    Arguments:
-        path {str} -- Path to a Vite JS/TS asset to include.
-        config {str} -- Configuration to use.
-
-    Returns:
-        str -- All tags to import this file in your HTML page.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If cannot find the file path in the
-            manifest (only in production).
-
-    Returns:
-        str -- The <script> tag and all <link> tags to import this
-            asset in your page.
-    """
-    assert path is not None
-    return DjangoViteAssetLoader.instance().generate_vite_asset(path, app, **kwargs)
-
-
-@register.simple_tag
-@mark_safe
-def vite_preload_asset(path: str, app: str = DEFAULT_APP_NAME) -> str:
-    """
-    Generates preloadmodule tag for this JS/TS asset and preloads
-    all of its CSS and JS dependencies by reading the manifest
-    file (for production only).
-    In development does nothing.
-
-    Arguments:
-        path {str} -- Path to a Vite JS/TS asset to include.
-
-    Returns:
-        str -- All tags to import this file in your HTML page.
-
-    Raises:
-        RuntimeError: If cannot find the file path in the
-            manifest (only in production).
-
-    """
-    assert path is not None
-    return DjangoViteAssetLoader.instance().preload_vite_asset(path, app)
-
-
-@register.simple_tag
-def vite_asset_url(path: str, app: str = DEFAULT_APP_NAME) -> str:
-    """
-    Generates only the URL of an asset managed by ViteJS.
-    Warning, this function does not generate URLs for dependant assets.
-
-    Arguments:
-        path {str} -- Path to a Vite asset.
-        config {str} -- Configuration to use.
-
-    Raises:
-        RuntimeError: If cannot find the asset path in the
-            manifest (only in production).
-
-    Returns:
-        str -- The URL of this asset.
-    """
-    assert path is not None
-    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, app)
-
-
-@register.simple_tag
-@mark_safe
-def vite_legacy_polyfills(app: str = DEFAULT_APP_NAME, **kwargs: Dict[str, str]) -> str:
-    """
-    Generates a <script> tag to the polyfills generated
-    by '@vitejs/plugin-legacy' if used.
-    This tag must be included at end of the <body> before including
-    other legacy scripts.
-
-    Arguments:
-        app {str} -- Configuration to use.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If polyfills path not found inside
-            the 'manifest.json' (only in production).
-
-    Returns:
-        str -- The script tag to the polyfills.
-    """
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        app, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_legacy_asset(
-    path: str,
-    app: str = DEFAULT_APP_NAME,
-    **kwargs: Dict[str, str],
-) -> str:
-    """
-    Generates a <script> tag for legacy assets JS/TS
-    generated by '@vitejs/plugin-legacy'
-    (in production only, in development do nothing).
-
-    Arguments:
-        path {str} -- Path to a Vite asset to include
-            (must contains '-legacy' in its name).
-        app {str} -- Configuration to use.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If cannot find the asset path in
-            the manifest (only in production).
-
-    Returns:
-        str -- The script tag of this legacy asset.
-    """
-
-    assert path is not None
-
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, app, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_react_refresh(
-    app: str = DEFAULT_APP_NAME,
-    **kwargs: Dict[str, str],
-) -> str:
-    """
-    Generates the script for the Vite React Refresh for HMR.
-    Only used in development, in production this method returns
-    an empty string.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Returns:
-        str -- The script or an empty string.
-    """
-    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(
-        app, **kwargs
-    )
+vite_hmr_client = register.simple_tag(mark_safe(base_vite_hmr_client))
+vite_asset = register.simple_tag(mark_safe(base_vite_asset))
+vite_preload_asset = register.simple_tag(mark_safe(base_vite_preload_asset))
+vite_asset_url = register.simple_tag(base_vite_asset_url)
+vite_legacy_polyfills = register.simple_tag(mark_safe(base_vite_legacy_polyfills))
+vite_legacy_asset = register.simple_tag(mark_safe(base_vite_legacy_asset))
+vite_react_refresh = register.simple_tag(mark_safe(base_vite_react_refresh))
diff --git a/django_vite/templatetags/jinja.py b/django_vite/templatetags/jinja.py
new file mode 100644
index 0000000..76d70db
--- /dev/null
+++ b/django_vite/templatetags/jinja.py
@@ -0,0 +1,48 @@
+from functools import wraps
+
+from jinja2.ext import Extension
+from markupsafe import Markup
+
+from django_vite.templatetags.base import (
+    vite_asset as base_vite_asset,
+    vite_asset_url as base_vite_asset_url,
+    vite_hmr_client as base_vite_hmr_client,
+    vite_legacy_asset as base_vite_legacy_asset,
+    vite_legacy_polyfills as base_vite_legacy_polyfills,
+    vite_preload_asset as base_vite_preload_asset,
+    vite_react_refresh as base_vite_react_refresh,
+)
+
+
+def mark_safe(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        return Markup(func(*args, **kwargs))
+
+    return wrapper
+
+
+vite_hmr_client = mark_safe(base_vite_hmr_client)
+vite_asset = mark_safe(base_vite_asset)
+vite_preload_asset = mark_safe(base_vite_preload_asset)
+vite_asset_url = base_vite_asset_url
+vite_legacy_polyfills = mark_safe(base_vite_legacy_polyfills)
+vite_legacy_asset = mark_safe(base_vite_legacy_asset)
+vite_react_refresh = mark_safe(base_vite_react_refresh)
+
+
+class DjangoViteExtension(Extension):
+    def __init__(self, environment):
+        super().__init__(environment)
+
+        environment.globals.update(
+            {
+                "vite_hmr_client": vite_hmr_client,
+                "vite_asset": vite_asset,
+                "vite_preload_asset": vite_preload_asset,
+                "vite_asset_url": vite_asset_url,
+                "vite_legacy_polyfills": vite_legacy_polyfills,
+                "vite_legacy_asset": vite_legacy_asset,
+                "vite_react_refresh": vite_react_refresh,
+            }
+        )
diff --git a/tests/conftest.py b/tests/conftest.py
index 9c0efac..3cbc94a 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,7 +1,10 @@
-from typing import Dict, Any
+from typing import Any, Dict
+
 import pytest
+from jinja2 import Environment
 
 from django_vite.core.asset_loader import DjangoViteAssetLoader
+from django_vite.templatetags.jinja import DjangoViteExtension
 
 __PYTEST_EMPTY__ = "__PYTEST_EMPTY__"
 __PYTEST_DELETE__ = "__PYTEST_DELETE__"
@@ -133,3 +136,9 @@ def dev_mode_all(request, patch_settings):
     settings that we support.
     """
     return patch_settings(request.param)
+
+
+@pytest.fixture()
+def jinja_env():
+    env = Environment(extensions=[DjangoViteExtension], autoescape=True)
+    return env
diff --git a/tests/tests/jinja/test_vite_asset.py b/tests/tests/jinja/test_vite_asset.py
new file mode 100644
index 0000000..ee8336f
--- /dev/null
+++ b/tests/tests/jinja/test_vite_asset.py
@@ -0,0 +1,287 @@
+import pytest
+from bs4 import BeautifulSoup
+
+from django_vite.core.exceptions import (
+    DjangoViteAssetNotFoundError,
+    DjangoViteConfigNotFoundError,
+)
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+@pytest.mark.parametrize("as_default", [True, False])
+def test_vite_asset_returns_dev_tags(jinja_env, as_default):
+    if as_default:
+        template = jinja_env.from_string(
+            """
+                {{ vite_asset("src/entry.ts", app="default") }}
+        """
+        )
+    else:
+        template = jinja_env.from_string(
+            """
+                {{ vite_asset("src/entry.ts") }}
+        """
+        )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:5173/static/src/entry.ts"
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+@pytest.mark.parametrize("as_default", [True, False])
+def test_vite_asset_returns_production_tags(jinja_env, as_default):
+    if as_default:
+        template = jinja_env.from_string(
+            """
+                {{ vite_asset("src/entry.ts", app="default") }}
+        """
+        )
+    else:
+        template = jinja_env.from_string(
+            """
+                {{ vite_asset("src/entry.ts") }}
+        """
+        )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_asset_raises_without_path(jinja_env):
+    template = jinja_env.from_string(
+        """
+            {{ vite_asset() }}
+    """
+    )
+    with pytest.raises(TypeError):
+        template.render({})
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_asset_raises_nonexistent_entry(jinja_env):
+    template = jinja_env.from_string(
+        """
+            {{ vite_asset("src/fake.ts") }}
+    """
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template.render({})
+
+
+@pytest.mark.parametrize(
+    "patch_settings",
+    [
+        {
+            "DJANGO_VITE_DEV_MODE": True,
+            "DJANGO_VITE_STATIC_URL_PREFIX": "custom/prefix",
+        },
+        {
+            "DJANGO_VITE_DEV_MODE": True,
+            "DJANGO_VITE_STATIC_URL_PREFIX": "custom/prefix/",
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": True,
+                    "static_url_prefix": "custom/prefix",
+                }
+            }
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": True,
+                    "static_url_prefix": "custom/prefix/",
+                }
+            }
+        },
+    ],
+    indirect=True,
+)
+def test_vite_asset_dev_prefix(jinja_env, patch_settings):
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert (
+        script_tag["src"] == "http://localhost:5173/static/custom/prefix/src/entry.ts"
+    )
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.parametrize(
+    "patch_settings",
+    [
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_STATIC_URL_PREFIX": "custom/prefix",
+        },
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_STATIC_URL_PREFIX": "custom/prefix/",
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": False,
+                    "static_url_prefix": "custom/prefix",
+                }
+            }
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": False,
+                    "static_url_prefix": "custom/prefix/",
+                }
+            }
+        },
+    ],
+    indirect=True,
+)
+def test_vite_asset_production_prefix(jinja_env, patch_settings):
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "custom/prefix/assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_asset_production_staticfiles_storage(jinja_env, patch_settings):
+    patch_settings(
+        {
+            "INSTALLED_APPS": ["django_vite", "django.contrib.staticfiles"],
+        }
+    )
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "/static/assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
+@pytest.mark.usefixtures("dev_mode_all")
+def test_vite_asset_override_default_attribute(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts", crossorigin="anonymous") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["crossorigin"] == "anonymous"
+
+
+@pytest.mark.usefixtures("dev_mode_all")
+def test_vite_asset_kebab_attribute(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts", data_item_track="reload", data_other="3") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["data-item-track"] == "reload"
+    assert script_tag["data-other"] == "3"
+
+
+def test_vite_asset_custom_attributes(jinja_env, dev_mode_all):
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts", foo="bar", hello="world") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["foo"] == "bar"
+    assert script_tag["hello"] == "world"
+
+
+def test_vite_asset_nonexistent_app(jinja_env, dev_mode_true):
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.ts", app="bad_app") }}
+    """
+    )
+    with pytest.raises(DjangoViteConfigNotFoundError) as error:
+        template.render({})
+
+    assert "Cannot find bad_app in DJANGO_VITE settings" in str(error)
+
+
+@pytest.fixture()
+def external_vue_app(jinja_env, patch_settings, settings):
+    def _wrapper(dev_mode: bool):
+        return patch_settings(
+            {
+                "DJANGO_VITE": {
+                    "external_vue_app": {
+                        "dev_mode": dev_mode,
+                        "static_url_prefix": "custom/prefix",
+                        "dev_server_port": 5555,
+                        "manifest_path": settings.STATIC_ROOT.parent
+                        / "external_vue_app"
+                        / "manifest.json",
+                    }
+                }
+            }
+        )
+
+    return _wrapper
+
+
+def test_vite_asset_external_app_dev(jinja_env, external_vue_app):
+    external_vue_app(dev_mode=True)
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.js", app="external_vue_app") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert (
+        script_tag["src"] == "http://localhost:5555/static/custom/prefix/src/entry.js"
+    )
+
+
+def test_vite_asset_external_app_production(jinja_env, external_vue_app):
+    external_vue_app(dev_mode=False)
+    template = jinja_env.from_string(
+        """
+        {{ vite_asset("src/entry.js", app="external_vue_app") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "custom/prefix/assets/entry-5c085aac.js"
diff --git a/tests/tests/jinja/test_vite_asset_url.py b/tests/tests/jinja/test_vite_asset_url.py
new file mode 100644
index 0000000..82eb993
--- /dev/null
+++ b/tests/tests/jinja/test_vite_asset_url.py
@@ -0,0 +1,41 @@
+import pytest
+from bs4 import BeautifulSoup
+
+from django_vite.core.exceptions import DjangoViteAssetNotFoundError
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_asset_url_returns_dev_url(jinja_env):
+    template = jinja_env.from_string(
+        """
+        <script src="{{ vite_asset_url('src/entry.ts') }}">
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:5173/static/src/entry.ts"
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_asset_url_returns_production_url(jinja_env):
+    template = jinja_env.from_string(
+        """
+        <script src="{{ vite_asset_url('src/entry.ts') }}">
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_asset_url_raises_nonexistent_entry(jinja_env):
+    template = jinja_env.from_string(
+        """
+        <script src="{{ vite_asset_url('src/fake.ts') }}">
+    """
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template.render({})
diff --git a/tests/tests/jinja/test_vite_hmr_client.py b/tests/tests/jinja/test_vite_hmr_client.py
new file mode 100644
index 0000000..a8598f3
--- /dev/null
+++ b/tests/tests/jinja/test_vite_hmr_client.py
@@ -0,0 +1,80 @@
+import pytest
+from bs4 import BeautifulSoup
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_hmr_client_returns_script_tag(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_hmr_client() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:5173/static/@vite/client"
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_hmr_client_kwargs(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_hmr_client(blocking="render") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag.has_attr("blocking")
+    assert script_tag["blocking"] == "render"
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_hmr_client_returns_nothing_with_dev_mode_false(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_hmr_client() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.parametrize(
+    "patch_settings",
+    [
+        {
+            "DJANGO_VITE_DEV_MODE": True,
+            "DJANGO_VITE_DEV_SERVER_PROTOCOL": "https",
+            "DJANGO_VITE_DEV_SERVER_HOST": "127.0.0.2",
+            "DJANGO_VITE_DEV_SERVER_PORT": "5174",
+            "DJANGO_VITE_STATIC_URL_PREFIX": "custom/prefix",
+            "DJANGO_VITE_WS_CLIENT_URL": "foo/bar",
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": True,
+                    "dev_server_protocol": "https",
+                    "dev_server_host": "127.0.0.2",
+                    "dev_server_port": "5174",
+                    "static_url_prefix": "custom/prefix",
+                    "ws_client_url": "foo/bar",
+                }
+            }
+        },
+    ],
+    indirect=True,
+)
+def test_vite_hmr_client_uses_correct_settings(jinja_env, patch_settings):
+    template = jinja_env.from_string(
+        """
+        {{ vite_hmr_client() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "https://127.0.0.2:5174/static/custom/prefix/foo/bar"
diff --git a/tests/tests/jinja/test_vite_legacy_asset.py b/tests/tests/jinja/test_vite_legacy_asset.py
new file mode 100644
index 0000000..3a19ad0
--- /dev/null
+++ b/tests/tests/jinja/test_vite_legacy_asset.py
@@ -0,0 +1,72 @@
+import pytest
+from bs4 import BeautifulSoup
+
+from django_vite.core.exceptions import DjangoViteAssetNotFoundError
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_legacy_asset_returns_nothing_with_dev_mode_on(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_legacy_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.fixture
+def patch_manifest_path(request, settings, patch_settings):
+    if request.param == "new_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE_DEV_MODE": False,
+                "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+                / "polyfills-manifest.json",
+            }
+        )
+    elif request.param == "legacy_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE": {
+                    "default": {
+                        "dev_mode": False,
+                        "manifest_path": settings.STATIC_ROOT
+                        / "polyfills-manifest.json",
+                    }
+                }
+            }
+        )
+
+
+@pytest.mark.parametrize(
+    "patch_manifest_path",
+    [
+        "new_settings",
+        "legacy_settings",
+    ],
+    indirect=True,
+)
+def test_vite_legacy_asset_returns_production_tags(jinja_env, patch_manifest_path):
+    template = jinja_env.from_string(
+        """
+        {{ vite_legacy_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-2e8a3a7a.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_legacy_asset_raises_nonexistent_entry(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_legacy_asset("src/fake.ts") }}
+    """
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template.render({})
diff --git a/tests/tests/jinja/test_vite_legacy_polyfills.py b/tests/tests/jinja/test_vite_legacy_polyfills.py
new file mode 100644
index 0000000..dc241be
--- /dev/null
+++ b/tests/tests/jinja/test_vite_legacy_polyfills.py
@@ -0,0 +1,156 @@
+import pytest
+from bs4 import BeautifulSoup
+
+from django_vite.core.exceptions import DjangoViteAssetNotFoundError
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_legacy_polyfills_returns_nothing_with_dev_mode_on(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_legacy_polyfills() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.fixture
+def patch_manifest_path(request, settings, patch_settings):
+    if request.param == "new_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE_DEV_MODE": False,
+                "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+                / "polyfills-manifest.json",
+            }
+        )
+    elif request.param == "legacy_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE": {
+                    "default": {
+                        "dev_mode": False,
+                        "manifest_path": settings.STATIC_ROOT
+                        / "polyfills-manifest.json",
+                    }
+                }
+            }
+        )
+
+
+@pytest.mark.parametrize(
+    "patch_manifest_path",
+    [
+        "new_settings",
+        "legacy_settings",
+    ],
+    indirect=True,
+)
+def test_vite_legacy_polyfills_production(jinja_env, patch_manifest_path):
+    template = jinja_env.from_string(
+        """
+        {{ vite_legacy_polyfills() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-f4c2b91e.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.fixture
+def patch_manifest_path_custom_motif(request, settings, patch_settings):
+    if request.param == "new_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE_DEV_MODE": False,
+                "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "custom-motif",
+                "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+                / "custom-motif-polyfills-manifest.json",
+            }
+        )
+    elif request.param == "legacy_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE": {
+                    "default": {
+                        "dev_mode": False,
+                        "legacy_polyfills_motif": "custom-motif",
+                        "manifest_path": settings.STATIC_ROOT
+                        / "custom-motif-polyfills-manifest.json",
+                    }
+                }
+            }
+        )
+
+
+@pytest.mark.parametrize(
+    "patch_manifest_path_custom_motif",
+    [
+        "new_settings",
+        "legacy_settings",
+    ],
+    indirect=True,
+)
+def test_vite_legacy_polyfills_custom_motif(
+    jinja_env, patch_manifest_path_custom_motif
+):
+    template = jinja_env.from_string(
+        """
+        {{ vite_legacy_polyfills() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-6e7a4b9c.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.fixture
+def patch_manifest_path_bad_motif(request, settings, patch_settings):
+    if request.param == "new_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE_DEV_MODE": False,
+                "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "fake-legacy-polyfills",
+                "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+                / "polyfills-manifest.json",
+            }
+        )
+    elif request.param == "legacy_settings":
+        return patch_settings(
+            {
+                "DJANGO_VITE": {
+                    "default": {
+                        "dev_mode": False,
+                        "legacy_polyfills_motif": "fake-legacy-polyfills",
+                        "manifest_path": settings.STATIC_ROOT
+                        / "polyfills-manifest.json",
+                    }
+                }
+            }
+        )
+
+
+@pytest.mark.parametrize(
+    "patch_manifest_path_bad_motif",
+    [
+        "new_settings",
+        "legacy_settings",
+    ],
+    indirect=True,
+)
+def test_vite_legacy_polyfills_nonexistent_motif(
+    jinja_env, patch_manifest_path_bad_motif
+):
+    template = jinja_env.from_string(
+        """
+            {{ vite_legacy_polyfills() }}
+    """
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template.render({})
diff --git a/tests/tests/jinja/test_vite_preload_asset.py b/tests/tests/jinja/test_vite_preload_asset.py
new file mode 100644
index 0000000..b9aff8b
--- /dev/null
+++ b/tests/tests/jinja/test_vite_preload_asset.py
@@ -0,0 +1,43 @@
+import pytest
+from bs4 import BeautifulSoup
+
+from django_vite.core.exceptions import DjangoViteAssetNotFoundError
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_preload_vite_asset_returns_nothing_with_dev_mode_on(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_preload_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_preload_vite_asset_returns_production_tags(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_preload_asset("src/entry.ts") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    links = soup.find_all("link")
+    assert len(links) == 14
+    first_link = links[0]
+    assert first_link["href"] == "assets/entry-29e38a60.js"
+    assert first_link["rel"] == ["modulepreload"]
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_preload_asset_raises_nonexistent_entry(jinja_env):
+    template = jinja_env.from_string(
+        """
+            {{ vite_preload_asset("src/fake.ts") }}
+    """
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template.render({})
diff --git a/tests/tests/jinja/test_vite_react_refresh.py b/tests/tests/jinja/test_vite_react_refresh.py
new file mode 100644
index 0000000..9fe9997
--- /dev/null
+++ b/tests/tests/jinja/test_vite_react_refresh.py
@@ -0,0 +1,134 @@
+import pytest
+from bs4 import BeautifulSoup
+
+
+@pytest.mark.usefixtures("dev_mode_true")
+def test_vite_react_refresh_returns_script_tag(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_react_refresh() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert not script_tag.has_attr("src")
+    assert script_tag.has_attr("type")
+    assert script_tag["type"] == "module"
+    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
+    assert "http://localhost:5173/static/@react-refresh" in script_tag.text
+
+
+@pytest.mark.parametrize(
+    "patch_settings",
+    [
+        {
+            "DJANGO_VITE_DEV_MODE": True,
+            "DJANGO_VITE_DEV_SERVER_PROTOCOL": "https",
+            "DJANGO_VITE_DEV_SERVER_HOST": "127.0.0.2",
+            "DJANGO_VITE_DEV_SERVER_PORT": "5174",
+            "DJANGO_VITE_STATIC_URL_PREFIX": "custom/prefix",
+            "DJANGO_VITE_REACT_REFRESH_URL": "foo/bar",
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": True,
+                    "dev_server_protocol": "https",
+                    "dev_server_host": "127.0.0.2",
+                    "dev_server_port": "5174",
+                    "static_url_prefix": "custom/prefix",
+                    "react_refresh_url": "foo/bar",
+                }
+            }
+        },
+    ],
+    indirect=True,
+)
+def test_vite_react_refresh_uses_correct_settings(jinja_env, patch_settings):
+    template = jinja_env.from_string(
+        """
+        {{ vite_react_refresh() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert not script_tag.has_attr("src")
+    assert script_tag.has_attr("type")
+    assert script_tag["type"] == "module"
+    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
+    assert "https://127.0.0.2:5174/static/custom/prefix/foo/bar" in script_tag.text
+
+
+@pytest.mark.usefixtures("dev_mode_false")
+def test_vite_react_refresh_returns_nothing_with_dev_mode_false(jinja_env):
+    template = jinja_env.from_string(
+        """
+        {{ vite_react_refresh() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.parametrize(
+    "patch_settings",
+    [
+        {
+            "DJANGO_VITE_DEV_MODE": True,
+            "DJANGO_VITE_REACT_REFRESH_URL": "foobar",
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": True,
+                    "react_refresh_url": "foobar",
+                }
+            }
+        },
+    ],
+    indirect=True,
+)
+def test_vite_react_refresh_url_setting(jinja_env, patch_settings):
+    template = jinja_env.from_string(
+        """
+        {{ vite_react_refresh() }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert "http://localhost:5173/static/foobar" in script_tag.text
+
+
+@pytest.mark.parametrize(
+    "patch_settings",
+    [
+        {
+            "DJANGO_VITE_DEV_MODE": True,
+            "DJANGO_VITE_REACT_REFRESH_URL": "foobar",
+        },
+        {
+            "DJANGO_VITE": {
+                "default": {
+                    "dev_mode": True,
+                    "react_refresh_url": "foobar",
+                }
+            }
+        },
+    ],
+    indirect=True,
+)
+def test_vite_react_refresh_uses_kwargs(jinja_env, patch_settings):
+    template = jinja_env.from_string(
+        """
+        {{ vite_react_refresh(nonce="woo-nonce") }}
+    """
+    )
+    html = template.render({})
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert script_tag.has_attr("nonce")
+    assert script_tag["nonce"] == "woo-nonce"
diff --git a/tests/tests/test_asset_loader.py b/tests/tests/test_asset_loader.py
index ca9c06d..dce6cff 100644
--- a/tests/tests/test_asset_loader.py
+++ b/tests/tests/test_asset_loader.py
@@ -1,8 +1,10 @@
 import pytest
 
-from django_vite.core.asset_loader import DjangoViteConfig, ManifestClient
-from django_vite.templatetags.django_vite import DjangoViteAssetLoader
 from django_vite.apps import check_loader_instance
+from django_vite.core.asset_loader import (
+    DjangoViteAssetLoader,
+    DjangoViteConfig,
+)
 
 
 def test_django_vite_asset_loader_cannot_be_instantiated():