fix: resolves #2067

wevm · Apr 3, 2024 · 8a8fab2 · 8a8fab2
1 parent f539b0d
commit 8a8fab2
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 13 deletions.
diff --git a/.changeset/hot-walls-dream.md b/.changeset/hot-walls-dream.md
@@ -0,0 +1,5 @@
+---
+"viem": patch
+---
+
+Fixed parsing of `stateOverrides`.
diff --git a/src/actions/public/call.test.ts b/src/actions/public/call.test.ts
@@ -1105,7 +1105,7 @@ describe('parsing overrides', () => {
 
     expect(parseAccountStateOverride(stateOverride)).toMatchInlineSnapshot(`
       {
-        "balance": "0x0000000000000000000000000000000000000000000000000000000000000001",
+        "balance": "0x1",
       }
     `)
 
@@ -1127,7 +1127,7 @@ describe('parsing overrides', () => {
 
     expect(parseAccountStateOverride(stateOverride)).toMatchInlineSnapshot(`
       {
-        "nonce": "0x0000000000000001",
+        "nonce": "0x1",
       }
     `)
 

diff --git a/src/actions/public/call.ts b/src/actions/public/call.ts
@@ -404,18 +404,12 @@ export function parseAccountStateOverride(
 ): RpcAccountStateOverride {
   const { balance, nonce, state, stateDiff, code } = args
   const rpcAccountStateOverride: RpcAccountStateOverride = {}
-  if (code !== undefined) {
-    rpcAccountStateOverride.code = code
-  }
-  if (balance !== undefined) {
-    rpcAccountStateOverride.balance = numberToHex(balance, { size: 32 })
-  }
-  if (nonce !== undefined) {
-    rpcAccountStateOverride.nonce = numberToHex(nonce, { size: 8 })
-  }
-  if (state !== undefined) {
+  if (code !== undefined) rpcAccountStateOverride.code = code
+  if (balance !== undefined)
+    rpcAccountStateOverride.balance = numberToHex(balance)
+  if (nonce !== undefined) rpcAccountStateOverride.nonce = numberToHex(nonce)
+  if (state !== undefined)
     rpcAccountStateOverride.state = parseStateMapping(state)
-  }
   if (stateDiff !== undefined) {
     if (rpcAccountStateOverride.state) throw new StateAssignmentConflictError()
     rpcAccountStateOverride.stateDiff = parseStateMapping(stateDiff)