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