def Run(self): def ReplaceCronJobUrn(): jobs = list(aff4_cronjobs.GetCronManager().ListJobs(token=self.token)) return {jobs[0]: "CreateAndRunGenericHuntFlow_1234"} flow_name = file_finder.FileFinder.__name__ flow_args = rdf_file_finder.FileFinderArgs( paths=["c:\\windows\\system32\\notepad.*"]) hunt_runner_args = rdf_hunts.HuntRunnerArgs() hunt_runner_args.client_rule_set.rules = [ foreman_rules.ForemanClientRule( os=foreman_rules.ForemanOsClientRule(os_windows=True)) ] hunt_runner_args.description = "Foobar! (cron)" self.Check( "CreateCronJob", args=cron_plugin.ApiCreateCronJobArgs( description="Foobar!", flow_name=flow_name, flow_args=flow_args, hunt_runner_args=hunt_runner_args, periodicity=604800, lifetime=3600), replace=ReplaceCronJobUrn)
def testAddForemanRulesHuntRunnerArgumentIsNotRespected(self): args = cron_plugin.ApiCreateCronJobArgs( flow_name=flow_test_lib.FlowWithOneNestedFlow.__name__, hunt_runner_args=rdf_hunts.HuntRunnerArgs( # Default is True. add_foreman_rules=False)) result = self.handler.Handle(args, context=self.context) self.assertTrue( result.args.hunt_cron_action.hunt_runner_args.add_foreman_rules)