def test_docker_agent_shutdown_terminates_child_processes(monkeypatch, api): monkeypatch.setattr("prefect.agent.agent.Client", MagicMock()) proc = MagicMock(is_alive=MagicMock(return_value=True)) agent = DockerAgent(show_flow_logs=True) agent.processes = [proc] agent.on_shutdown() assert proc.is_alive.called assert proc.terminate.called
def test_docker_agent_shutdown_terminates_child_processes(monkeypatch, cloud_api): monkeypatch.setattr("prefect.agent.agent.Client", MagicMock()) api = MagicMock() api.ping.return_value = True api.create_container.return_value = {"Id": "container_id"} monkeypatch.setattr( "prefect.agent.docker.agent.DockerAgent._get_docker_client", MagicMock(return_value=api), ) proc = MagicMock(is_alive=MagicMock(return_value=True)) agent = DockerAgent(show_flow_logs=True) agent.processes = [proc] agent.on_shutdown() assert proc.is_alive.called assert proc.terminate.called