diff --git a/src/web/bridge/tests.rs b/src/web/bridge/tests.rs index 481301e..251e566 100644 --- a/src/web/bridge/tests.rs +++ b/src/web/bridge/tests.rs @@ -225,28 +225,33 @@ fn committed_websocket_lane_escalates_only_pre_upgrade_failure() { assert!(page.body.contains( "if(settled)return;settled=true;if(openTimer)clearTimeout(openTimer);openTimer=null" )); - assert!(page - .body - .contains("if(closed||lanes.get(lane.id)!==lane||lane.socket!==opened)return")); + assert!( + page.body + .contains("if(closed||lanes.get(lane.id)!==lane||lane.socket!==opened)return") + ); assert!(page.body.contains( "if(!upgraded){lane.socket=null;opened.close();recoveryController.recover(reason,null);return}" )); - assert!(page - .body - .contains("recoveryController.recover(reason,null);return}finishLane(lane,true)")); - assert!(page - .body - .contains("openTimer=setTimeout(()=>finishSocket('timeout'),websocketOpenMs)")); + assert!( + page.body + .contains("recoveryController.recover(reason,null);return}finishLane(lane,true)") + ); + assert!( + page.body + .contains("openTimer=setTimeout(()=>finishSocket('timeout'),websocketOpenMs)") + ); assert!(page.body.contains( "if(closed||lanes.get(lane.id)!==lane||lane.socket!==opened){opened.close();return}" )); assert!(page.body.contains("upgraded=true;lane.ready=true")); - assert!(page - .body - .contains("lane.socket.onclose=()=>finishSocket(upgraded?'network':'upgrade')")); - assert!(page - .body - .contains("port.postMessage({t:'status',state})")); + assert!(page.body.contains( + "lane.socket.onmessage=event=>{\n if(openTimer)clearTimeout(openTimer);openTimer=null;\n if(closed||lanes.get(lane.id)!==lane||lane.socket!==opened||!(event.data instanceof ArrayBuffer))" + )); + assert!( + page.body + .contains("lane.socket.onclose=()=>finishSocket(upgraded?'network':'upgrade')") + ); + assert!(page.body.contains("port.postMessage({t:'status',state})")); assert!(!page.body.contains("port.postMessage({t:'status',state,")); }