diff --git a/tests/helpers.py b/tests/helpers.py index 72afdef..540a327 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -17,8 +17,6 @@ def make_transfer_trace( to_address: str, token_address: str, amount: int, - action={}, - subtraces=0, ): return CallTrace( transaction_hash=transaction_hash, @@ -33,8 +31,8 @@ def make_transfer_trace( "amount": amount, }, block_hash=str(block_number), - action=action, - subtraces=subtraces, + action={}, + subtraces=0.0, ) @@ -47,16 +45,14 @@ def make_swap_trace( abi_name: str, recipient_address: str, recipient_input_key: str, - action={}, - subtraces=0, ): return DecodedCallTrace( transaction_hash=transaction_hash, block_number=block_number, type=TraceType.call, trace_address=trace_address, - action=action, - subtraces=subtraces, + action={}, + subtraces=0, classification=Classification.swap, from_address=from_address, to_address=pool_address, @@ -70,15 +66,13 @@ def make_unknown_trace( block_number: int, transaction_hash: str, trace_address: List[int], - action={}, - subtraces=0, ): return ClassifiedTrace( block_number=block_number, transaction_hash=transaction_hash, trace_address=trace_address, - action=action, - subtraces=subtraces, + action={}, + subtraces=0, block_hash=str(block_number), type=TraceType.call, classification=Classification.unknown, @@ -89,13 +83,13 @@ def make_many_unknown_traces( block_number: int, transaction_hash: str, trace_addresses: List[List[int]], - action={}, - subtraces=0, ) -> List[ClassifiedTrace]: return [ make_unknown_trace( - block_number, transaction_hash, trace_address, action, subtraces + block_number, + transaction_hash, + trace_address, ) for trace_address in trace_addresses ]