Skip to content

Commit 55f942b

Browse files
committed
add transaction task to EVMMethods
- minor refactoring (renaming) of debug_trace tasks - add trace_transaction task to EVMMethods
1 parent 5709350 commit 55f942b

File tree

4 files changed

+21
-15
lines changed

4 files changed

+21
-15
lines changed

chainbench/profile/evm/debug_trace.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,29 +16,29 @@ def debug_trace_transaction_task(self):
1616
self.make_call(
1717
name="debug_trace_transaction",
1818
method="debug_traceTransaction",
19-
params=self._trace_transaction_params_factory(get_rng()),
19+
params=self._debug_trace_transaction_params_factory(get_rng()),
2020
),
2121

2222
@task(41)
2323
def debug_trace_call_task(self):
2424
self.make_call(
2525
name="debug_trace_call",
2626
method="debug_traceCall",
27-
params=self._trace_call_params_factory(get_rng()),
27+
params=self._debug_trace_call_params_factory(get_rng()),
2828
),
2929

3030
@task(36)
3131
def debug_trace_block_by_number_task(self):
3232
self.make_call(
3333
name="debug_trace_block_by_number",
3434
method="debug_traceBlockByNumber",
35-
params=self._trace_block_by_number_params_factory(),
35+
params=self._debug_trace_block_by_number_params_factory(),
3636
),
3737

3838
@task(1)
3939
def debug_trace_block_by_hash_task(self):
4040
self.make_call(
4141
name="debug_trace_block_by_hash",
4242
method="debug_traceBlockByHash",
43-
params=self._trace_block_by_hash_params_factory(get_rng()),
43+
params=self._debug_trace_block_by_hash_params_factory(get_rng()),
4444
),

chainbench/profile/evm/heavy.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def debug_trace_transaction_task(self):
1717
self.make_call(
1818
name="debug_trace_transaction",
1919
method="debug_traceTransaction",
20-
params=self._trace_transaction_params_factory(get_rng()),
20+
params=self._debug_trace_transaction_params_factory(get_rng()),
2121
),
2222

2323
@task
@@ -121,15 +121,15 @@ def debug_trace_block_by_number_task(self):
121121
self.make_call(
122122
name="debug_trace_block_by_number",
123123
method="debug_traceBlockByNumber",
124-
params=self._trace_block_by_number_params_factory(),
124+
params=self._debug_trace_block_by_number_params_factory(),
125125
)
126126

127127
@task
128128
def debug_trace_block_by_hash_task(self):
129129
self.make_call(
130130
name="debug_trace_block_by_hash",
131131
method="debug_traceBlockByHash",
132-
params=self._trace_block_by_hash_params_factory(get_rng()),
132+
params=self._debug_trace_block_by_hash_params_factory(get_rng()),
133133
)
134134

135135
@task

chainbench/profile/polygon/__init__.py

Whitespace-only changes.

chainbench/user/evm.py

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def _get_balance_params_factory(self, rng: RNG) -> list[Account | str]:
4040
"latest",
4141
]
4242

43-
def _trace_call_params_factory(self, rng: RNG) -> list[dict | BlockNumber]:
43+
def _debug_trace_call_params_factory(self, rng: RNG) -> list[dict | BlockNumber]:
4444
tx_data = self.test_data.get_random_tx(rng)
4545
tx_param = {
4646
"to": tx_data["to"],
@@ -67,13 +67,13 @@ def _trace_call_params_factory(self, rng: RNG) -> list[dict | BlockNumber]:
6767
{"tracer": "callTracer", "timeout": self._default_trace_timeout},
6868
]
6969

70-
def _trace_block_by_number_params_factory(self) -> list[str | dict]:
70+
def _debug_trace_block_by_number_params_factory(self) -> list[str | dict]:
7171
return [
7272
"latest",
7373
{"tracer": "callTracer", "timeout": self._default_trace_timeout},
7474
]
7575

76-
def _trace_block_by_hash_params_factory(self, rng: RNG) -> list[BlockHash | dict]:
76+
def _debug_trace_block_by_hash_params_factory(self, rng: RNG) -> list[BlockHash | dict]:
7777
return [
7878
self.test_data.get_random_block_hash(rng),
7979
{"tracer": "callTracer", "timeout": self._default_trace_timeout},
@@ -92,7 +92,7 @@ def _trace_replay_transaction_params_factory(self, rng: RNG) -> list[TxHash | li
9292
["vmTrace", "trace", "stateDiff"],
9393
]
9494

95-
def _trace_transaction_params_factory(self, rng: RNG) -> list[TxHash | dict]:
95+
def _debug_trace_transaction_params_factory(self, rng: RNG) -> list[TxHash | dict]:
9696
return [
9797
self.test_data.get_random_tx_hash(rng),
9898
{"tracer": "callTracer", "timeout": self._default_trace_timeout},
@@ -295,25 +295,25 @@ def eth_syncing_task(self) -> None:
295295
def debug_trace_block_by_hash_task(self) -> None:
296296
self.make_call(
297297
method="debug_traceBlockByHash",
298-
params=self._trace_block_by_hash_params_factory(self.rng.get_rng()),
298+
params=self._debug_trace_block_by_hash_params_factory(self.rng.get_rng()),
299299
)
300300

301301
def debug_trace_block_by_number_task(self) -> None:
302302
self.make_call(
303303
method="debug_traceBlockByNumber",
304-
params=self._trace_block_by_number_params_factory(),
304+
params=self._debug_trace_block_by_number_params_factory(),
305305
)
306306

307307
def debug_trace_call_task(self) -> None:
308308
self.make_call(
309309
method="debug_traceCall",
310-
params=self._trace_call_params_factory(self.rng.get_rng()),
310+
params=self._debug_trace_call_params_factory(self.rng.get_rng()),
311311
)
312312

313313
def debug_trace_transaction_task(self) -> None:
314314
self.make_call(
315315
method="debug_traceTransaction",
316-
params=self._trace_transaction_params_factory(self.rng.get_rng()),
316+
params=self._debug_trace_transaction_params_factory(self.rng.get_rng()),
317317
)
318318

319319
def net_listening_task(self) -> None:
@@ -349,6 +349,12 @@ def trace_replay_transaction_task(self) -> None:
349349
params=self._trace_replay_transaction_params_factory(self.rng.get_rng()),
350350
)
351351

352+
def trace_transaction_task(self) -> None:
353+
self.make_call(
354+
method="trace_transaction",
355+
params=[self.test_data.get_random_tx_hash(self.rng.get_rng())],
356+
)
357+
352358
def web3_client_version_task(self) -> None:
353359
self.make_call(
354360
method="web3_clientVersion",

0 commit comments

Comments
 (0)