Differential D4920 Diff 28833 integration_tests/python_modules/dagster-k8s-test-infra/dagster_k8s_test_infra/helm.py
Changeset View
Changeset View
Standalone View
Standalone View
integration_tests/python_modules/dagster-k8s-test-infra/dagster_k8s_test_infra/helm.py
| Show First 20 Lines • Show All 350 Lines • ▼ Show 20 Lines | helm_config = { | ||||
| "enabled": True, | "enabled": True, | ||||
| "dagit": {"host": "dagit.example.com"}, | "dagit": {"host": "dagit.example.com"}, | ||||
| "flower": {"flower": "flower.example.com"}, | "flower": {"flower": "flower.example.com"}, | ||||
| }, | }, | ||||
| "computeLogManager": { | "computeLogManager": { | ||||
| "type": "LocalComputeLogManager", | "type": "LocalComputeLogManager", | ||||
| "config": {"localComputeLogManager": {}}, | "config": {"localComputeLogManager": {}}, | ||||
| }, | }, | ||||
| "scheduler": {"k8sEnabled": True, "schedulerNamespace": namespace}, | "scheduler": { | ||||
| "type": "K8sScheduler", | |||||
| "config": {"k8sScheduler": {"schedulerNamespace": namespace}}, | |||||
| }, | |||||
| "serviceAccount": {"name": "dagit-admin"}, | "serviceAccount": {"name": "dagit-admin"}, | ||||
| "postgresqlPassword": "test", | "postgresqlPassword": "test", | ||||
| "postgresqlDatabase": "test", | "postgresqlDatabase": "test", | ||||
| "postgresqlUser": "test", | "postgresqlUser": "test", | ||||
| "dagsterDaemon": {"enabled": False}, | "dagsterDaemon": {"enabled": False}, | ||||
| } | } | ||||
| with _helm_chart_helper(namespace, should_cleanup, helm_config, helm_install_name="helm_chart"): | with _helm_chart_helper(namespace, should_cleanup, helm_config, helm_install_name="helm_chart"): | ||||
| Show All 26 Lines | helm_config = { | ||||
| }, | }, | ||||
| }, | }, | ||||
| "celery": {"enabled": False}, | "celery": {"enabled": False}, | ||||
| "k8sRunLauncher": {"enabled": True, "jobNamespace": namespace}, | "k8sRunLauncher": {"enabled": True, "jobNamespace": namespace}, | ||||
| "computeLogManager": { | "computeLogManager": { | ||||
| "type": "LocalComputeLogManager", | "type": "LocalComputeLogManager", | ||||
| "config": {"localComputeLogManager": {}}, | "config": {"localComputeLogManager": {}}, | ||||
| }, | }, | ||||
| "scheduler": {"k8sEnabled": True, "schedulerNamespace": namespace}, | "scheduler": { | ||||
| "type": "K8sScheduler", | |||||
| "config": {"k8sScheduler": {"schedulerNamespace": namespace}}, | |||||
| }, | |||||
| "serviceAccount": {"name": "dagit-admin"}, | "serviceAccount": {"name": "dagit-admin"}, | ||||
| "postgresqlPassword": "test", | "postgresqlPassword": "test", | ||||
| "postgresqlDatabase": "test", | "postgresqlDatabase": "test", | ||||
| "postgresqlUser": "test", | "postgresqlUser": "test", | ||||
| "dagsterDaemon": {"enabled": False}, | "dagsterDaemon": {"enabled": False}, | ||||
| } | } | ||||
| with _helm_chart_helper( | with _helm_chart_helper( | ||||
| ▲ Show 20 Lines • Show All 74 Lines • ▼ Show 20 Lines | helm_config = { | ||||
| "broker_transport_options": {"priority_steps": [9]}, | "broker_transport_options": {"priority_steps": [9]}, | ||||
| "worker_concurrency": 1, | "worker_concurrency": 1, | ||||
| }, | }, | ||||
| }, | }, | ||||
| "computeLogManager": { | "computeLogManager": { | ||||
| "type": "LocalComputeLogManager", | "type": "LocalComputeLogManager", | ||||
| "config": {"localComputeLogManager": {}}, | "config": {"localComputeLogManager": {}}, | ||||
| }, | }, | ||||
| "scheduler": {"k8sEnabled": True, "schedulerNamespace": namespace}, | "scheduler": { | ||||
| "type": "K8sScheduler", | |||||
| "config": {"k8sScheduler": {"schedulerNamespace": namespace}}, | |||||
| }, | |||||
| "serviceAccount": {"name": "dagit-admin"}, | "serviceAccount": {"name": "dagit-admin"}, | ||||
| "dagsterDaemon": {"enabled": False}, | "dagsterDaemon": {"enabled": False}, | ||||
| } | } | ||||
| with _helm_chart_helper( | with _helm_chart_helper( | ||||
| namespace, should_cleanup, helm_config, helm_install_name="helm_chart_for_user_deployments" | namespace, should_cleanup, helm_config, helm_install_name="helm_chart_for_user_deployments" | ||||
| ): | ): | ||||
| yield | yield | ||||
| ▲ Show 20 Lines • Show All 67 Lines • ▼ Show 20 Lines | helm_config = { | ||||
| "worker_concurrency": 1, | "worker_concurrency": 1, | ||||
| }, | }, | ||||
| "annotations": {"dagster-integration-tests": "celery-pod-annotation"}, | "annotations": {"dagster-integration-tests": "celery-pod-annotation"}, | ||||
| }, | }, | ||||
| "computeLogManager": { | "computeLogManager": { | ||||
| "type": "LocalComputeLogManager", | "type": "LocalComputeLogManager", | ||||
| "config": {"localComputeLogManager": {}}, | "config": {"localComputeLogManager": {}}, | ||||
| }, | }, | ||||
| "scheduler": {"k8sEnabled": False, "schedulerNamespace": namespace}, | "scheduler": {"type": "DagsterDaemonScheduler", "config": {}}, | ||||
| "serviceAccount": {"name": "dagit-admin"}, | "serviceAccount": {"name": "dagit-admin"}, | ||||
| "postgresqlPassword": "test", | "postgresqlPassword": "test", | ||||
| "postgresqlDatabase": "test", | "postgresqlDatabase": "test", | ||||
| "postgresqlUser": "test", | "postgresqlUser": "test", | ||||
| "dagsterDaemon": { | "dagsterDaemon": { | ||||
| "enabled": True, | "enabled": True, | ||||
| "image": {"repository": repository, "tag": tag, "pullPolicy": pull_policy}, | "image": {"repository": repository, "tag": tag, "pullPolicy": pull_policy}, | ||||
| "queuedRunCoordinator": {"enabled": True}, | "queuedRunCoordinator": {"enabled": True}, | ||||
| Show All 11 Lines | |||||