implements test by priority

Signed-off-by: Sagar Khandagre <sagar.khandagre998@gmail.com>

Author: sagarkhandagre998

ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/jsonrpc/internal/parameters/TransactionTraceParamsTest.java

@@ -85,4 +85,26 @@ public void nonOpcodeTracerShouldRespectExplicitDisableMemory() throws Exception
         .describedAs("explicit disableMemory=true should be respected")
         .isFalse();
   }
+
+  @Test
+  public void enableReturnDataTrueShouldSetTraceReturnData() throws Exception {
+    final TransactionTraceParams params =
+        MAPPER.readValue("{\"enableReturnData\": true}", TransactionTraceParams.class);
+    final OpCodeTracerConfig config = params.traceOptions().opCodeTracerConfig();
+
+    assertThat(config.traceReturnData())
+        .describedAs("enableReturnData: true should set traceReturnData to true")
+        .isTrue();
+  }
+
+  @Test
+  public void missingEnableReturnDataShouldDefaultToFalse() throws Exception {
+    final TransactionTraceParams params =
+        MAPPER.readValue("{}", TransactionTraceParams.class);
+    final OpCodeTracerConfig config = params.traceOptions().opCodeTracerConfig();
+
+    assertThat(config.traceReturnData())
+        .describedAs("traceReturnData should default to false when enableReturnData is absent")
+        .isFalse();
+  }
 }

ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/jsonrpc/internal/results/StructLogTest.java

@@ -246,4 +246,64 @@ public void testToCompactHexWithLargeData() {
     String result = StructLog.toCompactHex(bytes, true);
     assertEquals("0x102030405060708090a", result, "Expected correct hex output for large data");
   }
+
+  // --- returnData tests ---
+
+  @Test
+  public void returnDataShouldBePresentWhenCaptured() {
+    when(traceFrame.getDepth()).thenReturn(0);
+    when(traceFrame.getGasRemaining()).thenReturn(0L);
+    when(traceFrame.getGasCost()).thenReturn(OptionalLong.empty());
+    when(traceFrame.getGasRefund()).thenReturn(0L);
+    when(traceFrame.getMemory()).thenReturn(Optional.empty());
+    when(traceFrame.getOpcode()).thenReturn("PUSH1");
+    when(traceFrame.getPc()).thenReturn(0);
+    when(traceFrame.getStack()).thenReturn(Optional.empty());
+    when(traceFrame.getStorage()).thenReturn(Optional.empty());
+    when(traceFrame.getRevertReason()).thenReturn(Optional.empty());
+    when(traceFrame.getReturnData()).thenReturn(Optional.of(Bytes.fromHexString("0xdeadbeef")));
+
+    final StructLog log = new StructLog(traceFrame);
+
+    assertThat(log.returnData()).isEqualTo("0xdeadbeef");
+  }
+
+  @Test
+  public void returnDataShouldBeNullWhenNotCaptured() {
+    when(traceFrame.getDepth()).thenReturn(0);
+    when(traceFrame.getGasRemaining()).thenReturn(0L);
+    when(traceFrame.getGasCost()).thenReturn(OptionalLong.empty());
+    when(traceFrame.getGasRefund()).thenReturn(0L);
+    when(traceFrame.getMemory()).thenReturn(Optional.empty());
+    when(traceFrame.getOpcode()).thenReturn("PUSH1");
+    when(traceFrame.getPc()).thenReturn(0);
+    when(traceFrame.getStack()).thenReturn(Optional.empty());
+    when(traceFrame.getStorage()).thenReturn(Optional.empty());
+    when(traceFrame.getRevertReason()).thenReturn(Optional.empty());
+    when(traceFrame.getReturnData()).thenReturn(Optional.empty());
+
+    final StructLog log = new StructLog(traceFrame);
+
+    assertThat(log.returnData()).isNull();
+  }
+
+  @Test
+  public void returnDataShouldBeAbsentFromJsonWhenNotCaptured() throws Exception {
+    when(traceFrame.getDepth()).thenReturn(0);
+    when(traceFrame.getGasRemaining()).thenReturn(0L);
+    when(traceFrame.getGasCost()).thenReturn(OptionalLong.empty());
+    when(traceFrame.getGasRefund()).thenReturn(0L);
+    when(traceFrame.getMemory()).thenReturn(Optional.empty());
+    when(traceFrame.getOpcode()).thenReturn("PUSH1");
+    when(traceFrame.getPc()).thenReturn(0);
+    when(traceFrame.getStack()).thenReturn(Optional.empty());
+    when(traceFrame.getStorage()).thenReturn(Optional.empty());
+    when(traceFrame.getRevertReason()).thenReturn(Optional.empty());
+    when(traceFrame.getReturnData()).thenReturn(Optional.empty());
+
+    final StructLog log = new StructLog(traceFrame);
+    final String json = objectMapper.writeValueAsString(log);
+
+    assertThat(json).doesNotContain("returnData");
+  }
 }

ethereum/core/src/test/java/org/hyperledger/besu/ethereum/vm/DebugOperationTracerTest.java

@@ -222,6 +222,41 @@ void shouldNotRecordStorageWhenDisabled() {
     assertThat(traceFrame.getStorage()).isEmpty();
   }
 
+  @Test
+  void shouldRecordReturnDataWhenEnabled() {
+    final MessageFrame frame = validMessageFrame();
+    frame.setReturnData(Bytes.fromHexString("0xdeadbeef"));
+    final TraceFrame traceFrame =
+        traceFrame(
+            frame,
+            OpCodeTracerConfigBuilder.createFrom(OpCodeTracerConfig.DEFAULT)
+                .traceStorage(false)
+                .traceMemory(false)
+                .traceStack(false)
+                .traceReturnData(true)
+                .build(),
+            false);
+    assertThat(traceFrame.getReturnData()).isPresent();
+    assertThat(traceFrame.getReturnData().get()).isEqualTo(Bytes.fromHexString("0xdeadbeef"));
+  }
+
+  @Test
+  void shouldNotRecordReturnDataWhenDisabled() {
+    final MessageFrame frame = validMessageFrame();
+    frame.setReturnData(Bytes.fromHexString("0xdeadbeef"));
+    final TraceFrame traceFrame =
+        traceFrame(
+            frame,
+            OpCodeTracerConfigBuilder.createFrom(OpCodeTracerConfig.DEFAULT)
+                .traceStorage(false)
+                .traceMemory(false)
+                .traceStack(false)
+                .traceReturnData(false)
+                .build(),
+            false);
+    assertThat(traceFrame.getReturnData()).isEmpty();
+  }
+
   @Test
   void shouldNotAddGasWhenDisabled() {
     final TraceFrame traceFrame =