# test_tree.py

# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors

#

# This module is part of GitPython and is released under

# the BSD License: http://www.opensource.org/licenses/bsd-license.php

from io import BytesIO

import sys

from unittest import skipIf

from git import (

Tree,

Blob

)

from git.test.lib import TestBase

from git.util import HIDE_WINDOWS_KNOWN_ERRORS

import os.path as osp

class TestTree(TestBase):

@skipIf(HIDE_WINDOWS_KNOWN_ERRORS and sys.version_info[:2] == (3, 5), """

File "C:\\projects\\gitpython\\git\\cmd.py", line 559, in execute

raise GitCommandNotFound(command, err)

git.exc.GitCommandNotFound: Cmd('git') not found due to: OSError('[WinError 6] The handle is invalid')

cmdline: git cat-file --batch-check""")

def test_serializable(self):

# tree at the given commit contains a submodule as well

roottree = self.rorepo.tree('6c1faef799095f3990e9970bc2cb10aa0221cf9c')

for item in roottree.traverse(ignore_self=False):

if item.type != Tree.type:

continue

# END skip non-trees

tree = item

# trees have no dict

self.failUnlessRaises(AttributeError, setattr, tree, 'someattr', 1)

orig_data = tree.data_stream.read()

orig_cache = tree._cache

stream = BytesIO()

tree._serialize(stream)

assert stream.getvalue() == orig_data

stream.seek(0)

testtree = Tree(self.rorepo, Tree.NULL_BIN_SHA, 0, '')

testtree._deserialize(stream)

assert testtree._cache == orig_cache

# replaces cache, but we make sure of it

del(testtree._cache)

testtree._deserialize(stream)

# END for each item in tree

@skipIf(HIDE_WINDOWS_KNOWN_ERRORS and sys.version_info[:2] == (3, 5), """

File "C:\\projects\\gitpython\\git\\cmd.py", line 559, in execute

raise GitCommandNotFound(command, err)

git.exc.GitCommandNotFound: Cmd('git') not found due to: OSError('[WinError 6] The handle is invalid')

cmdline: git cat-file --batch-check""")

def test_traverse(self):

root = self.rorepo.tree('0.1.6')

num_recursive = 0

all_items = []

for obj in root.traverse():

if "/" in obj.path:

num_recursive += 1

assert isinstance(obj, (Blob, Tree))

all_items.append(obj)

# END for each object

assert all_items == root.list_traverse()

# limit recursion level to 0 - should be same as default iteration

assert all_items

assert 'CHANGES' in root

assert len(list(root)) == len(list(root.traverse(depth=1)))

# only choose trees

trees_only = lambda i, d: i.type == "tree"

trees = list(root.traverse(predicate=trees_only))

assert len(trees) == len([i for i in root.traverse() if trees_only(i, 0)])

# test prune

lib_folder = lambda t, d: t.path == "lib"

pruned_trees = list(root.traverse(predicate=trees_only, prune=lib_folder))

assert len(pruned_trees) < len(trees)

# trees and blobs

assert len(set(trees) | set(root.trees)) == len(trees)

assert len({b for b in root if isinstance(b, Blob)} | set(root.blobs)) == len(root.blobs)

subitem = trees[0][0]

assert "/" in subitem.path

assert subitem.name == osp.basename(subitem.path)

# assure that at some point the traversed paths have a slash in them

found_slash = False

for item in root.traverse():

assert osp.isabs(item.abspath)

if '/' in item.path:

found_slash = True

# END check for slash

# slashes in paths are supported as well

# NOTE: on py3, / doesn't work with strings anymore ...

assert root[item.path] == item == root / item.path

# END for each item

assert found_slash