fix change

quinn-proto/src/tests/mod.rs

@@ -2760,10 +2760,13 @@ fn reject_manually() {
     // The server should now reject incoming connections.
     let client_ch = pair.begin_connect(client_config());
     pair.drive();
-    let e = pair.server.assert_accept_error();
-    assert!(
-        matches!(e, crate::ConnectionError::ConnectionClosed(_)),
-        "wrong error"
-    );
-    assert!(pair.client.connections.get(&client_ch).unwrap().is_closed());
+    pair.server.assert_no_accept();
+    let client = pair.client.connections.get_mut(&client_ch).unwrap();
+    assert!(client.is_closed());
+    assert!(matches!(
+        client.poll(),
+        Some(Event::ConnectionLost {
+            reason: ConnectionError::ConnectionClosed(close)
+        }) if close.error_code == TransportErrorCode::CONNECTION_REFUSED
+    ))
 }

quinn-proto/src/tests/util.rs

@@ -475,6 +475,7 @@ impl TestEndpoint {
                     self.outbound
                         .extend(split_transmit(transmit, buf.split_to(size).freeze()));
                 }
+                self.accepted = Some(Err(e.clone()));
                 e
             })
     }