@@ -367,40 +367,42 @@ async def test_diagnostics_collection(home_data: HomeData) -> None:
367367async def test_unsupported_protocol_version () -> None :
368368 """Test the DeviceManager with some supported and unsupported product IDs."""
369369 with patch ("roborock.devices.device_manager.UserWebApiClient.get_home_data" ) as mock_home_data :
370- home_data = HomeData .from_dict ({
371- "id" : 1 ,
372- "name" : "Test Home" ,
373- "devices" : [
374- {
375- "duid" : "device-uid-1" ,
376- "name" : "Device 1" ,
377- "pv" : "1.0" ,
378- "productId" : "product-id-1" ,
379- "localKey" : mock_data .LOCAL_KEY ,
380- },
381- {
382- "duid" : "device-uid-2" ,
383- "name" : "Device 2" ,
384- "pv" : "unknown-pv" , # Fake new protocol version we've never seen
385- "productId" : "product-id-2" ,
386- "localKey" : mock_data .LOCAL_KEY ,
387- },
388- ],
389- "products" : [
390- {
391- "id" : "product-id-1" ,
392- "name" : "Roborock S7 MaxV" ,
393- "model" : "roborock.vacuum.a27" ,
394- "category" : "robot.vacuum.cleaner" ,
395- },
396- {
397- "id" : "product-id-2" ,
398- "name" : "New Roborock Model" ,
399- "model" : "roborock.vacuum.newmodel" ,
400- "category" : "robot.vacuum.cleaner" ,
401- },
402- ],
403- })
370+ home_data = HomeData .from_dict (
371+ {
372+ "id" : 1 ,
373+ "name" : "Test Home" ,
374+ "devices" : [
375+ {
376+ "duid" : "device-uid-1" ,
377+ "name" : "Device 1" ,
378+ "pv" : "1.0" ,
379+ "productId" : "product-id-1" ,
380+ "localKey" : mock_data .LOCAL_KEY ,
381+ },
382+ {
383+ "duid" : "device-uid-2" ,
384+ "name" : "Device 2" ,
385+ "pv" : "unknown-pv" , # Fake new protocol version we've never seen
386+ "productId" : "product-id-2" ,
387+ "localKey" : mock_data .LOCAL_KEY ,
388+ },
389+ ],
390+ "products" : [
391+ {
392+ "id" : "product-id-1" ,
393+ "name" : "Roborock S7 MaxV" ,
394+ "model" : "roborock.vacuum.a27" ,
395+ "category" : "robot.vacuum.cleaner" ,
396+ },
397+ {
398+ "id" : "product-id-2" ,
399+ "name" : "New Roborock Model" ,
400+ "model" : "roborock.vacuum.newmodel" ,
401+ "category" : "robot.vacuum.cleaner" ,
402+ },
403+ ],
404+ }
405+ )
404406 mock_home_data .return_value = home_data
405407
406408 device_manager = await create_device_manager (USER_PARAMS )
0 commit comments