File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 11import pytest
22
3+ from hexbytes import HexBytes
4+
35from vyper import compiler
46from vyper .exceptions import ArgumentException , StateAccessViolation
57from vyper .functions import get_create_forwarder_to_bytecode
@@ -63,15 +65,15 @@ def create_and_return_forwarder(inp: address) -> address:
6365 assert c2 .create_and_call_returnten (c .address ) == 10
6466 c2 .create_and_call_returnten (c .address , transact = {})
6567
66- expected_forwarder_code_mask = get_create_forwarder_to_bytecode ()[ 12 :]
68+ _ , preamble , callcode = get_create_forwarder_to_bytecode ()
6769
6870 c3 = c2 .create_and_return_forwarder (c .address , call = {})
6971 c2 .create_and_return_forwarder (c .address , transact = {})
7072
7173 c3_contract_code = w3 .toBytes (w3 .eth .getCode (c3 ))
7274
73- assert c3_contract_code [:14 ] == expected_forwarder_code_mask [: 14 ]
74- assert c3_contract_code [35 :] == expected_forwarder_code_mask [ 35 :]
75+ assert c3_contract_code [:10 ] == HexBytes ( preamble )
76+ assert c3_contract_code [- 15 :] == HexBytes ( callcode )
7577
7678 print ("Passed forwarder test" )
7779 # TODO: This one is special
You can’t perform that action at this time.
0 commit comments