def __init__(self, project_name, project_uuid, job_name, job_uuid, spec, k8s_config=None, namespace='default', in_cluster=False, job_container_name=None, job_docker_image=None, sidecar_container_name=None, sidecar_docker_image=None, role_label=None, type_label=None, ports=None, use_sidecar=False, sidecar_config=None, persist=False): self.spec = spec self.project_name = project_name self.project_uuid = project_uuid self.job_name = job_name self.job_uuid = job_uuid self.pod_manager = pods.PodManager( namespace=namespace, name=self.JOB_NAME, project_name=self.project_name, project_uuid=self.project_uuid, job_name=job_name, job_uuid=job_uuid, job_container_name=job_container_name, job_docker_image=job_docker_image, sidecar_container_name=sidecar_container_name, sidecar_docker_image=sidecar_docker_image, role_label=role_label, type_label=type_label, ports=ports, use_sidecar=use_sidecar, sidecar_config=sidecar_config, log_level=self.spec.log_level, health_check_url=get_job_health_url(job_name)) self.persist = persist super().__init__(k8s_config=k8s_config, namespace=namespace, in_cluster=in_cluster)
def __init__(self, project_name, project_uuid, job_name, job_uuid, k8s_config=None, namespace='default', version=None, in_cluster=False, job_container_name=None, job_docker_image=None, sidecar_container_name=None, sidecar_docker_image=None, role_label=None, type_label=None, use_sidecar=False, sidecar_config=None, log_level=None): self.project_name = project_name self.project_uuid = project_uuid self.job_name = job_name self.job_uuid = job_uuid self.resource_manager = manager.ResourceManager( namespace=namespace, version=version, name=JOB_NAME, project_name=project_name, project_uuid=project_uuid, job_name=job_name, job_uuid=job_uuid, job_docker_image=job_docker_image, job_container_name=job_container_name, sidecar_container_name=sidecar_container_name, sidecar_docker_image=sidecar_docker_image, role_label=role_label, type_label=type_label, use_sidecar=use_sidecar, sidecar_config=sidecar_config, health_check_url=get_job_health_url(job_name), log_level=log_level) super().__init__(k8s_config=k8s_config, namespace=namespace, in_cluster=in_cluster)
def test_get_job_health_url(self): self.assertEqual( get_job_health_url(unique_name=self.job.unique_name), '{}/_heartbeat'.format( get_job_url(unique_name=self.job.unique_name)))