~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2006 Canonical Ltd

# Authors:  Robert Collins <robert.collins@canonical.com>

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program 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; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""Tests for the TreeBuilder helper class."""

from bzrlib import errors, tests

from bzrlib.memorytree import MemoryTree

from bzrlib.tests import TestCaseWithTransport

from bzrlib.treebuilder import TreeBuilder

class FakeTree(object):

    """A pretend tree to test the calls made by TreeBuilder."""

    def __init__(self):

        self._calls = []

    def lock_tree_write(self):

        self._calls.append("lock_tree_write")

    def unlock(self):

        self._calls.append("unlock")

class TestFakeTree(TestCaseWithTransport):

    def testFakeTree(self):

        """Check that FakeTree works as required for the TreeBuilder tests."""

        tree = FakeTree()

        self.assertEqual([], tree._calls)

        tree.lock_tree_write()

        self.assertEqual(["lock_tree_write"], tree._calls)

        tree.unlock()

        self.assertEqual(["lock_tree_write", "unlock"], tree._calls)

class TestTreeBuilderMemoryTree(tests.TestCaseWithMemoryTransport):

    def test_create(self):

        builder = TreeBuilder()

    def test_start_tree_locks_write(self):

        builder = TreeBuilder()

        tree = FakeTree()

        builder.start_tree(tree)

        self.assertEqual(["lock_tree_write"], tree._calls)

    def test_start_tree_when_started_fails(self):

        builder = TreeBuilder()

        tree = FakeTree()

        builder.start_tree(tree)

        self.assertRaises(errors.AlreadyBuilding, builder.start_tree, tree)

    def test_finish_tree_not_started_errors(self):

        builder = TreeBuilder()

        self.assertRaises(errors.NotBuilding, builder.finish_tree)

    def test_finish_tree_unlocks(self):

        builder = TreeBuilder()

        tree = FakeTree()

        builder.start_tree(tree)

        builder.finish_tree()

        self.assertEqual(["lock_tree_write", "unlock"], tree._calls)

    def test_build_tree_not_started_errors(self):

        builder = TreeBuilder()

        self.assertRaises(errors.NotBuilding, builder.build, "foo")

    def test_build_tree(self):

        """Test building works using a MemoryTree."""

        branch = self.make_branch('branch')

        tree = MemoryTree.create_on_branch(branch)

        builder = TreeBuilder()

        builder.start_tree(tree)

        builder.build(['foo', "bar/", "bar/file"])

        self.assertEqual('contents of foo\n',

            tree.get_file(tree.path2id('foo')).read())

        self.assertEqual('contents of bar/file\n',

            tree.get_file(tree.path2id('bar/file')).read())

        builder.finish_tree()