async static files panel test

salty-ivy · salty-ivy · commit 182e90bf60c3 · 2024-08-19T20:55:02.000+05:30
diff --git a/tests/panels/test_staticfiles.py b/tests/panels/test_staticfiles.py
@@ -1,5 +1,7 @@
 from django.conf import settings
 from django.contrib.staticfiles import finders
+from django.shortcuts import render
+from django.test import AsyncRequestFactory
 
 from ..base import BaseTestCase
 
@@ -27,6 +29,17 @@ def test_default_case(self):
             self.panel.get_staticfiles_dirs(), finders.FileSystemFinder().locations
         )
 
+    async def test_store_staticfiles_with_async_context(self):
+        async def get_response(request):
+            # template contains one static file
+            return render(request, "staticfiles/async_static.html")
+
+        self._get_response = get_response
+        async_request = AsyncRequestFactory().get("/")
+        response = await self.panel.process_request(async_request)
+        self.panel.generate_stats(self.request, response)
+        self.assertNotEqual(self.panel.num_used, 0)
+
     def test_insert_content(self):
         """
         Test that the panel only inserts content after generate_stats and
diff --git a/tests/templates/base.html b/tests/templates/base.html
@@ -2,6 +2,7 @@
 <html>
     <head>
         <title>{{ title }}</title>
+        {% block head %}{% endblock %}
     </head>
     <body>
     {% block content %}{% endblock %}
diff --git a/tests/templates/staticfiles/async_static.html b/tests/templates/staticfiles/async_static.html
@@ -0,0 +1,6 @@
+{% extends "base.html" %}
+{% load static %}
+
+{% block head %}
+    <link rel="stylesheet" href="{% static 'additional_static/base.css' %}">
+{% endblock %}
