update tests

This commit is contained in:
banteg 2020-01-28 16:54:19 +07:00
parent bdcc6691eb
commit ee7aee83a5
2 changed files with 4 additions and 4 deletions

0
tests/__init__.py Normal file
View File

View File

@ -1,23 +1,23 @@
from eth_abi import encode_abi, decode_abi
from multicall import signature
from multicall import Signature
args = ((1, 2, 3), '0x' + 'f' * 40, b'data')
types = ['uint256[]', 'address', 'bytes']
def test_signature_parsing():
sig = signature.Signature('aggregate((address,bytes)[])(uint256,bytes[])')
sig = Signature('aggregate((address,bytes)[])(uint256,bytes[])')
assert sig.parts == ['aggregate', '((address,bytes)[])', '(uint256,bytes[])']
assert sig.input_types == '((address,bytes)[])'
assert sig.output_types == '(uint256,bytes[])'
def test_signature_encoding():
sig = signature.Signature('test(uint256[],address,bytes)()')
sig = Signature('test(uint256[],address,bytes)()')
assert sig.encode_data(args) == sig.fourbyte + encode_abi(types, args)
def test_signature_decoding():
sig = signature.Signature('test()(uint256[],address,bytes)')
sig = Signature('test()(uint256[],address,bytes)')
data = encode_abi(types, args)
assert sig.decode_data(data) == args