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