From d8d622ee99b632f1454b396177113366cb22ff0b Mon Sep 17 00:00:00 2001 From: Marten Seemann Date: Sat, 22 Feb 2020 16:16:25 +0700 Subject: [PATCH] fix flaky 0.5-RTT data test --- integrationtests/self/early_data_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/integrationtests/self/early_data_test.go b/integrationtests/self/early_data_test.go index d1d82ed7e..ab12ea378 100644 --- a/integrationtests/self/early_data_test.go +++ b/integrationtests/self/early_data_test.go @@ -31,6 +31,7 @@ var _ = Describe("early data", func() { done := make(chan struct{}) go func() { defer GinkgoRecover() + defer close(done) sess, err := ln.Accept(context.Background()) Expect(err).ToNot(HaveOccurred()) str, err := sess.OpenUniStream() @@ -40,8 +41,7 @@ var _ = Describe("early data", func() { Expect(str.Close()).To(Succeed()) // make sure the Write finished before the handshake completed Expect(sess.HandshakeComplete().Done()).ToNot(BeClosed()) - Eventually(sess.HandshakeComplete().Done(), protocol.DefaultHandshakeTimeout).Should(BeClosed()) - close(done) + Eventually(sess.Context().Done()).Should(BeClosed()) }() serverPort := ln.Addr().(*net.UDPAddr).Port proxy, err := quicproxy.NewQuicProxy("localhost:0", &quicproxy.Opts{