def testWorksCorrectlyWithTestOutputPluginOnFlowWithSingleResult(self):
result = self.handler.Handle(
hunt_plugin.ApiGetExportedHuntResultsArgs(
hunt_id=self.hunt.urn.Basename(),
plugin_name=instant_output_plugin_test.TestInstantOutputPlugin.
plugin_name),
token=self.token)
chunks = list(result.GenerateContent())
self.assertListEqual(
chunks,
["Start: %s" % utils.SmartStr(self.hunt.urn),
"Values of type: RDFString",
"First pass: oh",
"First pass: oh",
"First pass: oh",
"First pass: oh",
"First pass: oh",
"Second pass: oh",
"Second pass: oh",
"Second pass: oh",
"Second pass: oh",
"Second pass: oh",
"Finish: %s" % utils.SmartStr(self.hunt.urn)]) # pyformat: disable
def MockMethodIsCalled():
try:
mock_method.assert_called_once_with(
api_hunt.ApiGetExportedHuntResultsArgs(
hunt_id=hunt_urn.Basename(), plugin_name=plugin),
token=mock.ANY)
return True
except AssertionError:
return False
