

def testDownloadHuntResultCollection(self):
    'Check we can download files references in HuntResultCollection.'
    fd = aff4.FACTORY.Create('aff4:/testcoll', 'HuntResultCollection', token=self.token)
    fd.AddAsMessage(rdfvalue.RDFURN(self.out.Add('testfile1')), self.client_id)
    fd.AddAsMessage(rdf_client.StatEntry(aff4path=self.out.Add('testfile2')), self.client_id)
    fd.AddAsMessage(file_finder.FileFinderResult(stat_entry=rdf_client.StatEntry(aff4path=self.out.Add('testfile5'))), self.client_id)
    fd.AddAsMessage(collectors.ArtifactFilesDownloaderResult(downloaded_file=rdf_client.StatEntry(aff4path=self.out.Add('testfile6'))), self.client_id)
    fd.Close()
    self._VerifyDownload()

dataset/ETHPy150Open google/grr/grr/lib/export_utils_test.py TestExports.testDownloadHuntResultCollection/original