def runTest(self):
"""Launch our flows."""
for flow, args in [
("ListDirectory", {
"pathspec":
rdf_paths.PathSpec(
pathtype=rdf_paths.PathSpec.PathType.REGISTRY,
path=self.reg_path)
}),
("FindFiles", {
"findspec":
rdf_client.FindSpec(pathspec=rdf_paths.PathSpec(
path=self.reg_path,
pathtype=rdf_paths.PathSpec.PathType.REGISTRY),
path_regex="ProfileImagePath"),
"output":
self.output_path
})
]:
if self.local_worker:
self.session_id = debugging.StartFlowAndWorker(
self.client_id, flow, **args)
else:
self.session_id = flow_utils.StartFlowAndWait(self.client_id,
flow_name=flow,
token=self.token,
**args)
self.CheckFlow()
def runTest(self):
"""Launch our flows."""
for flow, args in [(filesystem.ListDirectory.__name__, {
"pathspec":
rdf_paths.PathSpec(
pathtype=rdf_paths.PathSpec.PathType.REGISTRY,
path=self.reg_path)
}), (find.FindFiles.__name__, {
"findspec":
rdf_client.FindSpec(
pathspec=rdf_paths.PathSpec(
path=self.reg_path,
pathtype=rdf_paths.PathSpec.PathType.REGISTRY),
path_regex="ProfileImagePath"),
})]:
if self.local_worker:
self.session_id = debugging.StartFlowAndWorker(self.client_id, flow,
**args)
else:
self.session_id = flow_utils.StartFlowAndWait(
self.client_id, flow_name=flow, token=self.token, **args)
results = self.CheckResultCollectionNotEmptyWithRetry(self.session_id)
for stat_entry in results:
self.assertTrue(isinstance(stat_entry, rdf_client.StatEntry))
self.assertTrue("ProfileImagePath" in stat_entry.pathspec.path)
def runTest(self):
if self.local_worker:
self.session_id = debugging.StartFlowAndWorker(
self.client_id, self.flow, **self.args)
else:
self.session_id = flow_utils.StartFlowAndWait(self.client_id,
flow_name=self.flow,
timeout=self.timeout,
token=self.token,
**self.args)
self.CheckFlow()
def runTest(self):
if self.client_min_version:
target_client = aff4.FACTORY.Open(self.client_id, token=self.token)
client_info = target_client.Get(target_client.Schema.CLIENT_INFO)
if client_info.client_version < self.client_min_version:
message = "Skipping version %s less than client_min_version: %s" % (
client_info.client_version, self.client_min_version)
return self.skipTest(message)
if self.local_worker:
self.session_id = debugging.StartFlowAndWorker(
self.client_id, self.flow, **self.args)
else:
self.session_id = flow_utils.StartFlowAndWait(
self.client_id, flow_name=self.flow,
timeout=self.timeout, token=self.token, **self.args)
self.CheckFlow()
def runTest(self):
if self.local_worker:
self.session_id = debugging.StartFlowAndWorker(
self.client_id,
self.flow,
cpu_limit=self.cpu_limit,
network_bytes_limit=self.network_bytes_limit,
**self.args)
else:
self.session_id = flow_utils.StartFlowAndWait(
self.client_id,
flow_name=self.flow,
cpu_limit=self.cpu_limit,
network_bytes_limit=self.network_bytes_limit,
token=self.token,
**self.args)
self.CheckFlow()