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