From 94154c76cdbe72b09a0302b65de5672c801feda1 Mon Sep 17 00:00:00 2001
From: Nick Ivons <14187277+Niicck@users.noreply.github.com>
Date: Mon, 16 Oct 2023 00:27:33 -0700
Subject: [PATCH] Add tests to achieve 100% Test Coverage (#91)

* create patch_settings fixture

* 100% test coverage

---------

Co-authored-by: Ben Beecher <120373+gone@users.noreply.github.com>
Co-authored-by: Eddy Brown <eddybrown@me.com>
Co-authored-by: Thijs Kramer <1274867+thijskramer@users.noreply.github.com>
Co-authored-by: Niicck <niicck@users.noreply.github.com>
---
 django_vite/apps.py                           |   5 +-
 django_vite/exceptions.py                     |  10 +
 django_vite/templatetags/django_vite.py       | 215 +++++++++++++++---
 pyproject.toml                                |   4 +
 setup.py                                      |   2 +-
 tests/conftest.py                             |  46 ++++
 .../custom-motif-polyfills-manifest.json      |  24 ++
 .../staticfiles/custom/prefix/manifest.json   |  72 ++++++
 tests/data/staticfiles/manifest.json          |  72 ++++++
 .../data/staticfiles/polyfills-manifest.json  |  24 ++
 tests/settings.py                             |  13 +-
 tests/test_asset_loader.py                    |   8 -
 tests/test_templatetags.py                    | 150 ------------
 tests/tests/templatetags/test_vite_asset.py   | 124 ++++++++++
 .../tests/templatetags/test_vite_asset_url.py |  43 ++++
 .../templatetags/test_vite_hmr_client.py      |  67 ++++++
 .../templatetags/test_vite_legacy_asset.py    |  49 ++++
 .../test_vite_legacy_polyfills.py             |  78 +++++++
 .../templatetags/test_vite_preload_asset.py   |  45 ++++
 .../templatetags/test_vite_react_refresh.py   |  74 ++++++
 tests/tests/test_asset_loader.py              |  33 +++
 tox.ini                                       |  21 ++
 22 files changed, 986 insertions(+), 193 deletions(-)
 create mode 100644 django_vite/exceptions.py
 create mode 100644 tests/conftest.py
 create mode 100644 tests/data/staticfiles/custom-motif-polyfills-manifest.json
 create mode 100644 tests/data/staticfiles/custom/prefix/manifest.json
 create mode 100644 tests/data/staticfiles/manifest.json
 create mode 100644 tests/data/staticfiles/polyfills-manifest.json
 delete mode 100644 tests/test_asset_loader.py
 delete mode 100644 tests/test_templatetags.py
 create mode 100644 tests/tests/templatetags/test_vite_asset.py
 create mode 100644 tests/tests/templatetags/test_vite_asset_url.py
 create mode 100644 tests/tests/templatetags/test_vite_hmr_client.py
 create mode 100644 tests/tests/templatetags/test_vite_legacy_asset.py
 create mode 100644 tests/tests/templatetags/test_vite_legacy_polyfills.py
 create mode 100644 tests/tests/templatetags/test_vite_preload_asset.py
 create mode 100644 tests/tests/templatetags/test_vite_react_refresh.py
 create mode 100644 tests/tests/test_asset_loader.py

diff --git a/django_vite/apps.py b/django_vite/apps.py
index 98dae96..29280ad 100644
--- a/django_vite/apps.py
+++ b/django_vite/apps.py
@@ -4,6 +4,7 @@
 from django.core.checks import Warning, register
 
 from .templatetags.django_vite import DjangoViteAssetLoader
+from .exceptions import DjangoViteManifestError
 
 
 class DjangoViteAppConfig(AppConfig):
@@ -11,7 +12,7 @@ class DjangoViteAppConfig(AppConfig):
     verbose_name = "Django Vite"
 
     def ready(self) -> None:
-        with suppress(RuntimeError):
+        with suppress(DjangoViteManifestError):
             # Create Loader instance at startup to prevent threading problems,
             # but do not crash while doing so.
             DjangoViteAssetLoader.instance()
@@ -25,7 +26,7 @@ def check_loader_instance(**kwargs):
         # Make Loader instance at startup to prevent threading problems
         DjangoViteAssetLoader.instance()
         return []
-    except RuntimeError as exception:
+    except DjangoViteManifestError as exception:
         return [
             Warning(
                 exception,
diff --git a/django_vite/exceptions.py b/django_vite/exceptions.py
new file mode 100644
index 0000000..1030ca8
--- /dev/null
+++ b/django_vite/exceptions.py
@@ -0,0 +1,10 @@
+class DjangoViteManifestError(RuntimeError):
+    """Manifest parsing failed."""
+
+    pass
+
+
+class DjangoViteAssetNotFoundError(RuntimeError):
+    """Vite Asset could not be found."""
+
+    pass
diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 4ff55de..52deacf 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -1,6 +1,6 @@
 import json
 from pathlib import Path
-from typing import Dict, List
+from typing import Dict, List, Callable
 from urllib.parse import urljoin
 
 from django import template
@@ -8,8 +8,9 @@
 from django.conf import settings
 from django.utils.safestring import mark_safe
 
-register = template.Library()
+from django_vite.exceptions import DjangoViteManifestError, DjangoViteAssetNotFoundError
 
+register = template.Library()
 
 # If using in development or production mode.
 DJANGO_VITE_DEV_MODE = getattr(settings, "DJANGO_VITE_DEV_MODE", False)
@@ -87,9 +88,10 @@ def generate_vite_asset(
         **kwargs: Dict[str, str],
     ) -> str:
         """
-        Generates a <script> tag for this JS/TS asset and a <link> tag for
-        all of its CSS dependencies by reading the manifest
-        file (for production only).
+        Generates a <script> tag for this JS/TS asset, a <link> tag for
+        all of its CSS dependencies, and a <link modulepreload>
+        for the js dependencies, as listed in the manifest file
+        (for production only).
         In development Vite loads all by itself.
 
         Arguments:
@@ -103,7 +105,7 @@ def generate_vite_asset(
                 script tags.
 
         Raises:
-            RuntimeError: If cannot find the file path in the
+            DjangoViteAssetNotFoundError: If cannot find the file path in the
                 manifest (only in production).
 
         Returns:
@@ -118,7 +120,7 @@ def generate_vite_asset(
             )
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -128,7 +130,7 @@ def generate_vite_asset(
         scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
 
         # Add dependent CSS
-        tags.extend(self._generate_css_files_of_asset(path, []))
+        tags.extend(self._load_css_files_of_asset(path, []))
 
         # Add the script by itself
         tags.append(
@@ -140,10 +142,118 @@ def generate_vite_asset(
             )
         )
 
+        # Preload imports
+        preload_attrs = {
+            "type": "text/javascript",
+            "crossorigin": "anonymous",
+            "rel": "modulepreload",
+            "as": "script",
+        }
+
+        for dep in manifest_entry.get("imports", []):
+            dep_manifest_entry = self._manifest[dep]
+            dep_file = dep_manifest_entry["file"]
+            url = DjangoViteAssetLoader._generate_production_server_url(dep_file)
+            tags.append(
+                DjangoViteAssetLoader._generate_preload_tag(
+                    url,
+                    attrs=preload_attrs,
+                )
+            )
+
         return "\n".join(tags)
 
-    def _generate_css_files_of_asset(
+    def preload_vite_asset(
+        self,
+        path: str,
+    ) -> str:
+        """
+        Generates a <link modulepreload> tag for this JS/TS asset, a
+        <link preload> tag for all of its CSS dependencies,
+        and a <link modulepreload> for the js dependencies.
+        In development this template tag renders nothing,
+        since files aren't compiled yet"
+
+        Arguments:
+            path {str} -- Path to a Vite JS/TS asset to preload.
+
+        Returns:
+            str -- All tags to preload this file in your HTML page.
+
+        Raises:
+            DjangoViteAssetNotFoundError: if cannot find the file path in the
+                manifest.
+
+        Returns:
+            str -- all <link> tags to preload
+                this asset.
+        """
+        if DJANGO_VITE_DEV_MODE:
+            return ""
+
+        if not self._manifest or path not in self._manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {DJANGO_VITE_MANIFEST_PATH}"
+            )
+
+        tags = []
+        manifest_entry = self._manifest[path]
+
+        # Add the script by itself
+        script_attrs = {
+            "type": "text/javascript",
+            "crossorigin": "anonymous",
+            "rel": "modulepreload",
+            "as": "script",
+        }
+
+        manifest_file = manifest_entry["file"]
+        url = DjangoViteAssetLoader._generate_production_server_url(manifest_file)
+        tags.append(
+            DjangoViteAssetLoader._generate_preload_tag(
+                url,
+                attrs=script_attrs,
+            )
+        )
+
+        # Add dependent CSS
+        tags.extend(self._preload_css_files_of_asset(path, []))
+
+        # Preload imports
+        for dep in manifest_entry.get("imports", []):
+            dep_manifest_entry = self._manifest[dep]
+            dep_file = dep_manifest_entry["file"]
+            url = DjangoViteAssetLoader._generate_production_server_url(dep_file)
+            tags.append(
+                DjangoViteAssetLoader._generate_preload_tag(
+                    url,
+                    attrs=script_attrs,
+                )
+            )
+
+        return "\n".join(tags)
+
+    def _preload_css_files_of_asset(
+        self, path: str, already_processed: List[str]
+    ) -> List[str]:
+        return self._generate_css_files_of_asset(
+            path,
+            already_processed,
+            DjangoViteAssetLoader._generate_stylesheet_preload_tag,
+        )
+
+    def _load_css_files_of_asset(
         self, path: str, already_processed: List[str]
+    ) -> List[str]:
+        return self._generate_css_files_of_asset(
+            path,
+            already_processed,
+            DjangoViteAssetLoader._generate_stylesheet_tag,
+        )
+
+    def _generate_css_files_of_asset(
+        self, path: str, already_processed: List[str], tag_generator: Callable
     ) -> List[str]:
         """
         Generates all CSS tags for dependencies of an asset.
@@ -162,7 +272,9 @@ def _generate_css_files_of_asset(
         if "imports" in manifest_entry:
             for import_path in manifest_entry["imports"]:
                 tags.extend(
-                    self._generate_css_files_of_asset(import_path, already_processed)
+                    self._generate_css_files_of_asset(
+                        import_path, already_processed, tag_generator
+                    )
                 )
 
         if "css" in manifest_entry:
@@ -171,7 +283,7 @@ def _generate_css_files_of_asset(
                     url = DjangoViteAssetLoader._generate_production_server_url(
                         css_path
                     )
-                    tags.append(DjangoViteAssetLoader._generate_stylesheet_tag(url))
+                    tags.append(tag_generator(url))
 
                 already_processed.append(css_path)
 
@@ -186,7 +298,7 @@ def generate_vite_asset_url(self, path: str) -> str:
             path {str} -- Path to a Vite asset.
 
         Raises:
-            RuntimeError: If cannot find the asset path in the
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
                 manifest (only in production).
 
         Returns:
@@ -197,7 +309,7 @@ def generate_vite_asset_url(self, path: str) -> str:
             return DjangoViteAssetLoader._generate_vite_server_url(path)
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -221,7 +333,7 @@ def generate_vite_legacy_polyfills(
                 script tags.
 
         Raises:
-            RuntimeError: If polyfills path not found inside
+            DjangoViteAssetNotFoundError: If polyfills path not found inside
                 the 'manifest.json' (only in production).
 
         Returns:
@@ -242,7 +354,7 @@ def generate_vite_legacy_polyfills(
                     attrs=scripts_attrs,
                 )
 
-        raise RuntimeError(
+        raise DjangoViteAssetNotFoundError(
             f"Vite legacy polyfills not found in manifest "
             f"at {DJANGO_VITE_MANIFEST_PATH}"
         )
@@ -266,7 +378,7 @@ def generate_vite_legacy_asset(
                 script tags.
 
         Raises:
-            RuntimeError: If cannot find the asset path in the
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
                 manifest (only in production).
 
         Returns:
@@ -277,7 +389,7 @@ def generate_vite_legacy_asset(
             return ""
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -297,7 +409,8 @@ def _parse_manifest(self) -> None:
         Read and parse the Vite manifest file.
 
         Raises:
-            RuntimeError: if cannot load the file or JSON in file is malformed.
+            DjangoViteManifestError: if cannot load the file or JSON in file is
+            malformed.
         """
 
         try:
@@ -305,7 +418,7 @@ def _parse_manifest(self) -> None:
                 manifest_content = manifest_file.read()
             self._manifest = json.loads(manifest_content)
         except Exception as error:
-            raise RuntimeError(
+            raise DjangoViteManifestError(
                 f"Cannot read Vite manifest file at "
                 f"{DJANGO_VITE_MANIFEST_PATH} : {str(error)}"
             ) from error
@@ -377,7 +490,7 @@ def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
     @staticmethod
     def _generate_stylesheet_tag(href: str) -> str:
         """
-        Generates and HTML <link> stylesheet tag for CSS.
+        Generates an HTML <link> stylesheet tag for CSS.
 
         Arguments:
             href {str} -- CSS file URL.
@@ -388,6 +501,26 @@ def _generate_stylesheet_tag(href: str) -> str:
 
         return f'<link rel="stylesheet" href="{href}" />'
 
+    @staticmethod
+    def _generate_stylesheet_preload_tag(href: str) -> str:
+        """
+        Generates an HTML <link> preload tag for CSS.
+
+        Arguments:
+            href {str} -- CSS file URL.
+
+        Returns:
+            str -- CSS link tag.
+        """
+
+        return f'<link rel="preload" href="{href}" as="style" />'
+
+    @staticmethod
+    def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
+
+        return f'<link href="{href}" {attrs_str} />'
+
     @staticmethod
     def _generate_vite_server_url(path: str) -> str:
         """
@@ -441,10 +574,11 @@ def _generate_production_server_url(path: str) -> str:
             str -- Full URL to the asset.
         """
 
-        prefix = DJANGO_VITE_STATIC_URL_PREFIX
-        if not DJANGO_VITE_STATIC_URL_PREFIX.endswith("/"):
-            prefix += "/"
-        production_server_url = urljoin(prefix, path)
+        production_server_url = path
+        if prefix := DJANGO_VITE_STATIC_URL_PREFIX:
+            if not DJANGO_VITE_STATIC_URL_PREFIX.endswith("/"):
+                prefix += "/"
+            production_server_url = urljoin(prefix, path)
 
         if apps.is_installed("django.contrib.staticfiles"):
             from django.contrib.staticfiles.storage import staticfiles_storage
@@ -480,9 +614,9 @@ def vite_asset(
     **kwargs: Dict[str, str],
 ) -> str:
     """
-    Generates a <script> tag for this JS/TS asset and a <link> tag for
-    all of its CSS dependencies by reading the manifest
-    file (for production only).
+    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:
@@ -509,6 +643,33 @@ def vite_asset(
     return DjangoViteAssetLoader.instance().generate_vite_asset(path, **kwargs)
 
 
+@register.simple_tag
+@mark_safe
+def vite_preload_asset(
+    path: str,
+) -> 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)
+
+
 @register.simple_tag
 def vite_asset_url(path: str) -> str:
     """
diff --git a/pyproject.toml b/pyproject.toml
index 0614b47..c332ae0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,8 +7,12 @@ addopts = '''
   --cov-report html
   --cov-report term-missing
   --cov-branch
+  --cov-fail-under=100
 '''
 
+[tool.black]
+line-length = 88
+
 [tool.ruff]
 select = [
   "E",   # pycodestyle
diff --git a/setup.py b/setup.py
index 35e7831..f2e4533 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="django-vite",
-    version="2.1.2",
+    version="2.1.3",
     description="Integration of Vite in a Django project.",
     long_description=README,
     long_description_content_type="text/markdown",
diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..2eedec1
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,46 @@
+from typing import Dict, Any
+import pytest
+from importlib import reload
+from django.apps import apps
+from django_vite.templatetags import django_vite
+
+
+def reload_django_vite():
+    reload(django_vite)
+    django_vite_app_config = apps.get_app_config("django_vite")
+    django_vite_app_config.ready()
+
+
+@pytest.fixture()
+def patch_settings(settings):
+    """
+    1. Patch new settings into django.conf.settings.
+    2. Reload django_vite module so that variables on the module level that use settings
+    get recalculated.
+    3. Restore the original settings once the test is over.
+
+    TODO: refactor django_vite so that we don't set variables on the module level using
+    settings.
+    """
+    __PYTEST_EMPTY__ = "__PYTEST_EMPTY__"
+    original_settings_cache = {}
+
+    def _patch_settings(new_settings: Dict[str, Any]):
+        for key, value in new_settings.items():
+            original_settings_cache[key] = getattr(settings, key, __PYTEST_EMPTY__)
+            setattr(settings, key, value)
+        reload_django_vite()
+
+    yield _patch_settings
+
+    for key, value in original_settings_cache.items():
+        if value == __PYTEST_EMPTY__:
+            delattr(settings, key)
+        else:
+            setattr(settings, key, value)
+    reload_django_vite()
+
+
+@pytest.fixture()
+def dev_mode_off(patch_settings):
+    patch_settings({"DJANGO_VITE_DEV_MODE": False})
diff --git a/tests/data/staticfiles/custom-motif-polyfills-manifest.json b/tests/data/staticfiles/custom-motif-polyfills-manifest.json
new file mode 100644
index 0000000..c954ade
--- /dev/null
+++ b/tests/data/staticfiles/custom-motif-polyfills-manifest.json
@@ -0,0 +1,24 @@
+{
+    "../../vite/custom-motif-legacy": {
+      "file": "assets/polyfills-legacy-6e7a4b9c.js",
+      "isEntry": true,
+      "src": "../../vite/custom-motif-legacy"
+    },
+    "src/entry-legacy.ts": {
+      "file": "assets/entry-legacy-4c50596f.js",
+      "isEntry": true,
+      "src": "src/entry-legacy.ts"
+    },
+    "src/entry.css": {
+      "file": "assets/entry-5e7d9c21.css",
+      "src": "src/entry.css"
+    },
+    "src/entry.ts": {
+      "css": [
+        "assets/entry-5e7d9c21.css"
+      ],
+      "file": "assets/entry-8a2f6b3d.js",
+      "isEntry": true,
+      "src": "src/entry.ts"
+    }
+}
diff --git a/tests/data/staticfiles/custom/prefix/manifest.json b/tests/data/staticfiles/custom/prefix/manifest.json
new file mode 100644
index 0000000..fbe4a74
--- /dev/null
+++ b/tests/data/staticfiles/custom/prefix/manifest.json
@@ -0,0 +1,72 @@
+{
+  "src/entry.ts": {
+    "css": ["assets/entry-74d0d5dd.css"],
+    "file": "assets/entry-29e38a60.js",
+    "imports": [
+      "_vue.esm-bundler-96356fb1.js",
+      "_index-62f37ad0.js",
+      "_vue.esm-bundler-49e6b475.js",
+      "__plugin-vue_export-helper-c27b6911.js",
+      "_messages-a0a9e13b.js",
+      "_use-outside-click-224980bf.js",
+      "_use-resolve-button-type-c5656cba.js",
+      "_use-event-listener-153dc639.js",
+      "_hidden-84ddb9a5.js",
+      "_apiClient-01d20438.js",
+      "_pinia-5d7892fd.js"
+    ],
+    "isEntry": true,
+    "src": "entry.ts"
+  },
+  "src/entry.css": {
+    "file": "assets/entry-74d0d5dd.css",
+    "src": "entry.css"
+  },
+  "src/extra.css": {
+    "file": "assets/extra-a9f3b2c1.css",
+    "src": "extra.css"
+  },
+  "_vue.esm-bundler-96356fb1.js": {
+    "file": "vue.esm-bundler-96356fb1.js"
+  },
+  "_index-62f37ad0.js": {
+    "file": "index-62f37ad0.js",
+    "imports": ["_vue.esm-bundler-49e6b475.js"]
+  },
+  "_vue.esm-bundler-49e6b475.js": {
+    "file": "vue.esm-bundler-49e6b475.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "__plugin-vue_export-helper-c27b6911.js": {
+    "file": "_plugin-vue_export-helper-c27b6911.js"
+  },
+  "_messages-a0a9e13b.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "messages-a0a9e13b.js",
+    "imports": ["_pinia-5d7892fd.js", "_vue.esm-bundler-96356fb1.js"]
+  },
+  "_pinia-5d7892fd.js": {
+    "file": "pinia-5d7892fd.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-outside-click-224980bf.js": {
+    "file": "use-outside-click-224980bf.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-resolve-button-type-c5656cba.js": {
+    "file": "use-resolve-button-type-c5656cba.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_use-event-listener-153dc639.js": {
+    "file": "use-event-listener-153dc639.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_hidden-84ddb9a5.js": {
+    "file": "hidden-84ddb9a5.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_apiClient-01d20438.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "apiClient-01d20438.js"
+  }
+}
diff --git a/tests/data/staticfiles/manifest.json b/tests/data/staticfiles/manifest.json
new file mode 100644
index 0000000..fbe4a74
--- /dev/null
+++ b/tests/data/staticfiles/manifest.json
@@ -0,0 +1,72 @@
+{
+  "src/entry.ts": {
+    "css": ["assets/entry-74d0d5dd.css"],
+    "file": "assets/entry-29e38a60.js",
+    "imports": [
+      "_vue.esm-bundler-96356fb1.js",
+      "_index-62f37ad0.js",
+      "_vue.esm-bundler-49e6b475.js",
+      "__plugin-vue_export-helper-c27b6911.js",
+      "_messages-a0a9e13b.js",
+      "_use-outside-click-224980bf.js",
+      "_use-resolve-button-type-c5656cba.js",
+      "_use-event-listener-153dc639.js",
+      "_hidden-84ddb9a5.js",
+      "_apiClient-01d20438.js",
+      "_pinia-5d7892fd.js"
+    ],
+    "isEntry": true,
+    "src": "entry.ts"
+  },
+  "src/entry.css": {
+    "file": "assets/entry-74d0d5dd.css",
+    "src": "entry.css"
+  },
+  "src/extra.css": {
+    "file": "assets/extra-a9f3b2c1.css",
+    "src": "extra.css"
+  },
+  "_vue.esm-bundler-96356fb1.js": {
+    "file": "vue.esm-bundler-96356fb1.js"
+  },
+  "_index-62f37ad0.js": {
+    "file": "index-62f37ad0.js",
+    "imports": ["_vue.esm-bundler-49e6b475.js"]
+  },
+  "_vue.esm-bundler-49e6b475.js": {
+    "file": "vue.esm-bundler-49e6b475.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "__plugin-vue_export-helper-c27b6911.js": {
+    "file": "_plugin-vue_export-helper-c27b6911.js"
+  },
+  "_messages-a0a9e13b.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "messages-a0a9e13b.js",
+    "imports": ["_pinia-5d7892fd.js", "_vue.esm-bundler-96356fb1.js"]
+  },
+  "_pinia-5d7892fd.js": {
+    "file": "pinia-5d7892fd.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-outside-click-224980bf.js": {
+    "file": "use-outside-click-224980bf.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-resolve-button-type-c5656cba.js": {
+    "file": "use-resolve-button-type-c5656cba.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_use-event-listener-153dc639.js": {
+    "file": "use-event-listener-153dc639.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_hidden-84ddb9a5.js": {
+    "file": "hidden-84ddb9a5.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_apiClient-01d20438.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "apiClient-01d20438.js"
+  }
+}
diff --git a/tests/data/staticfiles/polyfills-manifest.json b/tests/data/staticfiles/polyfills-manifest.json
new file mode 100644
index 0000000..ddebd6a
--- /dev/null
+++ b/tests/data/staticfiles/polyfills-manifest.json
@@ -0,0 +1,24 @@
+{
+    "../../vite/legacy-polyfills-legacy": {
+      "file": "assets/polyfills-legacy-f4c2b91e.js",
+      "isEntry": true,
+      "src": "../../vite/legacy-polyfills-legacy"
+    },
+    "src/entry-legacy.ts": {
+      "file": "assets/entry-legacy-32083566.js",
+      "isEntry": true,
+      "src": "src/entry-legacy.ts"
+    },
+    "src/entry.css": {
+      "file": "assets/entry-74d0d5dd.css",
+      "src": "src/entry.css"
+    },
+    "src/entry.ts": {
+      "css": [
+        "assets/entry-74d0d5dd.css"
+      ],
+      "file": "assets/entry-2e8a3a7a.js",
+      "isEntry": true,
+      "src": "src/entry.ts"
+    }
+}
diff --git a/tests/settings.py b/tests/settings.py
index 189a195..541df85 100644
--- a/tests/settings.py
+++ b/tests/settings.py
@@ -1,13 +1,11 @@
 import os
+from pathlib import Path
 
-BASE_DIR = os.path.dirname(__file__)
+BASE_DIR = Path(__file__).resolve().parent
 
-STATIC_URL = "static"
-DJANGO_VITE_DEV_MODE = True
-DJANGO_VITE_ASSETS_PATH = "/"
+STATIC_URL = "/static/"
 USE_TZ = True
 
-
 INSTALLED_APPS = [
     "django_vite",
 ]
@@ -26,3 +24,8 @@
         },
     },
 ]
+
+STATIC_ROOT = BASE_DIR / "data" / "staticfiles"
+
+DJANGO_VITE_DEV_MODE = True
+DJANGO_VITE_ASSETS_PATH = "/"
diff --git a/tests/test_asset_loader.py b/tests/test_asset_loader.py
deleted file mode 100644
index 7ee8935..0000000
--- a/tests/test_asset_loader.py
+++ /dev/null
@@ -1,8 +0,0 @@
-import pytest
-
-from django_vite.templatetags.django_vite import DjangoViteAssetLoader
-
-
-def test_django_vite_asset_loader_cannot_be_instantiated():
-    with pytest.raises(RuntimeError):
-        DjangoViteAssetLoader()
diff --git a/tests/test_templatetags.py b/tests/test_templatetags.py
deleted file mode 100644
index 79bb5c3..0000000
--- a/tests/test_templatetags.py
+++ /dev/null
@@ -1,150 +0,0 @@
-import pytest
-from bs4 import BeautifulSoup
-from django.template import Context, Template, TemplateSyntaxError
-
-
-@pytest.fixture()
-def override_setting(monkeypatch):
-    def _override_setting(setting, value):
-        monkeypatch.setattr(
-            f"django_vite.templatetags.django_vite.{setting}",
-            value,
-        )
-
-    return _override_setting
-
-
-def test_vite_hmr_client_returns_script_tag():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag["src"] == "http://localhost:3000/static/@vite/client"
-    assert script_tag["type"] == "module"
-
-
-def test_vite_hmr_client_kwargs():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client blocking="render" %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag.has_attr("blocking")
-    assert script_tag["blocking"] == "render"
-
-
-def test_vite_hmr_client_returns_nothing_with_dev_mode_off(settings, monkeypatch):
-    settings.DJANGO_VITE_DEV_MODE = False
-    monkeypatch.setattr(
-        "django_vite.templatetags.django_vite.DJANGO_VITE_DEV_MODE",
-        settings.DJANGO_VITE_DEV_MODE,
-    )
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    assert str(soup).strip() == ""
-
-
-def test_vite_hmr_client_uses_correct_settings(override_setting):
-    override_setting("DJANGO_VITE_DEV_SERVER_PROTOCOL", "https")
-    override_setting("DJANGO_VITE_DEV_SERVER_HOST", "127.0.0.2")
-    override_setting("DJANGO_VITE_DEV_SERVER_PORT", "5174")
-    override_setting("DJANGO_VITE_STATIC_URL", "static/")
-    override_setting("DJANGO_VITE_WS_CLIENT_URL", "foo/bar")
-
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag["src"] == "https://127.0.0.2:5174/static/foo/bar"
-
-
-def test_vite_asset_returns_script_tags():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_asset "src/entry.tsx" %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag["src"] == "http://localhost:3000/static/src/entry.tsx"
-    assert script_tag["type"] == "module"
-
-
-def test_vite_asset_raises_without_path():
-    with pytest.raises(TemplateSyntaxError):
-        Template(
-            """
-            {% load django_vite %}
-            {% vite_asset %}
-        """
-        )
-
-
-def test_vite_react_refresh_happy_flow():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_react_refresh %}
-    """
-    )
-    html = template.render(Context({}))
-    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:3000/static/@react-refresh" in script_tag.text
-
-
-def test_vite_react_refresh_returns_nothing_with_dev_mode_off(settings, monkeypatch):
-    settings.DJANGO_VITE_DEV_MODE = False
-    monkeypatch.setattr(
-        "django_vite.templatetags.django_vite.DJANGO_VITE_DEV_MODE",
-        settings.DJANGO_VITE_DEV_MODE,
-    )
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_react_refresh %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    assert str(soup).strip() == ""
-
-
-def test_vite_react_refresh_url_setting(override_setting):
-    override_setting("DJANGO_VITE_REACT_REFRESH_URL", "foobar")
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_react_refresh %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.script
-    assert "http://localhost:3000/static/foobar" in script_tag.text
diff --git a/tests/tests/templatetags/test_vite_asset.py b/tests/tests/templatetags/test_vite_asset.py
new file mode 100644
index 0000000..7923954
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_asset.py
@@ -0,0 +1,124 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template, TemplateSyntaxError
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_asset_returns_dev_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.ts"
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_returns_production_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    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
+
+
+def test_vite_asset_raises_without_path():
+    with pytest.raises(TemplateSyntaxError):
+        Template(
+            """
+            {% load django_vite %}
+            {% vite_asset %}
+        """
+        )
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
+
+
+@pytest.mark.parametrize("prefix", ["custom/prefix", "custom/prefix/"])
+def test_vite_asset_dev_prefix(prefix, patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_STATIC_URL_PREFIX": prefix,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert (
+        script_tag["src"] == "http://localhost:3000/static/custom/prefix/src/entry.ts"
+    )
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+@pytest.mark.parametrize("prefix", ["custom/prefix", "custom/prefix/"])
+def test_vite_asset_production_prefix(prefix, patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_STATIC_URL_PREFIX": prefix,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    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_off")
+def test_vite_asset_production_staticfiles_storage(patch_settings):
+    patch_settings(
+        {
+            "INSTALLED_APPS": ["django_vite", "django.contrib.staticfiles"],
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    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
diff --git a/tests/tests/templatetags/test_vite_asset_url.py b/tests/tests/templatetags/test_vite_asset_url.py
new file mode 100644
index 0000000..e0eef07
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_asset_url.py
@@ -0,0 +1,43 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_asset_url_returns_dev_url():
+    template = Template(
+        """
+        {% load django_vite %}
+        <script src="{% vite_asset_url "src/entry.ts" %}">
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.ts"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_url_returns_production_url():
+    template = Template(
+        """
+        {% load django_vite %}
+        <script src="{% vite_asset_url "src/entry.ts" %}">
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_url_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            <script src="{% vite_asset_url "src/fake.ts" %}">
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_hmr_client.py b/tests/tests/templatetags/test_vite_hmr_client.py
new file mode 100644
index 0000000..2fee91b
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_hmr_client.py
@@ -0,0 +1,67 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+
+
+def test_vite_hmr_client_returns_script_tag():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/@vite/client"
+    assert script_tag["type"] == "module"
+
+
+def test_vite_hmr_client_kwargs():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client blocking="render" %}
+    """
+    )
+    html = template.render(Context({}))
+    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_off")
+def test_vite_hmr_client_returns_nothing_with_dev_mode_off():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+def test_vite_hmr_client_uses_correct_settings(patch_settings):
+    patch_settings(
+        {
+            "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": "static/",
+            "DJANGO_VITE_WS_CLIENT_URL": "foo/bar",
+        }
+    )
+
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "https://127.0.0.2:5174/static/foo/bar"
diff --git a/tests/tests/templatetags/test_vite_legacy_asset.py b/tests/tests/templatetags/test_vite_legacy_asset.py
new file mode 100644
index 0000000..4c2111a
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_legacy_asset.py
@@ -0,0 +1,49 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_legacy_asset_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_asset_returns_production_tags(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    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_off")
+def test_vite_legacy_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_legacy_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_legacy_polyfills.py b/tests/tests/templatetags/test_vite_legacy_polyfills.py
new file mode 100644
index 0000000..409c032
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_legacy_polyfills.py
@@ -0,0 +1,78 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_legacy_polyfills_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_production(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-f4c2b91e.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_custom_motif(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "custom-motif",
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "custom-motif-polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-6e7a4b9c.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_nonexistent_motif(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+            "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "fake-legacy-polyfills",
+        }
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_legacy_polyfills %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_preload_asset.py b/tests/tests/templatetags/test_vite_preload_asset.py
new file mode 100644
index 0000000..e8ec44a
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_preload_asset.py
@@ -0,0 +1,45 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_preload_vite_asset_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_preload_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_preload_vite_asset_returns_production_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_preload_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    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_off")
+def test_vite_preload_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_preload_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_react_refresh.py b/tests/tests/templatetags/test_vite_react_refresh.py
new file mode 100644
index 0000000..e2c31e7
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_react_refresh.py
@@ -0,0 +1,74 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+
+
+def test_vite_react_refresh_returns_script_tag():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    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:3000/static/@react-refresh" in script_tag.text
+
+
+def test_vite_react_refresh_uses_correct_settings(patch_settings):
+    patch_settings(
+        {
+            "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": "static/",
+            "DJANGO_VITE_REACT_REFRESH_URL": "foo/bar",
+        }
+    )
+
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    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/foo/bar" in script_tag.text
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_react_refresh_returns_nothing_with_dev_mode_off():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+def test_vite_react_refresh_url_setting(patch_settings):
+    patch_settings({"DJANGO_VITE_REACT_REFRESH_URL": "foobar"})
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert "http://localhost:3000/static/foobar" in script_tag.text
diff --git a/tests/tests/test_asset_loader.py b/tests/tests/test_asset_loader.py
new file mode 100644
index 0000000..470867c
--- /dev/null
+++ b/tests/tests/test_asset_loader.py
@@ -0,0 +1,33 @@
+import pytest
+
+from django_vite.templatetags.django_vite import DjangoViteAssetLoader
+from django_vite.apps import check_loader_instance
+
+
+def test_django_vite_asset_loader_cannot_be_instantiated():
+    with pytest.raises(RuntimeError):
+        DjangoViteAssetLoader()
+
+
+def test_check_loader_instance_happy(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+        }
+    )
+    DjangoViteAssetLoader._instance = None
+    warnings = check_loader_instance()
+    assert len(warnings) == 0
+
+
+def test_check_loader_instance_warnings(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_MANIFEST_PATH": "fake.json",
+        }
+    )
+    DjangoViteAssetLoader._instance = None
+    warnings = check_loader_instance()
+    assert len(warnings) == 1
+    assert "Make sure you have generated a manifest file" in warnings[0].hint
diff --git a/tox.ini b/tox.ini
index 193fa9c..fb651ac 100644
--- a/tox.ini
+++ b/tox.ini
@@ -32,6 +32,27 @@ commands =
 ignore_outcome =
     django-latest: True
 
+[testenv:debugpy]
+description = run unit tests with debugpy
+basepython = python3.8
+deps =
+    beautifulsoup4
+    pytest>=7
+    debugpy
+    pytest-django
+    pytest-sugar
+    Django>=3.2,<4.0
+setenv =
+    DEBUGPY_PORT_PYTEST = 56789
+commands =
+    python \
+    -m debugpy --listen localhost:{env:DEBUGPY_PORT_PYTEST} --wait-for-client \
+    -m pytest \
+    -o addopts="" \
+    {posargs:tests}
+ignore_outcome =
+    django-latest: True
+
 [testenv:codestyle]
 basepython = python3
 commands =