diff --git a/tests/integration/test_runner_manager_openstack.py b/tests/integration/test_runner_manager_openstack.py index cb88d84ba..ba4d71b10 100644 --- a/tests/integration/test_runner_manager_openstack.py +++ b/tests/integration/test_runner_manager_openstack.py @@ -7,7 +7,7 @@ import json from pathlib import Path from secrets import token_hex -from typing import Iterator +from typing import AsyncGenerator, Iterator import pytest import pytest_asyncio @@ -105,7 +105,7 @@ async def openstack_runner_manager_fixture( proxy_config: ProxyConfig, runner_label: str, openstack_connection: OpenstackConnection, -) -> OpenStackRunnerManager: +) -> AsyncGenerator[OpenStackRunnerManager, None]: """Create OpenstackRunnerManager instance. The prefix args of OpenstackRunnerManager set to app_name to let openstack_connection_fixture @@ -133,7 +133,7 @@ async def openstack_runner_manager_fixture( ssh_debug_connections=None, repo_policy_compliance=None, ) - return OpenStackRunnerManager( + yield OpenStackRunnerManager( app_name, f"{app_name}-0", cloud_config, server_config, runner_config, service_config ) @@ -144,13 +144,13 @@ async def runner_manager_fixture( token: str, github_path: GitHubPath, log_dir_base_path: dict[str, Path], -) -> RunnerManager: +) -> AsyncGenerator[RunnerManager, None]: """Get RunnerManager instance. Import of log_dir_base_path to monkeypatch the runner logs path with tmp_path. """ config = RunnerManagerConfig(token, github_path) - return RunnerManager("test_runner", openstack_runner_manager, config) + yield RunnerManager("test_runner", openstack_runner_manager, config) @pytest_asyncio.fixture(scope="function", name="runner_manager_with_one_runner")