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)]