chore(dot/sync): narrow the type of syncTask.request

ChainSafe · Oct 22, 2024 · 0588ed0 · 0588ed0
1 parent 3da46d9
commit 0588ed0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dot/sync/fullsync.go b/dot/sync/fullsync.go
@@ -53,7 +53,7 @@ type importer interface {
 type syncTask struct {
 	id           TaskID
 	requestMaker network.RequestMaker
-	request      messages.P2PMessage
+	request      *messages.BlockRequestMessage
 }
 
 func (s *syncTask) ID() TaskID {
@@ -446,7 +446,7 @@ func validateResult(result TaskResult, badBlocks []string) (repChange *Change,
 		return
 	}
 
-	request := task.request.(*messages.BlockRequestMessage)
+	request := task.request
 	response := result.Result.(*messages.BlockResponseMessage)
 	if request.Direction == messages.Descending {
 		// reverse blocks before pre-validating and placing in ready queue

diff --git a/dot/sync/fullsync_test.go b/dot/sync/fullsync_test.go
@@ -70,7 +70,7 @@ func TestFullSyncNextActions(t *testing.T) {
 		task, err := fs.NextActions()
 		require.NoError(t, err)
 
-		request := task[0].(*syncTask).request.(*messages.BlockRequestMessage)
+		request := task[0].(*syncTask).request
 		require.Equal(t, uint(1), request.StartingBlock.RawValue())
 		require.Equal(t, uint32(128), *request.Max)
 	})