Add test to check for u:view.

adamghill · adamghill · commit c66012a8c618 · 2022-11-02T22:08:06.000-04:00
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")
