def test_submit_job():
pod = core_v1.PodSpec(containers=[
core_v1.Container(
name="Container1",
image="index.docker.io/library/ubuntu:latest",
args=["sleep", "10s"],
securityContext=core_v1.SecurityContext(runAsUser=1000),
resources=core_v1.ResourceRequirements(
requests={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
limits={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
),
)
], )
tester.submit_jobs(
queue="test",
job_set_id="test",
job_request_items=[
tester.create_job_request_item(priority=1, pod_spec=pod)
],
)
def submit_sleep_container(image: str):
"""
Simple armada job where image allows you to control
if this container fails or not.
"""
pod = core_v1.PodSpec(
containers=[
core_v1.Container(
name="sleep",
image=image,
args=["sleep", "10s"],
securityContext=core_v1.SecurityContext(runAsUser=1000),
resources=core_v1.ResourceRequirements(
requests={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
limits={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
),
)
],
)
return [submit_pb2.JobSubmitRequestItem(priority=1, pod_spec=pod)]
def submit_sleep_job():
"""
This is a PodSpec definition that allows you to run sleep.
This returns an array of JobSubmitRequestItems that allows you
to submit to Armada.
"""
pod = core_v1.PodSpec(containers=[
core_v1.Container(
name="sleep",
image="busybox",
args=["sleep", "10s"],
securityContext=core_v1.SecurityContext(runAsUser=1000),
resources=core_v1.ResourceRequirements(
requests={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
limits={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
),
)
], )
return [submit_pb2.JobSubmitRequestItem(priority=1, pod_spec=pod)]
def submit_sleep_job(client):
pod = core_v1.PodSpec(containers=[
core_v1.Container(
name="sleep",
image="alpine:latest",
args=["sleep", "30s"],
resources=core_v1.ResourceRequirements(
requests={
"cpu": api_resource.Quantity(string="0.2"),
"memory": api_resource.Quantity(string="64Mi"),
},
limits={
"cpu": api_resource.Quantity(string="0.2"),
"memory": api_resource.Quantity(string="64Mi"),
},
),
)
], )
return [client.create_job_request_item(priority=0, pod_spec=pod)]
def sleep_job():
pod = core_v1.PodSpec(containers=[
core_v1.Container(
name="container-1",
image="busybox",
args=["sleep", "10s"],
securityContext=core_v1.SecurityContext(runAsUser=1000),
resources=core_v1.ResourceRequirements(
requests={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
limits={
"cpu": api_resource.Quantity(string="120m"),
"memory": api_resource.Quantity(string="510Mi"),
},
),
)
], )
return [submit_pb2.JobSubmitRequestItem(priority=0, pod_spec=pod)]