crowetic/mev-inspect-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Run precommit

Browse Source
This commit is contained in:
Leo Arias 2021-10-11 17:51:38 +00:00
parent a67769cea3
commit ba73f58396
3 changed files with 40 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mev_inspect/decode.py
View File

@ -1,6 +1,7 @@
import eth_utils.abi
from typing import Dict, Optional
import eth_utils.abi
from hexbytes import HexBytes
from eth_abi import decode_abi
from eth_abi.exceptions import InsufficientDataBytes, NonEmptyPaddingBytes
@ -28,7 +29,8 @@ class ABIDecoder:
names = [input.name for input in func.inputs]
types = [
input.type if input.type != 'tuple'
input.type
if input.type != "tuple"
else eth_utils.abi.collapse_if_tuple(input.dict())
for input in func.inputs
]

6
mev_inspect/schemas/abi.py
View File

@ -48,9 +48,11 @@ class ABIFunctionDescription(BaseModel):
def get_signature(self) -> str:
joined_input_types = ",".join(
input.type if input.type != "tuple"
input.type
if input.type != "tuple"
else eth_utils.abi.collapse_if_tuple(input.dict())
for input in self.inputs)
for input in self.inputs
)
return f"{self.name}({joined_input_types})"

54
tests/test_decode.py
View File

@ -7,14 +7,16 @@ from mev_inspect.schemas import abi
def test_decode_function_with_simple_argument():
test_function_name = "testFunction"
test_parameter_name = "testParameter"
test_abi = pydantic.parse_obj_as(abi.ABI, [{
"name": test_function_name,
"type": "function",
"inputs": [{
"name": test_parameter_name,
"type": "uint256"
}]
}])
test_abi = pydantic.parse_obj_as(
abi.ABI,
[
{
"name": test_function_name,
"type": "function",
"inputs": [{"name": test_parameter_name, "type": "uint256"}],
}
],
)
# 4byte signature of the test function.
# https://www.4byte.directory/signatures/?bytes4_signature=0x350c530b
test_function_selector = "350c530b"
@ -23,7 +25,8 @@ def test_decode_function_with_simple_argument():
)
abi_decoder = decode.ABIDecoder(test_abi)
call_data = abi_decoder.decode(
"0x" + test_function_selector + test_function_argument)
"0x" + test_function_selector + test_function_argument
)
assert call_data.function_name == test_function_name
assert call_data.function_signature == "testFunction(uint256)"
assert call_data.inputs == {test_parameter_name: 1}
@ -33,18 +36,24 @@ def test_decode_function_with_tuple_argument():
test_function_name = "testFunction"
test_tuple_name = "testTuple"
test_parameter_name = "testParameter"
test_abi = pydantic.parse_obj_as(abi.ABI, [{
"name": test_function_name,
"type": "function",
"inputs": [{
"name": test_tuple_name,
"type": "tuple",
"components": [{
"name": test_parameter_name,
"type": "uint256"
}]
}]
}])
test_abi = pydantic.parse_obj_as(
abi.ABI,
[
{
"name": test_function_name,
"type": "function",
"inputs": [
{
"name": test_tuple_name,
"type": "tuple",
"components": [
{"name": test_parameter_name, "type": "uint256"}
],
}
],
}
],
)
# 4byte signature of the test function.
# https://www.4byte.directory/signatures/?bytes4_signature=0x98568079
test_function_selector = "98568079"
@ -53,7 +62,8 @@ def test_decode_function_with_tuple_argument():
)
abi_decoder = decode.ABIDecoder(test_abi)
call_data = abi_decoder.decode(
"0x" + test_function_selector + test_function_argument)
"0x" + test_function_selector + test_function_argument
)
assert call_data.function_name == test_function_name
assert call_data.function_signature == "testFunction((uint256))"
assert call_data.inputs == {test_tuple_name: (1,)}