deb-python-pygit2/test/test_treebuilder.py
Carlos Martín Nieto 242f3c2ffa TreeBuilder: allow the source to be a Tree
If the user passes a tree, use it as the source for the
TreeBuilder. On the way, make sure we free the tree we looked up, and
fix a test to make sure the TreeBuilder starts empty.
2012-03-09 12:57:22 +01:00

70 lines
2.2 KiB
Python

# -*- coding: utf-8 -*-
#
# Copyright 2012 Carlos Martín Nieto
#
# This file is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License, version 2,
# as published by the Free Software Foundation.
#
# In addition to the permissions in the GNU General Public License,
# the authors give you unlimited permission to link the compiled
# version of this file into combinations with other programs,
# and to distribute those combinations without any restriction
# coming from the use of this file. (The General Public License
# restrictions do apply in other respects; for example, they cover
# modification of the file, and distribution when not linked into
# a combined executable.)
#
# This file is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; see the file COPYING. If not, write to
# the Free Software Foundation, 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301, USA.
"""Tests for Index files."""
from __future__ import absolute_import
from __future__ import unicode_literals
import os
import unittest
import pygit2
from . import utils
__author__ = 'carlos@cmartin.tk (Carlos Martín Nieto)'
TREE_SHA = '967fce8df97cc71722d3c2a5930ef3e6f1d27b12'
class TreeBuilderTest(utils.BareRepoTestCase):
def test_new_empty_treebuilder(self):
bld = self.repo.TreeBuilder()
def test_noop_treebuilder(self):
tree = self.repo[TREE_SHA]
bld = self.repo.TreeBuilder(TREE_SHA)
result = bld.write()
self.assertEqual(tree.oid, result)
def test_noop_treebuilder_from_tree(self):
tree = self.repo[TREE_SHA]
bld = self.repo.TreeBuilder(tree)
result = bld.write()
self.assertEqual(tree.oid, result)
def test_rebuild_treebuilder(self):
tree = self.repo[TREE_SHA]
bld = self.repo.TreeBuilder()
for e in tree:
bld.insert(e)
result = bld.write()
self.assertEqual(tree.oid, result)
if __name__ == '__main__':
unittest.main()