Skip to content

Commit 9fddf51

Browse files
committed
Add default run method
1 parent 3a678d9 commit 9fddf51

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

test/asynchronous/utils_spec_runner.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ def __init__(self, name, *args, **kwargs):
6868
self.name = name
6969
self.stopped = False
7070
self.task = None
71+
if "target" in kwargs:
72+
self.target = kwargs["target"]
7173

7274
if not _IS_SYNC:
7375

@@ -81,6 +83,10 @@ async def join(self, timeout: float | None = 0): # type: ignore[override]
8183
def is_alive(self):
8284
return not self.stopped
8385

86+
async def run(self):
87+
if self.target:
88+
await self.target()
89+
8490

8591
class SpecRunnerTask(ConcurrentRunner):
8692
def __init__(self, name):

test/utils_spec_runner.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ def __init__(self, name, *args, **kwargs):
6868
self.name = name
6969
self.stopped = False
7070
self.task = None
71+
if "target" in kwargs:
72+
self.target = kwargs["target"]
7173

7274
if not _IS_SYNC:
7375

@@ -81,6 +83,10 @@ def join(self, timeout: float | None = 0): # type: ignore[override]
8183
def is_alive(self):
8284
return not self.stopped
8385

86+
def run(self):
87+
if self.target:
88+
self.target()
89+
8490

8591
class SpecRunnerThread(ConcurrentRunner):
8692
def __init__(self, name):

0 commit comments

Comments
 (0)