/brz/remove-bazaar

# Copyright (C) 2005 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

"""\

Blackbox tests for version_info

"""

import os

import sys

from bzrlib.tests import TestCase, TestCaseInTempDir

class TestVersionInfo(TestCaseInTempDir):

    def test_invalid_format(self):

        bzr = self.run_bzr

        bzr('version-info', '--format', 'quijibo', retcode=3)

    def create_branch(self):

        bzr = self.run_bzr

        os.mkdir('branch')

        os.chdir('branch')

        bzr('init')

        open('a', 'wb').write('a file\n')

        bzr('add')

        bzr('commit', '-m', 'adding a')

        open('b', 'wb').write('b file\n')

        bzr('add')

        bzr('commit', '-m', 'adding b')

        os.chdir('..')

    def get_revisions(self):

        os.chdir('branch')

        revisions = self.run_bzr('revision-history')[0].strip().split('\n')

        os.chdir('..')

        return revisions

    def test_rio(self):

        self.create_branch()

        def regen(*args):

            return self.run_bzr('version-info', '--format', 'rio', 

                                'branch', *args)[0]

        revisions = self.get_revisions()

        txt = regen()

        self.assertContainsRe(txt, 'date:')

        self.assertContainsRe(txt, 'revno: 2')

        self.assertContainsRe(txt, 'revision_id: ' + revisions[-1])

        txt = regen('--all')

        self.assertContainsRe(txt, 'date:')

        self.assertContainsRe(txt, 'revno: 2')

        self.assertContainsRe(txt, 'revision_id: ' + revisions[-1])

        self.assertContainsRe(txt, 'clean: True')

        self.assertContainsRe(txt, 'revisions:')

        for rev_id in revisions:

            self.assertContainsRe(txt, 'id: ' + rev_id)

        self.assertContainsRe(txt, 'message: adding a')

        self.assertContainsRe(txt, 'message: adding b')

        txt = regen('--check-clean')

        self.assertContainsRe(txt, 'clean: True')

        open('branch/c', 'wb').write('now unclean\n')

        txt = regen('--check-clean')

        self.assertContainsRe(txt, 'clean: False')

        os.remove('branch/c')

        # Make sure it works without a directory

        os.chdir('branch')

        txt = self.run_bzr('version-info', '--format', 'rio')

    def test_python(self):

        def bzr(*args, **kwargs):

            return self.run_bzr(*args, **kwargs)[0]

        def regen(*args):

            txt = self.run_bzr('version-info', '--format', 'python',

                               'branch', *args)[0]

            outf = open('test_version_information.py', 'wb')

            outf.write(txt)

            outf.close()

            try:

                sys.path.append(os.getcwdu())

                import test_version_information as tvi

                reload(tvi)

            finally:

                sys.path.pop()

            # Make sure the module isn't cached

            sys.modules.pop('tvi', None)

            sys.modules.pop('test_version_information', None)

            # Delete the compiled versions, because we are generating

            # a new file fast enough that python doen't detect it

            # needs to recompile, and using sleep() just makes the

            # test slow

            if os.path.exists('test_version_information.pyc'):

                os.remove('test_version_information.pyc')

            if os.path.exists('test_version_information.pyo'):

                os.remove('test_version_information.pyo')

            return tvi

        self.create_branch()

        revisions = self.get_revisions()

        tvi = regen()

        self.assertEqual(tvi.version_info['revno'], 2)

        self.failUnless(tvi.version_info.has_key('date'))

        self.assertEqual(revisions[-1], tvi.version_info['revision_id'])

        tvi = regen('--all')

        self.assertEqual([(revisions[0], 'adding a'),

                          (revisions[1], 'adding b')],

                         tvi.version_info['revisions'])

        self.assertEqual(True, tvi.version_info['clean'])