From 7da83e9e6c3edfd18f569420dbb7e948af8f1abe Mon Sep 17 00:00:00 2001 From: c9s Date: Wed, 20 Nov 2024 18:12:15 +0800 Subject: [PATCH] fix twap tests --- pkg/twap/v2/stream_executor_test.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pkg/twap/v2/stream_executor_test.go b/pkg/twap/v2/stream_executor_test.go index fe3f05cdb..e53658efc 100644 --- a/pkg/twap/v2/stream_executor_test.go +++ b/pkg/twap/v2/stream_executor_test.go @@ -81,6 +81,9 @@ func bindMockMarketDataStream(mockStream *mocks.MockStream, stream *types.Standa mockStream.EXPECT().OnConnect(Catch(func(x any) { stream.OnConnect(x.(func())) })).AnyTimes() + mockStream.EXPECT().OnDisconnect(Catch(func(x any) { + stream.OnDisconnect(x.(func())) + })).AnyTimes() } func bindMockUserDataStream(mockStream *mocks.MockStream, stream *types.StandardStream) {