Fix trace pending txns

NethermindEth · Jan 15, 2025 · 88f0cb3 · 88f0cb3
1 parent efe2165
commit 88f0cb3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/rpc/trace.go b/rpc/trace.go
@@ -142,6 +142,18 @@ func (h *Handler) traceTransaction(ctx context.Context, hash *felt.Felt) (*vm.Tr
 	httpHeader.Set(ExecutionStepsHeader, "0")
 
 	if err != nil {
+		if pendingB := h.syncReader.PendingBlock(); pendingB != nil {
+			txIndex := slices.IndexFunc(pendingB.Transactions, func(tx core.Transaction) bool {
+				return tx.Hash().Equal(hash)
+			})
+			if txIndex != -1 {
+				traceResults, header, traceBlockErr := h.traceBlockTransactions(ctx, pendingB)
+				if traceBlockErr != nil {
+					return nil, header, traceBlockErr
+				}
+				return traceResults[txIndex].TraceRoot, header, nil
+			}
+		}
 		return nil, httpHeader, ErrTxnHashNotFound
 	}
 

diff --git a/rpc/trace_test.go b/rpc/trace_test.go
@@ -87,8 +87,8 @@ func TestTraceTransaction(t *testing.T) {
 
 	t.Run("not found", func(t *testing.T) {
 		hash := utils.HexToFelt(t, "0xBBBB")
-		// Receipt() returns error related to db
 		mockReader.EXPECT().Receipt(hash).Return(nil, nil, uint64(0), db.ErrKeyNotFound)
+		mockSyncReader.EXPECT().PendingBlock().Return(nil)
 
 		trace, httpHeader, err := handler.TraceTransaction(context.Background(), *hash)
 		assert.Nil(t, trace)