GitPython/git/test/test_tree.py at 0.3.6 · githubdelegate/GitPython · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

# test_tree.py

# This module is part of GitPython and is released under

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

import os

from git.test.lib import TestBase

from git import (

Tree,

Blob

)

from io import BytesIO

class TestTree(TestBase):

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

# TEST CACHE MUTATOR

mod = testtree.cache

self.failUnlessRaises(ValueError, mod.add, "invalid sha", 0, "name")

self.failUnlessRaises(ValueError, mod.add, Tree.NULL_HEX_SHA, 0, "invalid mode")

self.failUnlessRaises(ValueError, mod.add, Tree.NULL_HEX_SHA, tree.mode, "invalid/name")

# add new item

name = "fake_dir"

mod.add(testtree.NULL_HEX_SHA, tree.mode, name)

assert name in testtree

# its available in the tree immediately

assert isinstance(testtree[name], Tree)

# adding it again will not cause multiple of them to be presents

cur_count = len(testtree)

mod.add(testtree.NULL_HEX_SHA, tree.mode, name)

assert len(testtree) == cur_count

# fails with a different sha - name exists

hexsha = "1" * 40

self.failUnlessRaises(ValueError, mod.add, hexsha, tree.mode, name)

# force it - replace existing one

mod.add(hexsha, tree.mode, name, force=True)

assert testtree[name].hexsha == hexsha

assert len(testtree) == cur_count

# unchecked addition always works, even with invalid items

invalid_name = "hi/there"

mod.add_unchecked(hexsha, 0, invalid_name)

assert len(testtree) == cur_count + 1

del(mod[invalid_name])

assert len(testtree) == cur_count

# del again, its fine

del(mod[invalid_name])

# have added one item, we are done

mod.set_done()

mod.set_done() # multiple times are okay

# serialize, its different now

stream = BytesIO()

testtree._serialize(stream)

stream.seek(0)

assert stream.getvalue() != orig_data

# replaces cache, but we make sure of it

del(testtree._cache)

testtree._deserialize(stream)

assert name in testtree

assert invalid_name not in testtree

# END for each item in tree

def test_traverse(self):

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

num_recursive = 0

all_items = list()

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(list(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(set(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 == os.path.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 os.path.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

GitPython/git/test/test_tree.py at 0.3.6 · githubdelegate/GitPython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

test_tree.py

Latest commit

History

Breadcrumbs

test_tree.py

File metadata and controls