Skip to content

Commit f80c393

Browse files
committed
test: fixup forwarder test
1 parent 7919574 commit f80c393

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/parser/functions/test_raw_call.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
import pytest
22

3+
from hexbytes import HexBytes
4+
35
from vyper import compiler
46
from vyper.exceptions import ArgumentException, StateAccessViolation
57
from 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

0 commit comments

Comments
 (0)