Skip to content

Commit dc26f35

Browse files
committed
refactor: rename self.servers to self.simulated_devices and restore hardcoded qiCNieZa in mock_data
1 parent 211653b commit dc26f35

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

roborock/testing/cloud.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def get_homes_callback(url, **kwargs):
142142

143143
devices = []
144144
products = []
145-
for server in self.cloud.servers.values():
145+
for server in self.cloud.simulated_devices.values():
146146
devices.append(server.device_info)
147147
products.append(server.product)
148148

@@ -182,16 +182,16 @@ def __init__(
182182
user_data: UserData | None = None,
183183
home_id: int = 123456,
184184
home_name: str = "Fake Home",
185-
):
186-
self.servers: dict[str, RoborockDeviceSimulator] = {}
185+
) -> None:
186+
self.simulated_devices: dict[str, RoborockDeviceSimulator] = {}
187187
self.user_data = user_data or DEFAULT_USER_DATA
188188
self.home_id = home_id
189189
self.home_name = home_name
190190
self.web_api = FakeWebApiClient(self)
191191

192192
def add_device(self, server: RoborockDeviceSimulator) -> None:
193193
"""Register a stateful device simulator in the cloud registry."""
194-
self.servers[server.duid] = server
194+
self.simulated_devices[server.duid] = server
195195

196196
@contextlib.contextmanager
197197
def patch_device_manager(self):
@@ -208,7 +208,7 @@ def mock_create_v1_channel(user_data, mqtt_params, mqtt_session, device, device_
208208
f"Simulating protocol {device.pv} is not yet supported. "
209209
"TODO: Implement stateful simulators for B01 (Q7/Q10) and A01 (Zeo/Dyad) devices."
210210
)
211-
server = self.servers.get(device.duid)
211+
server = self.simulated_devices.get(device.duid)
212212
if server is not None:
213213
if not isinstance(server, V1VacuumSimulator):
214214
raise TypeError(
@@ -225,7 +225,7 @@ def mock_create_mqtt_channel(user_data, mqtt_params, mqtt_session, device):
225225
f"Simulating protocol {device.pv} is not yet supported. "
226226
"TODO: Implement stateful simulators for B01 (Q7/Q10) and A01 (Zeo/Dyad) devices."
227227
)
228-
server = self.servers.get(device.duid)
228+
server = self.simulated_devices.get(device.duid)
229229
if server:
230230
return server.mqtt_channel
231231
return original_create_mqtt_channel(user_data, mqtt_params, mqtt_session, device)

0 commit comments

Comments
 (0)