@@ -93,10 +93,11 @@ def test_classic_ti_plan( # noqa: PLR0915
9393 is_product_lite : bool ,
9494 create_tip_plan_from_dashboard : Callable [[str ], dict [str , Any ]],
9595):
96+ # checks "Access TIP" button
97+ page .get_by_test_id ("userMenuBtn" ).locator ("div" ).click ()
98+ assert page .get_by_text ("Access Full TIP" ).is_visible ()
9699
97- assert page .get_by_text ("Access TIP" ).is_enabled ()
98- # TODO: check TIP upgrade
99-
100+ # press + button
100101 project_data = create_tip_plan_from_dashboard ("newTIPlanButton" )
101102 assert "workbench" in project_data , "Expected workbench to be in project data!"
102103 assert isinstance (
@@ -115,7 +116,9 @@ def test_classic_ti_plan( # noqa: PLR0915
115116 len (node_ids ) >= expected_number_of_steps
116117 ), f"Expected at least { expected_number_of_steps } nodes in the workbench"
117118
118- with log_context (logging .INFO , "Electrode Selector step (1)" ) as ctx :
119+ with log_context (
120+ logging .INFO , "Electrode Selector step (1/%s)" , expected_number_of_steps
121+ ) as ctx :
119122 # NOTE: creating the plan auto-triggers the first service to start, which might already triggers socket events
120123 electrode_selector_iframe = wait_for_service_running (
121124 page = page ,
@@ -164,7 +167,9 @@ def test_classic_ti_plan( # noqa: PLR0915
164167 response_body = response .json ()
165168 ctx .logger .info ("the following output was generated: %s" , response_body )
166169
167- with log_context (logging .INFO , "Classic TI step (2)" ) as ctx :
170+ with log_context (
171+ logging .INFO , "Classic TI step (2/%s)" , expected_number_of_steps
172+ ) as ctx :
168173 with page .expect_websocket (
169174 _JLabWaitForWebSocket (),
170175 timeout = _OUTER_EXPECT_TIMEOUT_RATIO
@@ -250,7 +255,9 @@ def test_classic_ti_plan( # noqa: PLR0915
250255 if is_product_lite :
251256 assert expected_number_of_steps == 2
252257 else :
253- with log_context (logging .INFO , "Exposure Analysis step (3)" ):
258+ with log_context (
259+ logging .INFO , "Exposure Analysis step (3/%s)" , expected_number_of_steps
260+ ):
254261 with expected_service_running (
255262 page = page ,
256263 node_id = node_ids [2 ],
0 commit comments