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

Enforce data typpe for helpers

Browse Source
This commit is contained in:
Gui Heise 2021-08-17 12:57:16 -04:00
parent cfba66fb4d
commit 27b318dacd
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/helpers.py
View File

@ -1,7 +1,11 @@
from typing import List from typing import List
from mev_inspect.schemas.blocks import TraceType from mev_inspect.schemas.blocks import TraceType
from mev_inspect.schemas.classified_traces import Classification, ClassifiedTrace from mev_inspect.schemas.classified_traces import (
Classification,
ClassifiedTrace,
ClassifiedCallData,
)
def make_transfer_trace( def make_transfer_trace(
@ -13,7 +17,7 @@ def make_transfer_trace(
token_address: str, token_address: str,
amount: int, amount: int,
): ):
return ClassifiedTrace( return ClassifiedCallData(
transaction_hash=transaction_hash, transaction_hash=transaction_hash,
block_number=block_number, block_number=block_number,
trace_type=TraceType.call, trace_type=TraceType.call,
@ -38,7 +42,7 @@ def make_swap_trace(
recipient_address: str, recipient_address: str,
recipient_input_key: str, recipient_input_key: str,
): ):
return ClassifiedTrace( return ClassifiedCallData(
transaction_hash=transaction_hash, transaction_hash=transaction_hash,
block_number=block_number, block_number=block_number,
trace_type=TraceType.call, trace_type=TraceType.call,