From 393695d7743eee23f153dbf7c62b7d536fbb445e Mon Sep 17 00:00:00 2001 From: Marten Seemann Date: Tue, 1 Oct 2019 10:13:46 +0700 Subject: [PATCH] fix flaky HTTP/3 client unit test --- http3/client_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/http3/client_test.go b/http3/client_test.go index 925d9f74b..be16367af 100644 --- a/http3/client_test.go +++ b/http3/client_test.go @@ -309,9 +309,9 @@ var _ = Describe("Client", func() { canceled := make(chan struct{}) gomock.InOrder( str.EXPECT().CancelWrite(quic.ErrorCode(errorRequestCanceled)).Do(func(quic.ErrorCode) { close(canceled) }), - str.EXPECT().CancelWrite(gomock.Any()).MaxTimes(1).Do(func(quic.ErrorCode) { close(done) }), + str.EXPECT().CancelRead(quic.ErrorCode(errorRequestCanceled)).Do(func(quic.ErrorCode) { close(done) }), ) - str.EXPECT().CancelRead(quic.ErrorCode(errorRequestCanceled)) + str.EXPECT().CancelWrite(gomock.Any()).MaxTimes(1) str.EXPECT().Read(gomock.Any()).DoAndReturn(func([]byte) (int, error) { cancel() <-canceled