Yield `Services` in `open_test_pikerd()` fixture

samplerd_service
Tyler Goodlet 2023-01-24 15:13:24 -05:00
parent 61e20a86cc
commit 7cfd431a2b
1 changed files with 6 additions and 2 deletions

View File

@ -7,6 +7,9 @@ from piker import (
# log, # log,
config, config,
) )
from piker._daemon import (
Services,
)
def pytest_addoption(parser): def pytest_addoption(parser):
@ -142,10 +145,10 @@ async def _open_test_pikerd(
maybe_open_pikerd( maybe_open_pikerd(
registry_addr=reg_addr, registry_addr=reg_addr,
**kwargs, **kwargs,
) as pikerd_portal, ) as service_manager,
): ):
# this proc/actor is the pikerd # this proc/actor is the pikerd
assert pikerd_portal is None assert service_manager is Services
async with tractor.wait_for_actor( async with tractor.wait_for_actor(
'pikerd', 'pikerd',
@ -157,6 +160,7 @@ async def _open_test_pikerd(
raddr[0], raddr[0],
raddr[1], raddr[1],
portal, portal,
service_manager,
) )