Fix test compatibility with panel-material-ui < 0.10

- Use hasattr fallback for footer_actions/footer_objects in tests
- Disable follow_up_suggestions in edit test to avoid MockLLM exhaustion

Author: ghostiee-11

lumen/tests/ai/test_ui.py

@@ -1466,6 +1466,7 @@ async def test_edit_on_second_exploration_preserves_first(explorer_ui):
     second one, only the second exploration (sql_2 + spec_2) should be
     removed. The first exploration (sql_1 + spec_1) must remain intact."""
     ui = explorer_ui
+    ui.follow_up_suggestions = False
     test_source = ui.context["source"]
     SQLQueryWithTables = make_sql_model([(test_source.name, "test_table")])
 
@@ -1673,7 +1674,8 @@ async def test_followup_suggestions_after_query(explorer_ui):
     await asyncio.sleep(0.5)
 
     last_msg = ui.interface.objects[-1]
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 1
     assert followup_icons[0].icon == "lightbulb"
@@ -1708,7 +1710,8 @@ async def test_followup_suggestions_disabled(explorer_ui):
     await asyncio.sleep(0.3)
 
     last_msg = ui.interface.objects[-1]
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 0
 
@@ -1719,7 +1722,8 @@ async def test_followup_suggestions_not_on_error(explorer_ui_with_error):
     await asyncio.sleep(0.3)
 
     last_msg = ui.interface.objects[-1]
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 0
 
@@ -1758,7 +1762,8 @@ def raise_error():
     # Should not crash, no followup suggestions, but plan succeeded
     assert len(ui._explorations.items) == 2
     last_msg = ui.interface.objects[-1]
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 0
 
@@ -1799,7 +1804,8 @@ async def test_followup_suggestion_click_sends_query(explorer_ui):
 
     # Get the follow-up icon button
     last_msg = ui.interface.objects[-1]
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 1
 
@@ -1846,7 +1852,8 @@ async def test_followup_icon_cycles_suggestions(explorer_ui):
     await asyncio.sleep(0.5)
 
     last_msg = ui.interface.objects[-1]
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 1
     icon = followup_icons[0]
@@ -1917,7 +1924,8 @@ async def invoke_with_extra_message(*args, **kwargs):
     # The extra message means num_objects changed, so no FollowUp should appear
     footer_with_followup = []
     for msg in ui.interface.objects:
-        footer_actions = msg.footer_actions or []
+        _attr = "footer_actions" if hasattr(msg, "footer_actions") else "footer_objects"
+        footer_actions = getattr(msg, _attr) or []
         footer_with_followup.extend(
             f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"
         )
@@ -1959,7 +1967,8 @@ async def test_analysis_suggestions_still_use_default_group_name(explorer_ui):
     last_msg = ui.interface.objects[-1]
 
     # FollowUp is now in footer_actions (icon button), not footer_objects
-    footer_actions = last_msg.footer_actions or []
+    _attr = "footer_actions" if hasattr(last_msg, "footer_actions") else "footer_objects"
+    footer_actions = getattr(last_msg, _attr) or []
     followup_icons = [f for f in footer_actions if getattr(f, 'name', '') == "FollowUp"]
     assert len(followup_icons) == 1