~abentley/bzrtools/bzrtools.dev

#!/usr/bin/python

import bzrlib.tests

class ShelfTests(bzrlib.tests.TestCaseInTempDir):

    ORIGINAL = '\n\nhello test world\n\n'

    MODIFIED = '\n\ngoodbye test world\n\n'

    DIFF_HEADER = "=== modified file 'test_file'\n"

    DIFF_1 = """--- test_file\t

+++ test_file\t

@@ -1,4 +1,4 @@

-hello test world

+goodbye test world

"""

    DIFF_2 = """--- test_file\t

+++ test_file\t

@@ -1,4 +1,4 @@

-goodbye test world

+hello test world

"""

    def test_shelf(self):

        self.__test_loop(1)

    def test_shelf_multi(self):

        self.__test_loop(10)

    def __test_loop(self, count):

        from bzrlib.branch import Branch

        b = Branch.initialize('.')

        # Create a test file and commit it

        file('test_file', 'w').write(self.ORIGINAL)

        b.working_tree().add('test_file')

        b.working_tree().commit(message='add test_file')

        while count > 0:

            count -= 1

            # Modify the test file

            file('test_file', 'w').write(self.MODIFIED)

            # Check the diff is right

            self.assertEqual(self.capture('diff', retcode=1),

                self.DIFF_HEADER + self.DIFF_1 + '\n')

            # Shelve the changes

            self.run_bzr('shelve', '--all', retcode=True)

            # Make sure there is no diff anymore

            self.assertEqual(self.capture('diff', retcode=0), '')

            # Make sure the file is actually back the way it was

            self.assertEqual(file('test_file').read(), self.ORIGINAL)

            # Check the shelf is right

            shelf = b._transport.get('.bzr/x-shelf/default/00').read()

            self.assertEqual(shelf, self.DIFF_1)

            # Unshelve

            self.run_bzr('unshelve', retcode=True)

            # Check the diff is right again

            self.assertEqual(self.capture('diff', retcode=1),

                self.DIFF_HEADER + self.DIFF_1 + '\n')

            # Make sure the file is back the way it should be

            self.assertEqual(file('test_file').read(), self.MODIFIED)

    def test_shelf_nothing_to_shelve(self):

        import os.path

        from bzrlib.branch import Branch

        b = Branch.initialize('.')

        # Create a test file and commit it

        file('test_file', 'w').write(self.ORIGINAL)

        b.working_tree().add('test_file')

        b.working_tree().commit(message='add test_file')

        # Shelve the changes

        self.run_bzr('shelve', '--all', retcode=True)

        if b._transport.has('.bzr/x-shelf/default/00'):

            self.fail("Shelf exists, but it shouldn't")

    def test_shelf_with_revision(self):

        from bzrlib.branch import Branch

        b = Branch.initialize('.')

        # Create a test file and commit it

        file('test_file', 'w').write(self.ORIGINAL)

        b.working_tree().add('test_file')

        b.working_tree().commit(message='add test_file')

        # Modify the test file and commit it

        file('test_file', 'w').write(self.MODIFIED)

        b.working_tree().commit(message='update test_file')

        # Shelve the changes

        self.run_bzr('shelve', '-r', '1', '--all', retcode=True)

        # Check the diff is right

        self.assertEqual(self.capture('diff', retcode=1),

            self.DIFF_HEADER + self.DIFF_2 + '\n')

        # Make sure the file is the way it should be

        self.assertEqual(file('test_file').read(), self.ORIGINAL)

        # Unshelve

        self.run_bzr('unshelve', retcode=True)

        # Make sure the file is back the way it should be

        self.assertEqual(file('test_file').read(), self.MODIFIED)

    def test_shelf_with_two_revisions(self):

        from bzrlib.branch import Branch

        b = Branch.initialize('.')

        cmd = 'shelve -r 1..2'

        (stdout, stderr) = self.run_bzr_captured(cmd.split(), retcode=None)

        self.assertEqual(stderr.split('\n')[0],

            'bzr: ERROR: bzrlib.errors.BzrCommandError: shelve only ' \

                'accepts a single revision parameter.')