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)
