

def _VerifyDownload(self):
    with utils.TempDirectory() as tmpdir:
        export_utils.DownloadCollection('aff4:/testcoll', tmpdir, overwrite=True, dump_client_info=True, token=self.token, max_threads=2)
        expected_outdir = os.path.join(tmpdir, self.out.Path()[1:])
        self.assertTrue(('testfile1' in os.listdir(expected_outdir)))
        self.assertTrue(('testfile2' in os.listdir(expected_outdir)))
        self.assertTrue(('testfile5' in os.listdir(expected_outdir)))
        self.assertTrue(('testfile6' in os.listdir(expected_outdir)))
        expected_rootdir = os.path.join(tmpdir, self.client_id.Basename())
        self.assertTrue(('client_info.yaml' in os.listdir(expected_rootdir)))

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