Merge remote-tracking branch 'origin'

adamghill · adamghill · commit db90c03ffd4e · 2022-11-02T22:10:59.000-04:00
diff --git a/django_unicorn/components/unicorn_template_response.py b/django_unicorn/components/unicorn_template_response.py
@@ -209,7 +209,9 @@ def get_root_element(soup: BeautifulSoup) -> Tag:
     for element in soup.contents:
         if isinstance(element, Tag) and element.name:
             if element.name == "html":
-                view_element = element.find_next(attrs={"unicorn:view": True})
+                view_element = element.find_next(
+                    attrs={"unicorn:view": True}
+                ) or element.find_next(attrs={"u:view": True})
 
                 if not view_element:
                     raise MissingComponentViewElement(
diff --git a/tests/components/test_unicorn_template_response.py b/tests/components/test_unicorn_template_response.py
@@ -50,7 +50,7 @@ def test_get_root_element_no_element():
         assert str(actual) == expected
 
 
-def test_get_root_element_direct_view():
+def test_get_root_element_direct_view_unicorn():
     # Annoyingly beautifulsoup adds a blank string on the attribute
     expected = '<div unicorn:view="">test</div>'
 
@@ -63,6 +63,17 @@ def test_get_root_element_direct_view():
     assert str(actual) == expected
 
 
+def test_get_root_element_direct_view_u():
+    # Annoyingly beautifulsoup adds a blank string on the attribute
+    expected = '<div u:view="">test</div>'
+
+    component_html = "<html><head></head>≤body><div u:view>test</div></body></html>"
+    soup = BeautifulSoup(component_html, features="html.parser")
+    actual = get_root_element(soup)
+
+    assert str(actual) == expected
+
+
 def test_get_root_element_direct_view_no_view():
     component_html = "<html><head></head>≤body><div>test</div></body></html>"
     soup = BeautifulSoup(component_html, features="html.parser")
