def Start(self):
self.state.hunt_id = None
self.state.client_ids = set()
self.state.client_ids_failures = set()
self.state.client_ids_result_reported = set()
self.state.client_ids = list(
base.GetClientTestTargets(token=self.token))
if not self.state.client_ids:
self.Log("No clients to test on, define them in "
"Test.end_to_end_client_ids")
return
# SetUID is required to run a hunt on the configured end-to-end client
# targets without an approval.
token = access_control.ACLToken(
username="******", reason="Running endtoend tests.").SetUID()
runner_args = rdf_flows.FlowRunnerArgs(flow_name="EndToEndTestFlow")
flow_request = hunts_standard.FlowRequest(
client_ids=self.state.client_ids,
args=flows_endtoend.EndToEndTestFlowArgs(),
runner_args=runner_args)
bogus_rule = rdf_foreman.ForemanRegexClientRule(
attribute_name="System", attribute_regex="Does not match anything")
client_rule_set = rdf_foreman.ForemanClientRuleSet(rules=[
rdf_foreman.ForemanClientRule(
rule_type=rdf_foreman.ForemanClientRule.Type.REGEX,
regex=bogus_rule)
])
hunt_args = hunts_standard.VariableGenericHuntArgs(
flows=[flow_request])
hunt_args.output_plugins = self.GetOutputPlugins()
with hunts.GRRHunt.StartHunt(hunt_name="VariableGenericHunt",
args=hunt_args,
client_rule_set=client_rule_set,
client_rate=0,
expiry_time="1d",
token=token) as hunt:
self.state.hunt_id = hunt.session_id
hunt.SetDescription("EndToEnd tests run by cron")
hunt.Run()
hunt.ManuallyScheduleClients(token=token)
# Set a callback to check the results after 50 minutes. This should be
# plenty of time for the clients to receive the hunt and run the tests, but
# not so long that the flow lease will expire.
wait_duration = rdfvalue.Duration(
config_lib.CONFIG.Get("Test.end_to_end_result_check_wait"))
completed_time = rdfvalue.RDFDatetime.Now() + wait_duration
self.CallState(next_state="CheckResults", start_time=completed_time)
def RunVariableGenericHunt(self):
args = standard.VariableGenericHuntArgs()
self._AppendFlowRequest(args.flows, 1, 1)
self._AppendFlowRequest(args.flows, 2, 2)
self._AppendFlowRequest(args.flows, 2, 3)
with hunts.GRRHunt.StartHunt(hunt_name="VariableGenericHunt",
args=args, client_rate=0,
token=self.token) as hunt:
hunt.Run()
hunt.ManuallyScheduleClients()
# Run the hunt.
client_mock = test_lib.SampleHuntMock(failrate=100)
test_lib.TestHuntHelper(client_mock, self.client_ids, False, self.token)
with aff4.FACTORY.Open(hunt.session_id, mode="rw",
token=self.token) as hunt:
hunt.Stop()
return hunt
