@@ -54,11 +54,11 @@ async def test_full_workflow(session_mocker, tmp_path: Path):
5454 assert pilot .app .screen ._start_search_button .disabled is True
5555
5656 # Test partition list default state
57- filter_checkbox = pilot .app .query ("Checkbox" ).only_one ()
57+ filter_checkbox = pilot .app .screen . query ("Checkbox" ).only_one ()
5858 assert filter_checkbox is not None
5959 assert filter_checkbox .value is True
6060
61- items = list (pilot .app .query ("ListItem" ).results ())
61+ items = list (pilot .app .screen . query ("ListItem" ).results ())
6262 assert len (items ) == VISIBLE_PARTITION_COUNT
6363
6464 # Test partition list unfiltered
@@ -67,10 +67,10 @@ async def test_full_workflow(session_mocker, tmp_path: Path):
6767
6868 assert filter_checkbox .value is False
6969 await assert_with_timeout (
70- lambda : len (list (pilot .app .query ("ListItem" ).results ()))
70+ lambda : len (list (pilot .app .screen . query ("ListItem" ).results ()))
7171 == UNFILTERED_PARTITION_COUNT ,
7272 UNFILTERED_PARTITION_COUNT ,
73- len (list (pilot .app .query ("ListItem" ).results ())),
73+ len (list (pilot .app .screen . query ("ListItem" ).results ())),
7474 )
7575
7676 # Test partition list filtered
@@ -79,10 +79,10 @@ async def test_full_workflow(session_mocker, tmp_path: Path):
7979
8080 assert filter_checkbox .value is True
8181 await assert_with_timeout (
82- lambda : len (list (pilot .app .query ("ListItem" ).results ()))
82+ lambda : len (list (pilot .app .screen . query ("ListItem" ).results ()))
8383 == VISIBLE_PARTITION_COUNT ,
8484 VISIBLE_PARTITION_COUNT ,
85- len (list (pilot .app .query ("ListItem" ).results ())),
85+ len (list (pilot .app .screen . query ("ListItem" ).results ())),
8686 )
8787
8888 # Test input search parameters
@@ -133,14 +133,14 @@ async def test_full_workflow(session_mocker, tmp_path: Path):
133133 len (pilot .app .screen ._grep_result_list .grep_results ),
134134 )
135135 await assert_with_timeout (
136- lambda : len (list (pilot .app .query ("ListItem" ).results ()))
136+ lambda : len (list (pilot .app .screen . query ("ListItem" ).results ()))
137137 == GREP_RESULT_COUNT ,
138138 GREP_RESULT_COUNT ,
139- len (list (pilot .app .query ("ListItem" ).results ())),
139+ len (list (pilot .app .screen . query ("ListItem" ).results ())),
140140 )
141141
142142 # Test select search results
143- list_items = list (pilot .app .query ("ListItem" ).results ())
143+ list_items = list (pilot .app .screen . query ("ListItem" ).results ())
144144
145145 for item in list_items :
146146 await pilot .click (f"#{ item .id } " )
@@ -223,7 +223,7 @@ async def test_full_workflow(session_mocker, tmp_path: Path):
223223 add_expected_save_result (pilot )
224224
225225 # Test select previous result
226- previous_button = pilot .app .query ("#previous-button" ).only_one ()
226+ previous_button = pilot .app .screen . query ("#previous-button" ).only_one ()
227227 current_inode = pilot .app .screen ._inode
228228 await pilot .click ("#previous-button" )
229229 await pilot .pause ()
0 commit comments