def test_to_dict(self):
job = beam_job_domain.BeamJob(
base_validation_jobs.AuditAllStorageModelsJob)
self.assertEqual(job.to_dict(), {
'name': 'AuditAllStorageModelsJob',
'argument_names': [],
})
def get_beam_jobs():
"""Returns the list of all registered Apache Beam jobs.
Returns:
list(BeamJob). The list of registered Apache Beam jobs.
"""
return [beam_job_domain.BeamJob(j) for j in jobs_registry.get_all_jobs()]
def test_get_returns_registered_jobs(self) -> None:
job = beam_job_domain.BeamJob(FooJob)
get_beam_jobs_swap = self.swap_to_always_return(
beam_job_services, 'get_beam_jobs', value=[job])
with get_beam_jobs_swap:
response = self.get_json('/beam_job')
self.assertEqual(response, {
'jobs': [{'name': 'FooJob'}],
})
def test_to_dict(self) -> None:
job = beam_job_domain.BeamJob(
model_validation_jobs.AuditAllStorageModelsJob)
self.assertEqual(job.to_dict(), {'name': 'AuditAllStorageModelsJob'})
def test_usage(self) -> None:
job = beam_job_domain.BeamJob(
model_validation_jobs.AuditAllStorageModelsJob)
self.assertEqual(job.name, 'AuditAllStorageModelsJob')
def test_usage(self):
job = beam_job_domain.BeamJob(
base_validation_jobs.AuditAllStorageModelsJob)
self.assertEqual(job.name, 'AuditAllStorageModelsJob')
self.assertEqual(job.argument_names, [])
def test_usage(self):
job = beam_job_domain.BeamJob(validation_jobs.AuditAllStorageModelsJob)
self.assertEqual(job.name, 'AuditAllStorageModelsJob')
self.assertEqual(job.parameter_names, [])
