test fixes

Author: willmcgugan

tests/animations/test_loading_indicator_animation.py

@@ -4,7 +4,6 @@
 """
 
 from textual.app import App
-from textual.widgets import LoadingIndicator
 
 
 async def test_loading_indicator_is_not_static_on_full() -> None:
@@ -15,7 +14,7 @@ async def test_loading_indicator_is_not_static_on_full() -> None:
     async with app.run_test() as pilot:
         app.screen.loading = True
         await pilot.pause()
-        indicator = app.query_one(LoadingIndicator)
+        indicator = app.screen._cover_widget
         assert str(indicator.render()) != "Loading..."
 
 
@@ -27,7 +26,7 @@ async def test_loading_indicator_is_not_static_on_basic() -> None:
     async with app.run_test() as pilot:
         app.screen.loading = True
         await pilot.pause()
-        indicator = app.query_one(LoadingIndicator)
+        indicator = app.screen._cover_widget
         assert str(indicator.render()) != "Loading..."
 
 
@@ -39,5 +38,5 @@ async def test_loading_indicator_is_static_on_none() -> None:
     async with app.run_test() as pilot:
         app.screen.loading = True
         await pilot.pause()
-        indicator = app.query_one(LoadingIndicator)
+        indicator = app.screen._cover_widget
         assert str(indicator.render()) == "Loading..."

tests/test_widget.py

@@ -425,23 +425,23 @@ def compose(self) -> ComposeResult:
         app = pilot.app
         label = app.query_one(Label)
         assert label.loading == False
-        assert len(label.query(LoadingIndicator)) == 0
+        assert label._cover_widget is None
 
         label.loading = True
         await pilot.pause()
-        assert len(label.query(LoadingIndicator)) == 1
+        assert label._cover_widget is not None
 
         label.loading = True  # Setting to same value is a null-op
         await pilot.pause()
-        assert len(label.query(LoadingIndicator)) == 1
+        assert label._cover_widget is not None
 
         label.loading = False
         await pilot.pause()
-        assert len(label.query(LoadingIndicator)) == 0
+        assert label._cover_widget is None
 
         label.loading = False  # Setting to same value is a null-op
         await pilot.pause()
-        assert len(label.query(LoadingIndicator)) == 0
+        assert label._cover_widget is None
 
 
 async def test_is_mounted_property():