def test_run_beam_job_using_job_class(self) -> None: model = beam_job_services.create_beam_job_run_model('NoOpJob') with self.swap_to_always_return(jobs_manager, 'run_job', value=model): run = beam_job_services.run_beam_job(job_class=NoOpJob) self.assertEqual( beam_job_services.get_beam_job_run_from_model(model).to_dict(), run.to_dict())
def test_run_beam_job(self) -> None: run_model = beam_job_services.create_beam_job_run_model('WorkingJob') get_job_class_by_name_swap = self.swap_to_always_return( jobs_registry, 'get_job_class_by_name') run_job_swap = self.swap_to_always_return( jobs_manager, 'run_job', value=run_model) with get_job_class_by_name_swap, run_job_swap: run = beam_job_services.run_beam_job('WorkingJob') self.assertEqual( beam_job_services.get_beam_job_run_from_model(run_model).to_dict(), run.to_dict())
def put(self) -> None: job_name: str = (self.normalized_payload.get('job_name') if self.normalized_payload else '') beam_job_run = beam_job_services.run_beam_job(job_name) self.render_json(beam_job_run.to_dict())
def test_run_beam_job_without_args_raises_an_exception(self) -> None: with self.assertRaisesRegex( ValueError, 'Must specify the job'): # type: ignore[no-untyped-call] beam_job_services.run_beam_job()
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job( job_class=(suggestion_stats_computation_jobs. GenerateTranslationContributionStatsJob))
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job( job_class=exp_search_indexing_jobs.IndexExplorationsInSearchJob)
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job( job_class=(exp_recommendation_computation_jobs. ComputeExplorationRecommendationsJob))
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job(job_class=( user_stats_computation_jobs.CollectWeeklyDashboardStatsJob))
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job( job_class=cron_jobs.GenerateTranslationContributionStats)
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job( job_class=cron_jobs.IndexExplorationsInSearch)
def get(self): """Handles GET requests.""" beam_job_services.run_beam_job( job_class=cron_jobs.CollectWeeklyDashboardStats)