test: watchdog

Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
Gaius 2022-05-24 21:50:22 +08:00
parent 3e8072daaa
commit 8ddd197367
No known key found for this signature in database
GPG Key ID: 8B4E5D1290FA2FFB
1 changed files with 2 additions and 2 deletions

View File

@ -27,7 +27,7 @@ import (
logger "d7y.io/dragonfly/v2/internal/dflog" logger "d7y.io/dragonfly/v2/internal/dflog"
"d7y.io/dragonfly/v2/pkg/rpc/scheduler" "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) { func Test_watchdog(t *testing.T) {
@ -54,7 +54,7 @@ func Test_watchdog(t *testing.T) {
for _, tt := range testCases { for _, tt := range testCases {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
peer := &scheduler.PeerPacket_DestPeer{} peer := &scheduler.PeerPacket_DestPeer{}
pps := mocks.NewMockPeerPacketStream(ctrl) pps := mocks.NewMockScheduler_ReportPieceResultClient(ctrl)
watchdog := &synchronizerWatchdog{ watchdog := &synchronizerWatchdog{
done: make(chan struct{}), done: make(chan struct{}),
mainPeer: atomic.Value{}, mainPeer: atomic.Value{},