|
29 | 29 | MINUTE, |
30 | 30 | SECOND, |
31 | 31 | AutoRegisteredUser, |
32 | | - RestartableWebSocket, |
| 32 | + RobustWebSocket, |
33 | 33 | RunningState, |
34 | 34 | ServiceType, |
35 | 35 | SocketIOEvent, |
@@ -366,7 +366,7 @@ def log_in_and_out( |
366 | 366 | register: Callable[[], AutoRegisteredUser], |
367 | 367 | store_browser_context: bool, |
368 | 368 | context: BrowserContext, |
369 | | -) -> Iterator[RestartableWebSocket]: |
| 369 | +) -> Iterator[RobustWebSocket]: |
370 | 370 | with log_context( |
371 | 371 | logging.INFO, |
372 | 372 | f"Open {product_url=} using {user_name=}/{user_password=}/{auto_register=} with {browser.browser_type.name}:{browser.version}({browser.browser_type.executable_path})", |
@@ -410,7 +410,7 @@ def log_in_and_out( |
410 | 410 | assert response_info.value.ok, f"{response_info.value.json()}" |
411 | 411 |
|
412 | 412 | assert not ws_info.value.is_closed() |
413 | | - restartable_wb = RestartableWebSocket.create(page, ws_info.value) |
| 413 | + restartable_wb = RobustWebSocket.create(page, ws_info.value) |
414 | 414 |
|
415 | 415 | # Welcome to Sim4Life |
416 | 416 | page.wait_for_timeout(5000) |
@@ -473,7 +473,7 @@ def _select_service_version(page: Page, *, version: str) -> None: |
473 | 473 | @pytest.fixture |
474 | 474 | def create_new_project_and_delete( # noqa: C901, PLR0915 |
475 | 475 | page: Page, |
476 | | - log_in_and_out: RestartableWebSocket, |
| 476 | + log_in_and_out: RobustWebSocket, |
477 | 477 | is_product_billable: bool, |
478 | 478 | api_request_context: APIRequestContext, |
479 | 479 | product_url: AnyUrl, |
@@ -735,7 +735,7 @@ def _( |
735 | 735 | def start_and_stop_pipeline( |
736 | 736 | product_url: AnyUrl, |
737 | 737 | page: Page, |
738 | | - log_in_and_out: RestartableWebSocket, |
| 738 | + log_in_and_out: RobustWebSocket, |
739 | 739 | api_request_context: APIRequestContext, |
740 | 740 | ) -> Iterator[Callable[[], SocketIOEvent]]: |
741 | 741 | started_pipeline_ids = [] |
|
0 commit comments