From a4bf8bc169ab7073518c4db5f4635c0a2c950baf Mon Sep 17 00:00:00 2001
From: Benjamin Saunders <ben.e.saunders@gmail.com>
Date: Fri, 11 Jan 2019 18:19:46 -0800
Subject: [PATCH] Strengthen 0-RTT test

---
 quinn-proto/src/tests.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/quinn-proto/src/tests.rs b/quinn-proto/src/tests.rs
index 2fe7c6f60..7bd80a83e 100644
--- a/quinn-proto/src/tests.rs
+++ b/quinn-proto/src/tests.rs
@@ -664,10 +664,12 @@ fn zero_rtt() {
         .client
         .connect(pair.server.addr, &config, "localhost")
         .unwrap();
+    assert!(pair.client.connection(client_conn).has_0rtt());
     let s = pair.client.open(client_conn, Directionality::Uni).unwrap();
     const MSG: &[u8] = b"Hello, 0-RTT!";
     pair.client.write(client_conn, s, MSG).unwrap();
     pair.drive();
+    assert!(pair.client.connection(client_conn).accepted_0rtt());
     let server_conn = if let Some(c) = pair.server.accept() {
         c
     } else {