Fixed tests

iommirocks · Jan 31, 2024 · 18b45b0 · 18b45b0
1 parent 48f0407
commit 18b45b0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/iommi/style__tests.py b/iommi/style__tests.py
@@ -785,7 +785,7 @@ class TestTable(Table):
         foo = Column.boolean()
 
     verify_table_html(
-        table=TestTable(rows=(x for x in [Struct(foo=True)]), iommi_style='base'),
+        table=TestTable(rows=(x for x in [Struct(foo=True)]), iommi_style='base', attrs__class__table=True),
         # language=html
         expected_html="""
             <table class="table" data-endpoint="/endpoints/tbody" data-iommi-id="">
@@ -804,7 +804,7 @@ class TestTable(Table):
     )
 
     verify_table_html(
-        table=TestTable(rows=(x for x in [Struct(foo=True)]), iommi_style='bulma'),
+        table=TestTable(rows=(x for x in [Struct(foo=True)]), iommi_style='bulma', attrs__class__table=True),
         # language=html
         expected_html="""
             <table class="is-fullwidth is-hoverable table" data-endpoint="/endpoints/tbody" data-iommi-id="">
@@ -823,7 +823,7 @@ class TestTable(Table):
     )
 
     verify_table_html(
-        table=TestTable(rows=(x for x in [Struct(foo=True)]), iommi_style='bootstrap5'),
+        table=TestTable(rows=(x for x in [Struct(foo=True)]), iommi_style='bootstrap5', attrs__class__table=True),
         # language=html
         expected_html="""
             <table class="table table-sm" data-endpoint="/endpoints/tbody" data-iommi-id="">

diff --git a/tests/helpers.py b/tests/helpers.py
@@ -87,9 +87,10 @@ def verify_html(*, actual_html: str, find=None, expected_html: str = None):
     actual_soup = BeautifulSoup(actual_html, 'html.parser')
 
     if find is not None:
+        actual_soup_orig = actual_soup
         actual_soup = actual_soup.find(**find)
         if not actual_soup:  # pragma: no cover
-            prettied_actual = reindent(actual_soup.prettify()).strip()
+            prettied_actual = reindent(actual_soup_orig.prettify()).strip()
             print(prettied_actual)
             assert False, f"Couldn't find selector {find} in actual output"