Function test_docker_deploy

integration-tests/test_docker_deploy.py:30–123

Deploy and run a flow in a Docker container with runtime package installation using uv. Demonstrates using EXTRA_PIP_PACKAGES to install dependencies at runtime.

()

def test_docker_deploy():
    """
    Deploy and run a flow in a Docker container with runtime package installation using uv.
    Demonstrates using EXTRA_PIP_PACKAGES to install dependencies at runtime.
    """
    try:
        subprocess.check_call(
            [
                "uv",
                "run",
                "--isolated",
                "prefect",
                "work-pool",
                "create",
                "test-docker-pool",
                "-t",
                "docker",
            ],
            stdout=sys.stdout,
            stderr=sys.stderr,
        )

        dockerfile = Path("docker-deploy.Dockerfile")

        dockerfile.write_text(
            dedent(
                """
                FROM prefecthq/prefect:3-latest
                COPY integration-tests/test_docker_deploy.py /opt/prefect/integration-tests/test_docker_deploy.py
                """
            )
        )

        flow_that_needs_pandas.deploy(
            name="docker-demo-deployment",
            work_pool_name="test-docker-pool",
            job_variables={
                "env": {"EXTRA_PIP_PACKAGES": "pandas"},
                "image": "prefect-integration-test-docker-deploy",
            },
            image=DockerImage(
                name="prefect-integration-test-docker-deploy",
                dockerfile=str(dockerfile),
            ),
            build=True,
            push=False,
        )

        dockerfile.unlink()

        flow_run = run_deployment(
            "flow-that-needs-pandas/docker-demo-deployment",
            timeout=0,
        )

        # Execute the flow run
        subprocess.check_call(
            [