fix: DualVMToken totalSupply

kkrt-labs · Nov 4, 2024 · 171db61 · 171db61
1 parent 6cbc820
commit 171db61
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/solidity_contracts/src/CairoPrecompiles/DualVmToken.sol b/solidity_contracts/src/CairoPrecompiles/DualVmToken.sol
@@ -79,7 +79,8 @@ contract DualVmToken {
 
     function totalSupply() external view returns (uint256) {
         bytes memory returnData = starknetToken.staticcallCairo("total_supply");
-        return abi.decode(returnData, (uint256));
+        (uint128 valueLow, uint128 valueHigh) = abi.decode(returnData, (uint128, uint128));
+        return uint256(valueLow) + (uint256(valueHigh) << 128);
     }
 
     /// @dev This function is used to get the balance of an evm account

diff --git a/tests/end_to_end/CairoPrecompiles/test_dual_vm_token.py b/tests/end_to_end/CairoPrecompiles/test_dual_vm_token.py
@@ -12,7 +12,7 @@
 @pytest_asyncio.fixture(scope="function")
 async def starknet_token(owner):
     address = await deploy_starknet(
-        "StarknetToken", int(1e18), owner.starknet_contract.address
+        "StarknetToken", int(2**256 - 1), owner.starknet_contract.address
     )
     return get_contract_starknet("StarknetToken", address=address)