def testBaseSessionIdFlowRunnerArgumentIsNotRespected(self):
args = cron_plugin.ApiCronJob(
flow_name=standard.CreateAndRunGenericHuntFlow.__name__,
flow_runner_args=rdf_flows.FlowRunnerArgs(
base_session_id="aff4:/foo"))
result = self.handler.Handle(args, token=self.token)
self.assertFalse(result.flow_runner_args.HasField("base_session_id"))
def Run(self):
def ReplaceCronJobUrn():
jobs = list(cronjobs.CRON_MANAGER.ListJobs(token=self.token))
return {jobs[0].Basename(): "CreateAndRunGeneicHuntFlow_1234"}
flow_args = standard.CreateGenericHuntFlowArgs()
flow_args.hunt_args.flow_args = rdf_file_finder.FileFinderArgs(
paths=["c:\\windows\\system32\\notepad.*"])
flow_args.hunt_args.flow_runner_args.flow_name = (
file_finder.FileFinder.__name__)
flow_args.hunt_runner_args.client_rule_set.rules = [
rdf_foreman.ForemanClientRule(os=rdf_foreman.ForemanOsClientRule(
os_windows=True))
]
flow_args.hunt_runner_args.description = "Foobar! (cron)"
self.Check(
"CreateCronJob",
args=cron_plugin.ApiCronJob(
description="Foobar!",
flow_name=standard.CreateAndRunGenericHuntFlow.__name__,
periodicity=604800,
lifetime=3600,
flow_args=flow_args),
replace=ReplaceCronJobUrn)