From 8ddd1973677c55bf0c4900fe528ad3166ba9033c Mon Sep 17 00:00:00 2001 From: Gaius Date: Tue, 24 May 2022 21:50:22 +0800 Subject: [PATCH] test: watchdog Signed-off-by: Gaius --- client/daemon/peer/peertask_piecetask_synchronizer_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/client/daemon/peer/peertask_piecetask_synchronizer_test.go b/client/daemon/peer/peertask_piecetask_synchronizer_test.go index d71f8b2e4..0a8beb2de 100644 --- a/client/daemon/peer/peertask_piecetask_synchronizer_test.go +++ b/client/daemon/peer/peertask_piecetask_synchronizer_test.go @@ -27,7 +27,7 @@ import ( logger "d7y.io/dragonfly/v2/internal/dflog" "d7y.io/dragonfly/v2/pkg/rpc/scheduler" - "d7y.io/dragonfly/v2/pkg/rpc/scheduler/client/mocks" + "d7y.io/dragonfly/v2/pkg/rpc/scheduler/mocks" ) func Test_watchdog(t *testing.T) { @@ -54,7 +54,7 @@ func Test_watchdog(t *testing.T) { for _, tt := range testCases { t.Run(tt.name, func(t *testing.T) { peer := &scheduler.PeerPacket_DestPeer{} - pps := mocks.NewMockPeerPacketStream(ctrl) + pps := mocks.NewMockScheduler_ReportPieceResultClient(ctrl) watchdog := &synchronizerWatchdog{ done: make(chan struct{}), mainPeer: atomic.Value{},