add test for FrontendMedia

fluent_contents/tests/test_rendering.py

@@ -5,7 +5,8 @@
 from fluent_contents.extensions import PluginContext
 from fluent_contents.models import Placeholder, DEFAULT_TIMEOUT
 from fluent_contents.rendering import utils as rendering_utils
-from fluent_contents.tests.testapp.models import TestPage, RawHtmlTestItem, TimeoutTestItem, OverrideBase
+from fluent_contents.tests import factories
+from fluent_contents.tests.testapp.models import TestPage, RawHtmlTestItem, TimeoutTestItem, OverrideBase, MediaTestItem
 from fluent_contents.tests.utils import AppTestCase
 
 
@@ -39,6 +40,18 @@ def test_render_timeout(self):
         self.assertEqual(output.html, '<b>Item1!</b><b>Item2!</b>')
         self.assertEqual(output.cache_timeout, 60)  # this is that timeout that should be used for the placeholder cache item.
 
+    def test_render_media(self):
+        """
+        Test that 'class FrontendMedia' works.
+        """
+        placeholder = factories.create_placeholder()
+        factories.create_content_item(MediaTestItem, placeholder=placeholder, html='MEDIA_TEST')
+
+        output = rendering.render_placeholder(self.dummy_request, placeholder)
+        self.assertEqual(output.html.strip(), 'MEDIA_TEST')
+        self.assertEqual(output.media._js, ['testapp/media_item.js'])
+        self.assertEqual(output.media._css, {'screen': ['testapp/media_item.css']})
+
     def test_debug_is_method_overwritten(self):
         """
         Test the "is method overwritten" logic to detect template changes

fluent_contents/tests/testapp/content_plugins.py

@@ -1,6 +1,6 @@
 from django.utils.safestring import mark_safe
 from fluent_contents.extensions import ContentPlugin, plugin_pool
-from fluent_contents.tests.testapp.models import RawHtmlTestItem, TimeoutTestItem
+from fluent_contents.tests.testapp.models import RawHtmlTestItem, TimeoutTestItem, MediaTestItem
 
 
 @plugin_pool.register
@@ -28,3 +28,20 @@ def render(self, request, instance, **kwargs):
     def get_render_template(self, request, instance, **kwargs):
         # This is for test_debug_is_method_overwritten()
         return super(TimeoutTestPlugin, self).get_render_template(request, instance, **kwargs)
+
+
+@plugin_pool.register
+class MediaTestPlugin(ContentPlugin):
+    """
+    Testing a plugin timeout.
+    """
+    model = MediaTestItem
+    render_template = "testapp/media_item.html"
+
+    class FrontendMedia:
+        css = {
+            'screen': ('testapp/media_item.css',),
+        }
+        js = (
+            'testapp/media_item.js',
+        )

fluent_contents/tests/testapp/migrations/0001_initial.py

@@ -49,6 +49,19 @@ class Migration(migrations.Migration):
             },
             bases=('fluent_contents.contentitem',),
         ),
+        migrations.CreateModel(
+            name='MediaTestItem',
+            fields=[
+                ('contentitem_ptr', models.OneToOneField(parent_link=True, auto_created=True, primary_key=True, serialize=False, to='fluent_contents.ContentItem')),
+                ('html', models.TextField(verbose_name=b'HTML code')),
+            ],
+            options={
+                'db_table': 'contentitem_testapp_mediatestitem',
+                'verbose_name': 'Timeout test',
+                'verbose_name_plural': 'Timeout test',
+            },
+            bases=('fluent_contents.contentitem',),
+        ),
         migrations.CreateModel(
             name='TestPage',
             fields=[

fluent_contents/tests/testapp/models.py

@@ -76,3 +76,19 @@ class Meta:
 
     def __str__(self):
         return self.html
+
+
+@python_2_unicode_compatible
+class MediaTestItem(ContentItem):
+    """
+    The most basic "media HTML" content item, for testing.
+    """
+    html = models.TextField("HTML code")
+
+    class Meta:
+        app_label = 'testapp'
+        verbose_name = 'Media test'
+        verbose_name_plural = 'Media test'
+
+    def __str__(self):
+        return self.html

fluent_contents/tests/testapp/templates/testapp/media_item.html

@@ -0,0 +1 @@
+{{ instance.html|safe }}