Method test_adding_tree

tools/mini_tar/mini_tar_test.py:145–196

(self)

    self.assertTarFileContent(self.tempfile, content)

  def test_adding_tree(self):
    content = [
        {
            "name": "./a",
            "mode": 0o750
        },
        {
            "name": "./a/b",
            "data": b"ab",
            "mode": 0o640
        },
        {
            "name": "./a/c",
            "mode": 0o750
        },
        {
            "name": "./a/c/d",
            "data": b"acd",
            "mode": 0o640
        },
    ]
    tempdir = os.path.join(os.environ["TEST_TMPDIR"], "test_dir")
    # Iterate over the `content` array to create the directory
    # structure it describes.
    for c in content:
      if "data" in c:
        p = os.path.join(tempdir, c["name"])
        os.makedirs(os.path.dirname(p))
        with open(p, "wb") as f:
          f.write(c["data"])
    with mini_tar.TarFileWriter(self.tempfile) as f:
      f.add_file_at_dest(in_path=tempdir, dest_path=".", mode=0o640)
    self.assertTarFileContent(self.tempfile, content)

    # Try it again, but re-rooted
    with mini_tar.TarFileWriter(self.tempfile, root_directory="foo") as f:
      f.add_file_at_dest(in_path=tempdir, dest_path="x", mode=0o640)
    n_content = [
        {
            "name": "foo",
            "mode": 0o755
        },
        {
            "name": "foo/x",
            "mode": 0o750
        },
    ]
    for c in content:
      nc = copy.copy(c)
      nc["name"] = "foo/x/" + c["name"][2:]
      n_content.append(nc)
    self.assertTarFileContent(self.tempfile, n_content)


if __name__ == "__main__":