def Run(self): self.CreateCronJob(flow_name=cron_system.OSBreakDown.__name__, token=self.token) self.CreateCronJob(flow_name=cron_system.GRRVersionBreakDown.__name__, token=self.token) self.Check("ModifyCronJob", args=cron_plugin.ApiModifyCronJobArgs( cron_job_id=cron_system.OSBreakDown.__name__, state="ENABLED")) self.Check("ModifyCronJob", args=cron_plugin.ApiModifyCronJobArgs( cron_job_id=cron_system.GRRVersionBreakDown.__name__, state="DISABLED"))
def testCronJobMethodsAreAccessChecked(self): args = api_cron.ApiForceRunCronJobArgs(cron_job_id="TestCronJob") self.CheckMethodIsAccessChecked( self.router.ForceRunCronJob, "CheckCronJobAccess", args=args) args = api_cron.ApiModifyCronJobArgs(cron_job_id="TestCronJob") self.CheckMethodIsAccessChecked( self.router.ModifyCronJob, "CheckCronJobAccess", args=args) args = api_cron.ApiDeleteCronJobArgs(cron_job_id="TestCronJob") self.CheckMethodIsAccessChecked( self.router.DeleteCronJob, "CheckCronJobAccess", args=args)
def Run(self): with test_lib.FakeTime(44): cron_job_id1 = self.CreateCronJob( flow_name=cron_system.OSBreakDown.__name__, token=self.token) cron_job_id2 = self.CreateCronJob( flow_name=cron_system.GRRVersionBreakDown.__name__, token=self.token) self.Check("ModifyCronJob", args=cron_plugin.ApiModifyCronJobArgs( cron_job_id=cron_job_id1, enabled=True), replace={ cron_job_id1: "OSBreakDown", cron_job_id2: "GRRVersionBreakDown" }) self.Check("ModifyCronJob", args=cron_plugin.ApiModifyCronJobArgs( cron_job_id=cron_job_id2, enabled=False), replace={ cron_job_id1: "OSBreakDown", cron_job_id2: "GRRVersionBreakDown" })