def testExportDir(self): """Check we can export a dir without errors.""" self.CreateDir("testdir") self.CreateFile("testdir/testfile1") self.CreateFile("testdir/testfile2") self.CreateDir("testdir/testdir1") self.CreateFile("testdir/testdir1/testfile3") self.CreateDir("testdir/testdir1/testdir2") self.CreateFile("testdir/testdir1/testdir2/testfile4") plugin = file_plugin.FileExportPlugin() parser = argparse.ArgumentParser() plugin.ConfigureArgParser(parser) with utils.TempDirectory() as tmpdir: plugin.Run( parser.parse_args(args=[ "--path", str(self.out.Add("testdir")), "--output", tmpdir ])) expected_outdir = os.path.join(tmpdir, self.out.Add("testdir").Path()[1:]) self.assertTrue("testfile1" in os.listdir(expected_outdir)) full_outdir = os.path.join(expected_outdir, "testdir1", "testdir2") self.assertTrue("testfile4" in os.listdir(full_outdir))
def testExportFile(self): """Check we can export a file without errors.""" self.CreateFile("testfile1") plugin = file_plugin.FileExportPlugin() parser = argparse.ArgumentParser() plugin.ConfigureArgParser(parser) with utils.TempDirectory() as tmpdir: plugin.Run( parser.parse_args(args=[ "--no_legacy_warning_pause", "--path", str( self.out.Add("testfile1")), "--output", tmpdir ])) expected_outdir = os.path.join(tmpdir, self.out.Path()[1:]) self.assertTrue("testfile1" in os.listdir(expected_outdir))