def test_startup_probe_enabled(template: HelmTemplate, enabled: bool): deployment = create_simple_user_deployment("foo") deployment.startupProbe = kubernetes.StartupProbe.construct(enabled=enabled) helm_values = DagsterHelmValues.construct( dagsterUserDeployments=UserDeployments.construct(deployments=[deployment]) ) dagster_user_deployment = template.render(helm_values) assert len(dagster_user_deployment) == 1 dagster_user_deployment = dagster_user_deployment[0] assert len(dagster_user_deployment.spec.template.spec.containers) == 1 container = dagster_user_deployment.spec.template.spec.containers[0] assert (container.startup_probe is not None) == enabled
def test_readiness_probes(template: HelmTemplate): deployment = create_simple_user_deployment("foo") deployment.readinessProbe = kubernetes.ReadinessProbe.construct(timeout_seconds=3) helm_values = DagsterHelmValues.construct( dagsterUserDeployments=UserDeployments.construct(deployments=[deployment]) ) dagster_user_deployment = template.render(helm_values) assert len(dagster_user_deployment) == 1 dagster_user_deployment = dagster_user_deployment[0] assert len(dagster_user_deployment.spec.template.spec.containers) == 1 container = dagster_user_deployment.spec.template.spec.containers[0] assert container.startup_probe is None assert container.startup_probe is None assert container.readiness_probe is not None
def test_startup_probe_exec(template: HelmTemplate): deployment = create_simple_user_deployment("foo") deployment.startupProbe = kubernetes.StartupProbe.construct( enabled=True, exec=dict(command=["my", "command"])) helm_values = DagsterHelmValues.construct( dagsterUserDeployments=UserDeployments.construct( deployments=[deployment])) dagster_user_deployment = template.render(helm_values) assert len(dagster_user_deployment) == 1 dagster_user_deployment = dagster_user_deployment[0] assert len(dagster_user_deployment.spec.template.spec.containers) == 1 container = dagster_user_deployment.spec.template.spec.containers[0] assert container.startup_probe._exec.command == [ # pylint:disable=protected-access "my", "command", ]
def test_startup_probe_default_exec(template: HelmTemplate): deployment = create_simple_user_deployment("foo") deployment.startupProbe = kubernetes.StartupProbe.construct(enabled=True) helm_values = DagsterHelmValues.construct( dagsterUserDeployments=UserDeployments.construct( deployments=[deployment])) dagster_user_deployment = template.render(helm_values) assert len(dagster_user_deployment) == 1 dagster_user_deployment = dagster_user_deployment[0] assert len(dagster_user_deployment.spec.template.spec.containers) == 1 container = dagster_user_deployment.spec.template.spec.containers[0] assert container.startup_probe._exec.command == [ # pylint: disable=protected-access "dagster", "api", "grpc-health-check", "-p", str(deployment.port), ]