From d91f014b4b3448ae08532c444b5da59da2205032 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Emil=20Stenstr=C3=B6m?= <em@kth.se>
Date: Sun, 24 Nov 2019 11:09:40 +0100
Subject: [PATCH] Move test components into test cases.

---
 tests/test_component.py | 37 ++++++++++++++++++-------------------
 1 file changed, 18 insertions(+), 19 deletions(-)

diff --git a/tests/test_component.py b/tests/test_component.py
index 5c23f4ab..81d9820b 100644
--- a/tests/test_component.py
+++ b/tests/test_component.py
@@ -6,27 +6,21 @@ from django_components import component
 
 from .django_test_setup import *  # NOQA
 
-
-class SimpleComponent(component.Component):
-    def context(self, variable=None):
-        return {
-            "variable": variable,
-        }
-
-    def template(self, context):
-        return "simple_template.html"
-
-    class Media:
-        css = {"all": ["style.css"]}
-        js = ["script.js"]
-
-class MultistyleComponent(SimpleComponent):
-    class Media:
-        css = {"all": ["style.css", "style2.css"]}
-        js = ["script.js", "script2.js"]
-
 class ComponentRegistryTest(SimpleTestCase):
     def test_simple_component(self):
+        class SimpleComponent(component.Component):
+            def context(self, variable=None):
+                return {
+                    "variable": variable,
+                }
+
+            def template(self, context):
+                return "simple_template.html"
+
+            class Media:
+                css = {"all": ["style.css"]}
+                js = ["script.js"]
+
         comp = SimpleComponent()
 
         self.assertHTMLEqual(comp.render_dependencies(), dedent("""
@@ -39,6 +33,11 @@ class ComponentRegistryTest(SimpleTestCase):
         """).lstrip())
 
     def test_component_with_list_of_styles(self):
+        class MultistyleComponent(component.Component):
+            class Media:
+                css = {"all": ["style.css", "style2.css"]}
+                js = ["script.js", "script2.js"]
+
         comp = MultistyleComponent()
 
         self.assertHTMLEqual(comp.render_dependencies(), dedent("""