/brz/remove-bazaar

# Copyright (C) 2005 by Canonical Ltd

# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import os

from bzrlib.selftest import TestCaseInTempDir

from bzrlib.branch import Branch

from bzrlib.commit import Commit

class TestCommit(TestCaseInTempDir):

    def test_simple_commit(self):

        """Commit and check two versions of a single file."""

        b = Branch('.', init=True)

        file('hello', 'w').write('hello world')

        b.add('hello')

        b.commit(message='add hello')

        file_id = b.working_tree().path2id('hello')

        file('hello', 'w').write('version 2')

        b.commit(message='commit 2')

        eq = self.assertEquals

        eq(b.revno(), 2)

        rh = b.revision_history()

        rev = b.get_revision(rh[0])

        eq(rev.message, 'add hello')

        tree1 = b.revision_tree(rh[0])

        text = tree1.get_file_text(file_id)

        eq(text, 'hello world')

        tree2 = b.revision_tree(rh[1])

        eq(tree2.get_file_text(file_id), 'version 2')

    def test_delete_commit(self):

        """Test a commit with a deleted file"""

        b = Branch('.', init=True)

        file('hello', 'w').write('hello world')

        b.add(['hello'], ['hello-id'])

        b.commit(message='add hello')

        os.remove('hello')

        b.commit('removed hello', rev_id='rev2')

        tree = b.revision_tree('rev2')

        self.assertFalse(tree.has_id('hello-id'))

    def test_removed_commit(self):

        """Test a commit with a removed file"""

        b = Branch('.', init=True)

        file('hello', 'w').write('hello world')

        b.add(['hello'], ['hello-id'])

        b.commit(message='add hello')

        b.remove('hello')

        b.commit('removed hello', rev_id='rev2')

        tree = b.revision_tree('rev2')

        self.assertFalse(tree.has_id('hello-id'))

if __name__ == '__main__':

    import unittest

    unittest.main()