def testRendersRequestedCronJobApproval(self): cron_manager = aff4_cronjobs.CronManager() cron_args = aff4_cronjobs.CreateCronJobFlowArgs(periodicity="1d", allow_overruns=False) cron_job_urn = cron_manager.ScheduleFlow(cron_args=cron_args, token=self.token) self.RequestCronJobApproval(cron_job_urn.Basename(), reason=self.token.reason, approver="approver", requestor=self.token.username) args = user_plugin.ApiListCronJobApprovalsArgs() result = self.handler.Handle(args, token=self.token) self.assertEqual(len(result.items), 1)
def ListCronJobApprovals(self, requestor=None): requestor = requestor or self.token.username handler = api_user.ApiListCronJobApprovalsHandler() return handler.Handle( api_user.ApiListCronJobApprovalsArgs(), token=access_control.ACLToken(username=requestor)).items