Skip to content

Commit 1a43cf0

Browse files
calvingilescopybara-github
authored andcommitted
Copybara import of the project:
-- 73826d2 by Calvin Giles <calvin.giles@trademe.co.nz>: fix: Enable InMemoryRunner to be used in async tests COPYBARA_INTEGRATE_REVIEW=#868 from calvingiles:enable-test-runner-in-async fb9033e PiperOrigin-RevId: 763233472
1 parent a380d70 commit 1a43cf0

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

tests/unittests/testing_utils.py

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -177,17 +177,13 @@ def __init__(
177177
@property
178178
def session(self) -> Session:
179179
if not self.session_id:
180-
session = asyncio.run(
181-
self.runner.session_service.create_session(
182-
app_name='test_app', user_id='test_user'
183-
)
180+
session = self.runner.session_service.create_session_sync(
181+
app_name='test_app', user_id='test_user'
184182
)
185183
self.session_id = session.id
186184
return session
187-
return asyncio.run(
188-
self.runner.session_service.get_session(
189-
app_name='test_app', user_id='test_user', session_id=self.session_id
190-
)
185+
return self.runner.session_service.get_session_sync(
186+
app_name='test_app', user_id='test_user', session_id=self.session_id
191187
)
192188

193189
def run(self, new_message: types.ContentUnion) -> list[Event]:
@@ -199,6 +195,16 @@ def run(self, new_message: types.ContentUnion) -> list[Event]:
199195
)
200196
)
201197

198+
async def run_async(self, new_message: types.ContentUnion) -> list[Event]:
199+
events = []
200+
async for event in self.runner.run_async(
201+
user_id=self.session.user_id,
202+
session_id=self.session.id,
203+
new_message=get_user_content(new_message),
204+
):
205+
events.append(event)
206+
return events
207+
202208
def run_live(self, live_request_queue: LiveRequestQueue) -> list[Event]:
203209
collected_responses = []
204210

0 commit comments

Comments
 (0)