|
4 | 4 |
|
5 | 5 | import pytest |
6 | 6 |
|
7 | | -from roborock_local_server.onboarding_cli import GuidedOnboardingConfig, run_guided_onboarding |
| 7 | +from start_onboarding import GuidedOnboardingConfig, run_guided_onboarding |
8 | 8 |
|
9 | 9 |
|
10 | 10 | class FakeApi: |
@@ -97,7 +97,7 @@ def test_guided_onboarding_happy_path(monkeypatch: pytest.MonkeyPatch, config: G |
97 | 97 | answers = iter(["1", ""]) |
98 | 98 | monkeypatch.setattr("builtins.input", lambda prompt="": next(answers)) |
99 | 99 | monkeypatch.setattr( |
100 | | - "roborock_local_server.onboarding_cli.poll_session_until_progress", |
| 100 | + "start_onboarding.poll_session_until_progress", |
101 | 101 | lambda *args, **kwargs: next(poll_results), |
102 | 102 | ) |
103 | 103 |
|
@@ -177,7 +177,7 @@ def test_guided_onboarding_handles_extra_cycles(monkeypatch: pytest.MonkeyPatch, |
177 | 177 | answers = iter(["1", "", "", ""]) |
178 | 178 | monkeypatch.setattr("builtins.input", lambda prompt="": next(answers)) |
179 | 179 | monkeypatch.setattr( |
180 | | - "roborock_local_server.onboarding_cli.poll_session_until_progress", |
| 180 | + "start_onboarding.poll_session_until_progress", |
181 | 181 | lambda *args, **kwargs: next(poll_results), |
182 | 182 | ) |
183 | 183 |
|
@@ -247,7 +247,7 @@ def test_guided_onboarding_timeout_can_retry_without_restart( |
247 | 247 | answers = iter(["1", "", "retry", ""]) |
248 | 248 | monkeypatch.setattr("builtins.input", lambda prompt="": next(answers)) |
249 | 249 | monkeypatch.setattr( |
250 | | - "roborock_local_server.onboarding_cli.poll_session_until_progress", |
| 250 | + "start_onboarding.poll_session_until_progress", |
251 | 251 | lambda *args, **kwargs: next(poll_results), |
252 | 252 | ) |
253 | 253 |
|
@@ -307,7 +307,7 @@ def test_guided_onboarding_duplicate_names_still_selects_requested_device( |
307 | 307 | answers = iter(["2", ""]) |
308 | 308 | monkeypatch.setattr("builtins.input", lambda prompt="": next(answers)) |
309 | 309 | monkeypatch.setattr( |
310 | | - "roborock_local_server.onboarding_cli.poll_session_until_progress", |
| 310 | + "start_onboarding.poll_session_until_progress", |
311 | 311 | lambda *args, **kwargs: next(poll_results), |
312 | 312 | ) |
313 | 313 |
|
|
0 commit comments