From 7cfd431a2bd38708d37739d20af94a5739b9aca4 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Tue, 24 Jan 2023 15:13:24 -0500 Subject: [PATCH] Yield `Services` in `open_test_pikerd()` fixture --- tests/conftest.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index d781cf61..2cfaad7a 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -7,6 +7,9 @@ from piker import ( # log, config, ) +from piker._daemon import ( + Services, +) def pytest_addoption(parser): @@ -142,10 +145,10 @@ async def _open_test_pikerd( maybe_open_pikerd( registry_addr=reg_addr, **kwargs, - ) as pikerd_portal, + ) as service_manager, ): # this proc/actor is the pikerd - assert pikerd_portal is None + assert service_manager is Services async with tractor.wait_for_actor( 'pikerd', @@ -157,6 +160,7 @@ async def _open_test_pikerd( raddr[0], raddr[1], portal, + service_manager, )