def test_default_toleration_preserved(image_name):
pod_spec = clean_pod_template(
make_pod_spec(
image=image_name,
extra_pod_config={
"tolerations": [
{
"key": "example.org/toleration",
"operator": "Exists",
"effect": "NoSchedule",
}
]
},
)
)
tolerations = pod_spec.to_dict()["spec"]["tolerations"]
assert {
"key": "k8s.dask.org/dedicated",
"operator": "Equal",
"value": "worker",
"effect": "NoSchedule",
"toleration_seconds": None,
} in tolerations
assert {
"key": "k8s.dask.org_dedicated",
"operator": "Equal",
"value": "worker",
"effect": "NoSchedule",
"toleration_seconds": None,
} in tolerations
assert {
"key": "example.org/toleration",
"operator": "Exists",
"effect": "NoSchedule",
} in tolerations
def pod_spec(docker_image):
yield clean_pod_template(
make_pod_spec(
image=docker_image,
extra_container_config={"imagePullPolicy": "IfNotPresent"},
)
)
def pod_spec(image_name):
yield clean_pod_template(
make_pod_spec(
image=image_name, extra_container_config={"imagePullPolicy": "IfNotPresent"}
)
)
