Update pilot.app.query_one call.

Author: JoeZiminski

tests/tests_tui/test_tui_datatypes.py

@@ -50,7 +50,9 @@ async def test_select_displayed_datatypes_create(
 
             for datatype in narrow_datatype_names:
                 assert (
-                    pilot.app.query_one(f"#create_{datatype}_checkbox").value
+                    pilot.app.screen.query_one(
+                        f"#create_{datatype}_checkbox"
+                    ).value
                     is False
                 )
 
@@ -96,7 +98,9 @@ async def test_select_displayed_datatypes_create(
             for datatype in broad_datatype_names:
                 # check all are shown and False again (because False on reset)
                 assert (
-                    pilot.app.query_one(f"#create_{datatype}_checkbox").value
+                    pilot.app.screen.query_one(
+                        f"#create_{datatype}_checkbox"
+                    ).value
                     is False
                 )
 
@@ -116,13 +120,15 @@ async def test_select_displayed_datatypes_create(
             for datatype in broad_datatype_names:
                 # check all are shown and False again
                 assert (
-                    pilot.app.query_one(f"#create_{datatype}_checkbox").value
+                    pilot.app.screen.query_one(
+                        f"#create_{datatype}_checkbox"
+                    ).value
                     is False
                 )
 
             # Confirm also that narrow datatypes are not shown.
             with pytest.raises(BaseException):
-                pilot.app.query_one(
+                pilot.app.screen.query_one(
                     f"#create_{narrow_datatype_names[0]}_checkbox"
                 )
 
@@ -165,7 +171,9 @@ async def test_select_displayed_datatypes_transfer(
 
             for datatype in narrow_datatype_names:
                 assert (
-                    pilot.app.query_one(f"#transfer_{datatype}_checkbox").value
+                    pilot.app.screen.query_one(
+                        f"#transfer_{datatype}_checkbox"
+                    ).value
                     is False
                 )
 

tests/tests_tui/test_tui_validate.py

@@ -58,7 +58,9 @@ async def test_validate_on_project_manager_output(
 
             written_lines = [
                 ele.text
-                for ele in pilot.app.query_one("#validate_richlog").lines
+                for ele in pilot.app.screen.query_one(
+                    "#validate_richlog"
+                ).lines
             ]
 
             assert len(written_lines) == 3
@@ -157,7 +159,7 @@ async def test_validate_on_project_manager_kwargs(
                 "validate_path_container",
             ]:
                 with pytest.raises(textual.css.query.InvalidQueryFormat):
-                    pilot.app.query_one(id)
+                    pilot.app.screen.query_one(id)
 
     @pytest.mark.asyncio
     async def test_validate_at_path_kwargs(self, setup_project_paths, mocker):
@@ -198,4 +200,4 @@ async def test_validate_at_path_kwargs(self, setup_project_paths, mocker):
 
             # Check removed widgets, this should be removed because always local
             with pytest.raises(textual.css.query.InvalidQueryFormat):
-                pilot.app.query_one("validate_include_central_checkbox")
+                pilot.app.screen.query_one("validate_include_central_checkbox")