Skip to content

Commit 3f5f763

Browse files
feat: add start time arg in devnet infra (#144)
1 parent e00b6b9 commit 3f5f763

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

python/test_utils/starknet_test_utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ def __init__(
100100
fork_network: Optional[str],
101101
fork_block: Optional[int],
102102
request_body_size_limit: Optional[int],
103+
start_time: Optional[int],
103104
):
104105
self.starknet = Starknet(
105106
port=port,
@@ -110,6 +111,7 @@ def __init__(
110111
fork_network=fork_network,
111112
fork_block=fork_block,
112113
request_body_size_limit=request_body_size_limit,
114+
start_time=start_time,
113115
)
114116
self.accounts = self.starknet.accounts
115117

@@ -128,6 +130,7 @@ def context_manager(
128130
fork_network: Optional[str] = None,
129131
fork_block: Optional[int] = None,
130132
request_body_size_limit: Optional[int] = None,
133+
start_time: Optional[int] = None,
131134
backoff: float = 0.1,
132135
):
133136
"""
@@ -146,6 +149,7 @@ def context_manager(
146149
fork_network=fork_network,
147150
fork_block=fork_block,
148151
request_body_size_limit=request_body_size_limit,
152+
start_time=start_time,
149153
)
150154
yield res
151155
return
@@ -180,6 +184,7 @@ def __init__(
180184
fork_network: Optional[str] = None,
181185
fork_block: Optional[int] = None,
182186
request_body_size_limit: Optional[int] = None,
187+
start_time: Optional[int] = None,
183188
):
184189
"""
185190
Runs starknet.
@@ -213,6 +218,9 @@ def __init__(
213218
if request_body_size_limit is not None:
214219
command.extend(["--request-body-size-limit", str(request_body_size_limit)])
215220

221+
if start_time is not None:
222+
command.extend(["--start-time", str(start_time)])
223+
216224
self.starknet_proc = subprocess.Popen(
217225
command,
218226
stdout=subprocess.DEVNULL,

0 commit comments

Comments
 (0)