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))
