def test_jinja_templated_fields(self):
task = KubernetesPodOperator(
namespace='default',
image="{{ image_jinja }}:16.04",
cmds=["bash", "-cx"],
name="test_pod",
task_id="task",
)
assert task.image == "{{ image_jinja }}:16.04"
task.render_template_fields(context={"image_jinja": "ubuntu"})
assert task.image == "ubuntu:16.04"
def test_env_vars(self):
k = KubernetesPodOperator(
namespace="default",
image="ubuntu:16.04",
cmds=["bash", "-cx"],
arguments=["echo 10"],
env_vars=[k8s.V1EnvVar(name="{{ bar }}", value="{{ foo }}")],
labels={"foo": "bar"},
name="test",
task_id="task",
in_cluster=False,
do_xcom_push=False,
)
k.render_template_fields(context={"foo": "footemplated", "bar": "bartemplated"})
assert k.env_vars[0].value == "footemplated"
assert k.env_vars[0].name == "bartemplated"
def test_env_vars(self):
# WHEN
env_vars = [k8s.V1EnvVar(name="{{ bar }}", value='{{ foo }}')]
from tests.models import DEFAULT_DATE
with DAG("test-dag", start_date=DEFAULT_DATE):
k = KubernetesPodOperator(
namespace='default',
image="ubuntu:16.04",
cmds=["bash", "-cx"],
arguments=["echo 10"],
env_vars=env_vars,
labels={"foo": "bar"},
name="test",
task_id="task",
in_cluster=False,
do_xcom_push=False,
)
k.render_template_fields(context={
"foo": "footemplated",
"bar": "bartemplated"
})
assert k.env_vars[0].value == "footemplated"
assert k.env_vars[0].name == "bartemplated"
