review

quinn-proto/src/tests/util.rs

@@ -456,10 +456,7 @@ impl TestEndpoint {
         now: Instant,
     ) -> Result<ConnectionHandle, ConnectionError> {
         let mut buf = BytesMut::new();
-        match self
-            .endpoint
-            .accept(incoming, now, &mut buf, Default::default())
-        {
+        match self.endpoint.accept(incoming, now, &mut buf, None) {
             Ok((ch, conn)) => {
                 self.connections.insert(ch, conn);
                 self.accepted = Some(Ok(ch));

quinn/src/incoming.rs

@@ -24,12 +24,22 @@ impl Incoming {
     }
 
     /// Attempt to accept this incoming connection (an error may still occur)
-    pub fn accept(
+    pub fn accept(mut self) -> Result<Connecting, ConnectionError> {
+        let state = self.0.take().unwrap();
+        state.endpoint.accept(state.inner, None)
+    }
+
+    /// Accept this incoming connection using a custom configuration.
+    ///
+    /// See [`accept()`] for more details.
+    ///
+    /// [`accept()`]: Incoming::accept
+    pub fn accept_with(
         mut self,
-        server_config: Option<Arc<ServerConfig>>,
+        server_config: Arc<ServerConfig>,
     ) -> Result<Connecting, ConnectionError> {
         let state = self.0.take().unwrap();
-        state.endpoint.accept(state.inner, server_config)
+        state.endpoint.accept(state.inner, Some(server_config))
     }
 
     /// Reject this incoming connection attempt
@@ -124,6 +134,6 @@ impl IntoFuture for Incoming {
     type IntoFuture = IncomingFuture;
 
     fn into_future(self) -> Self::IntoFuture {
-        IncomingFuture(self.accept(None))
+        IncomingFuture(self.accept())
     }
 }

quinn/src/tests.rs

@@ -293,7 +293,7 @@ async fn zero_rtt() {
     let endpoint2 = endpoint.clone();
     tokio::spawn(async move {
         for _ in 0..2 {
-            let incoming = endpoint2.accept().await.unwrap().accept(None).unwrap();
+            let incoming = endpoint2.accept().await.unwrap().accept().unwrap();
             let (connection, established) = incoming.into_0rtt().unwrap_or_else(|_| unreachable!());
             let c = connection.clone();
             tokio::spawn(async move {