fix account copy issue

ethereum/core/src/main/java/org/hyperledger/besu/ethereum/trie/diffbased/bonsai/BonsaiAccount.java

@@ -82,6 +82,7 @@ public BonsaiAccount(
         toCopy.nonce,
         toCopy.balance,
         toCopy.codeHash,
+        toCopy.code,
         mutable);
     this.storageRoot = toCopy.storageRoot;
     updatedStorage.putAll(toCopy.updatedStorage);
@@ -96,6 +97,7 @@ public BonsaiAccount(
         tracked.getNonce(),
         tracked.getBalance(),
         tracked.getCodeHash(),
+        tracked.getCode(),
         true);
     this.storageRoot = Hash.EMPTY_TRIE_HASH;
     updatedStorage.putAll(tracked.getUpdatedStorage());

ethereum/core/src/main/java/org/hyperledger/besu/ethereum/trie/diffbased/common/DiffBasedAccount.java

@@ -60,6 +60,25 @@ public DiffBasedAccount(
     this.immutable = !mutable;
   }
 
+  public DiffBasedAccount(
+      final DiffBasedWorldView context,
+      final Address address,
+      final Hash addressHash,
+      final long nonce,
+      final Wei balance,
+      final Hash codeHash,
+      final Bytes code,
+      final boolean mutable) {
+    this.context = context;
+    this.address = address;
+    this.addressHash = addressHash;
+    this.nonce = nonce;
+    this.balance = balance;
+    this.codeHash = codeHash;
+    this.code = code;
+    this.immutable = !mutable;
+  }
+
   public DiffBasedAccount(
       final DiffBasedWorldView context,
       final Address address,

ethereum/core/src/test/java/org/hyperledger/besu/ethereum/trie/diffbased/bonsai/BonsaiAccountTest.java

@@ -0,0 +1,74 @@
+/*
+ * Copyright Hyperledger Besu Contributors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+package org.hyperledger.besu.ethereum.trie.diffbased.bonsai;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import org.hyperledger.besu.datatypes.Address;
+import org.hyperledger.besu.datatypes.Hash;
+import org.hyperledger.besu.datatypes.Wei;
+import org.hyperledger.besu.ethereum.trie.diffbased.bonsai.worldview.BonsaiWorldState;
+import org.hyperledger.besu.evm.worldstate.UpdateTrackingAccount;
+
+import org.apache.tuweni.bytes.Bytes;
+import org.apache.tuweni.units.bigints.UInt256;
+import org.junit.jupiter.api.Test;
+import org.mockito.Mock;
+
+public class BonsaiAccountTest {
+
+  @Mock BonsaiWorldState bonsaiWorldState;
+
+  @Test
+  void shouldCopyTrackedBonsaiAccountCorrectly() {
+    final BonsaiAccount trackedAccount =
+        new BonsaiAccount(
+            bonsaiWorldState,
+            Address.ZERO,
+            Hash.hash(Address.ZERO),
+            0,
+            Wei.ONE,
+            Hash.EMPTY_TRIE_HASH,
+            Hash.EMPTY,
+            true);
+    trackedAccount.setCode(Bytes.of(1));
+    final UpdateTrackingAccount<BonsaiAccount> bonsaiAccountUpdateTrackingAccount =
+        new UpdateTrackingAccount<>(trackedAccount);
+    bonsaiAccountUpdateTrackingAccount.setStorageValue(UInt256.ONE, UInt256.ONE);
+
+    final BonsaiAccount expectedAccount = new BonsaiAccount(trackedAccount, bonsaiWorldState, true);
+    expectedAccount.setStorageValue(UInt256.ONE, UInt256.ONE);
+    assertThat(new BonsaiAccount(bonsaiWorldState, bonsaiAccountUpdateTrackingAccount))
+        .isEqualToComparingFieldByField(expectedAccount);
+  }
+
+  @Test
+  void shouldCopyBonsaiAccountCorrectly() {
+    final BonsaiAccount account =
+        new BonsaiAccount(
+            bonsaiWorldState,
+            Address.ZERO,
+            Hash.hash(Address.ZERO),
+            0,
+            Wei.ONE,
+            Hash.EMPTY_TRIE_HASH,
+            Hash.EMPTY,
+            true);
+    account.setCode(Bytes.of(1));
+    account.setStorageValue(UInt256.ONE, UInt256.ONE);
+    assertThat(new BonsaiAccount(account, bonsaiWorldState, true))
+        .isEqualToComparingFieldByField(account);
+  }
+}