def testSlicesLogsWhenMultiplePlugins(self):
hunt_urn = self.RunHuntWithOutputPlugins(self.output_plugins)
result = self.handler.Handle(
hunt_plugin.ApiListHuntOutputPluginLogsArgs(
hunt_id=hunt_urn.Basename(),
offset=2,
count=2,
plugin_id=test_plugins.DummyHuntTestOutputPlugin.__name__ +
"_1"),
token=self.token)
self.assertEqual(result.total_count, 5)
self.assertEqual(len(result.items), 2)
def testReturnsLogsWhenMultiplePlugins(self):
hunt_urn = self.RunHuntWithOutputPlugins(self.output_plugins)
result = self.handler.Handle(
hunt_plugin.ApiListHuntOutputPluginLogsArgs(
hunt_id=hunt_urn.Basename(),
plugin_id=test_plugins.DummyHuntTestOutputPlugin.__name__ +
"_1"),
token=self.token)
self.assertEqual(result.total_count, 5)
self.assertEqual(len(result.items), 5)
for item in result.items:
self.assertEqual("bar",
item.plugin_descriptor.plugin_args.filename_regex)
def Run(self):
with test_lib.FakeTime(42, increment=1):
hunt_urn = self.StartHunt(
description="the hunt",
output_plugins=[
output_plugin.OutputPluginDescriptor(
plugin_name=test_plugins.DummyHuntTestOutputPlugin.
__name__,
plugin_args=test_plugins.DummyHuntTestOutputPlugin.
args_type(filename_regex="blah!", fetch_binaries=True))
])
self.client_ids = self.SetupClients(2)
for index, client_id in enumerate(self.client_ids):
self.AssignTasksToClients(client_ids=[client_id])
self.RunHunt(failrate=-1)
with test_lib.FakeTime(100042 + index * 100):
self.ProcessHuntOutputPlugins()
self.Check("ListHuntOutputPluginLogs",
args=hunt_plugin.ApiListHuntOutputPluginLogsArgs(
hunt_id=hunt_urn.Basename(),
plugin_id="DummyHuntTestOutputPlugin_0"),
replace={hunt_urn.Basename(): "H:123456"})
