django-components/tests/test_component_highlight.py

36 lines
1.4 KiB
Python

from django_components.util.component_highlight import apply_component_highlight, COLORS
from .django_test_setup import setup_test_config
from .testutils import BaseTestCase
setup_test_config({"autodiscover": False})
class ComponentHighlightTests(BaseTestCase):
def test_component_highlight(self):
# Test component highlighting
test_html = "<div>Test content</div>"
component_name = "TestComponent"
result = apply_component_highlight("component", test_html, component_name)
# Check that the output contains the component name
self.assertIn(component_name, result)
# Check that the output contains the original HTML
self.assertIn(test_html, result)
# Check that the component colors are used
self.assertIn(COLORS["component"].text_color, result)
self.assertIn(COLORS["component"].border_color, result)
def test_slot_highlight(self):
# Test slot highlighting
test_html = "<span>Slot content</span>"
slot_name = "content-slot"
result = apply_component_highlight("slot", test_html, slot_name)
# Check that the output contains the slot name
self.assertIn(slot_name, result)
# Check that the output contains the original HTML
self.assertIn(test_html, result)
# Check that the slot colors are used
self.assertIn(COLORS["slot"].text_color, result)
self.assertIn(COLORS["slot"].border_color, result)