This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new c3c90eaa94 Add options to KubernetesPodOperator (#30992)
c3c90eaa94 is described below
commit c3c90eaa949b34c18f7b49052b2733ed817f8bed
Author: RachitSharma2001 <[email protected]>
AuthorDate: Wed May 10 02:53:02 2023 -0700
Add options to KubernetesPodOperator (#30992)
* Add options to KubernetesPodOperator
---
airflow/providers/cncf/kubernetes/operators/pod.py | 12 ++++++++
.../cncf/kubernetes/operators/test_pod.py | 35 ++++++++++++++++++++++
2 files changed, 47 insertions(+)
diff --git a/airflow/providers/cncf/kubernetes/operators/pod.py
b/airflow/providers/cncf/kubernetes/operators/pod.py
index 1bfa2ba034..fc09783a39 100644
--- a/airflow/providers/cncf/kubernetes/operators/pod.py
+++ b/airflow/providers/cncf/kubernetes/operators/pod.py
@@ -197,6 +197,9 @@ class KubernetesPodOperator(BaseOperator):
:param security_context: security options the pod should run with
(PodSecurityContext).
:param container_security_context: security options the container should
run with.
:param dnspolicy: dnspolicy for the pod.
+ :param dns_config: dns configuration (ip addresses, searches, options) for
the pod.
+ :param hostname: hostname for the pod.
+ :param subdomain: subdomain for the pod.
:param schedulername: Specify a schedulername for the pod
:param full_pod_spec: The complete podSpec
:param init_containers: init container for the launched Pod
@@ -282,6 +285,9 @@ class KubernetesPodOperator(BaseOperator):
security_context: dict | None = None,
container_security_context: dict | None = None,
dnspolicy: str | None = None,
+ dns_config: k8s.V1PodDNSConfig | None = None,
+ hostname: str | None = None,
+ subdomain: str | None = None,
schedulername: str | None = None,
full_pod_spec: k8s.V1Pod | None = None,
init_containers: list[k8s.V1Container] | None = None,
@@ -351,6 +357,9 @@ class KubernetesPodOperator(BaseOperator):
self.security_context = security_context or {}
self.container_security_context = container_security_context
self.dnspolicy = dnspolicy
+ self.dns_config = dns_config
+ self.hostname = hostname
+ self.subdomain = subdomain
self.schedulername = schedulername
self.full_pod_spec = full_pod_spec
self.init_containers = init_containers or []
@@ -816,8 +825,11 @@ class KubernetesPodOperator(BaseOperator):
image_pull_secrets=self.image_pull_secrets,
service_account_name=self.service_account_name,
host_network=self.hostnetwork,
+ hostname=self.hostname,
+ subdomain=self.subdomain,
security_context=self.security_context,
dns_policy=self.dnspolicy,
+ dns_config=self.dns_config,
scheduler_name=self.schedulername,
restart_policy="Never",
priority_class_name=self.priority_class_name,
diff --git a/tests/providers/cncf/kubernetes/operators/test_pod.py
b/tests/providers/cncf/kubernetes/operators/test_pod.py
index f3fd713f9b..c220b34429 100644
--- a/tests/providers/cncf/kubernetes/operators/test_pod.py
+++ b/tests/providers/cncf/kubernetes/operators/test_pod.py
@@ -324,6 +324,41 @@ class TestKubernetesPodOperator:
"already_checked!=True,!airflow-worker"
)
+ @patch(HOOK_CLASS, new=MagicMock)
+ def test_pod_dns_options(self):
+ dns_config = k8s.V1PodDNSConfig(
+ nameservers=["192.0.2.1", "192.0.2.3"],
+ searches=["ns1.svc.cluster-domain.example",
"my.dns.search.suffix"],
+ options=[
+ k8s.V1PodDNSConfigOption(
+ name="ndots",
+ value="2",
+ )
+ ],
+ )
+ hostname = "busybox-2"
+ subdomain = "busybox-subdomain"
+
+ k = KubernetesPodOperator(
+ namespace="default",
+ image="ubuntu:16.04",
+ cmds=["bash", "-cx"],
+ labels={"foo": "bar"},
+ name="test",
+ task_id="task",
+ in_cluster=False,
+ do_xcom_push=False,
+ dns_config=dns_config,
+ hostname=hostname,
+ subdomain=subdomain,
+ )
+
+ self.run_pod(k)
+ pod_spec = k.pod.spec
+ assert pod_spec.dns_config == dns_config
+ assert pod_spec.subdomain == subdomain
+ assert pod_spec.hostname == hostname
+
@pytest.mark.parametrize(
"val",
[
