/brz/remove-bazaar

# Copyright (C) 2005 by Canonical Ltd

# -*- coding: utf-8 -*-

# 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

"""Black-box tests for bzr.

These check that it behaves properly when it's invoked through the regular

command-line interface. This doesn't actually run a new interpreter but 

rather starts again from the run_bzr function.

"""

from cStringIO import StringIO

import os

import re

import shutil

import sys

from bzrlib.branch import Branch

from bzrlib.clone import copy_branch

from bzrlib.errors import BzrCommandError

from bzrlib.osutils import has_symlinks

from bzrlib.selftest import TestCaseInTempDir, BzrTestBase

from bzrlib.selftest.HTTPTestUtil import TestCaseWithWebserver

class ExternalBase(TestCaseInTempDir):

    def runbzr(self, args, retcode=0, backtick=False):

        if isinstance(args, basestring):

            args = args.split()

        if backtick:

            return self.run_bzr_captured(args, retcode=retcode)[0]

        else:

            return self.run_bzr_captured(args, retcode=retcode)

class TestCommands(ExternalBase):

    def test_help_commands(self):

        self.runbzr('--help')

        self.runbzr('help')

        self.runbzr('help commands')

        self.runbzr('help help')

        self.runbzr('commit -h')

    def test_init_branch(self):

        self.runbzr(['init'])

    def test_whoami(self):

        # this should always identify something, if only "john@localhost"

        self.runbzr("whoami")

        self.runbzr("whoami --email")

        self.assertEquals(self.runbzr("whoami --email",

                                      backtick=True).count('@'), 1)

    def test_whoami_branch(self):

        """branch specific user identity works."""

        self.runbzr('init')

        f = file('.bzr/email', 'wt')

        f.write('Branch Identity <branch@identi.ty>')

        f.close()

        bzr_email = os.environ.get('BZREMAIL')

        if bzr_email is not None:

            del os.environ['BZREMAIL']

        whoami = self.runbzr("whoami",backtick=True)

        whoami_email = self.runbzr("whoami --email",backtick=True)

        self.assertTrue(whoami.startswith('Branch Identity <branch@identi.ty>'))

        self.assertTrue(whoami_email.startswith('branch@identi.ty'))

        # Verify that the environment variable overrides the value 

        # in the file

        os.environ['BZREMAIL'] = 'Different ID <other@environ.ment>'

        whoami = self.runbzr("whoami",backtick=True)

        whoami_email = self.runbzr("whoami --email",backtick=True)

        self.assertTrue(whoami.startswith('Different ID <other@environ.ment>'))

        self.assertTrue(whoami_email.startswith('other@environ.ment'))

        if bzr_email is not None:

            os.environ['BZREMAIL'] = bzr_email

    def test_invalid_commands(self):

        self.runbzr("pants", retcode=1)

        self.runbzr("--pants off", retcode=1)

        self.runbzr("diff --message foo", retcode=1)

    def test_empty_commit(self):

        self.runbzr("init")

        self.build_tree(['hello.txt'])

        self.runbzr("commit -m empty", retcode=1)

        self.runbzr("add hello.txt")

        self.runbzr("commit -m added")

    def test_empty_commit_message(self):

        self.runbzr("init")

        file('foo.c', 'wt').write('int main() {}')

        self.runbzr(['add', 'foo.c'])

        self.runbzr(["commit", "-m", ""] , retcode=1) 

    def test_ignore_patterns(self):

        from bzrlib.branch import Branch

        b = Branch.initialize('.')

        self.assertEquals(list(b.unknowns()), [])

        file('foo.tmp', 'wt').write('tmp files are ignored')

        self.assertEquals(list(b.unknowns()), [])

        assert self.capture('unknowns') == ''

        file('foo.c', 'wt').write('int main() {}')

        self.assertEquals(list(b.unknowns()), ['foo.c'])

        assert self.capture('unknowns') == 'foo.c\n'

        self.runbzr(['add', 'foo.c'])

        assert self.capture('unknowns') == ''

        # 'ignore' works when creating the .bzignore file

        file('foo.blah', 'wt').write('blah')

        self.assertEquals(list(b.unknowns()), ['foo.blah'])

        self.runbzr('ignore *.blah')

        self.assertEquals(list(b.unknowns()), [])

        assert file('.bzrignore', 'rU').read() == '*.blah\n'

        # 'ignore' works when then .bzrignore file already exists

        file('garh', 'wt').write('garh')

        self.assertEquals(list(b.unknowns()), ['garh'])

        assert self.capture('unknowns') == 'garh\n'

        self.runbzr('ignore garh')

        self.assertEquals(list(b.unknowns()), [])

        assert file('.bzrignore', 'rU').read() == '*.blah\ngarh\n'

    def test_revert(self):

        self.runbzr('init')

        file('hello', 'wt').write('foo')

        self.runbzr('add hello')

        self.runbzr('commit -m setup hello')

        file('goodbye', 'wt').write('baz')

        self.runbzr('add goodbye')

        self.runbzr('commit -m setup goodbye')

        file('hello', 'wt').write('bar')

        file('goodbye', 'wt').write('qux')

        self.runbzr('revert hello')

        self.check_file_contents('hello', 'foo')

        self.check_file_contents('goodbye', 'qux')

        self.runbzr('revert')

        self.check_file_contents('goodbye', 'baz')

        os.mkdir('revertdir')

        self.runbzr('add revertdir')

        self.runbzr('commit -m f')

        os.rmdir('revertdir')

        self.runbzr('revert')

        os.symlink('/unlikely/to/exist', 'symlink')

        self.runbzr('add symlink')

        self.runbzr('commit -m f')

        os.unlink('symlink')

        self.runbzr('revert')

        self.failUnlessExists('symlink')

        os.unlink('symlink')

        os.symlink('a-different-path', 'symlink')

        self.runbzr('revert')

        self.assertEqual('/unlikely/to/exist',

                         os.readlink('symlink'))

        file('hello', 'wt').write('xyz')

        self.runbzr('commit -m xyz hello')

        self.runbzr('revert -r 1 hello')

        self.check_file_contents('hello', 'foo')

        self.runbzr('revert hello')

        self.check_file_contents('hello', 'xyz')

        os.chdir('revertdir')

        self.runbzr('revert')

        os.chdir('..')

    def test_mv_modes(self):

        """Test two modes of operation for mv"""

        from bzrlib.branch import Branch

        b = Branch.initialize('.')

        self.build_tree(['a', 'c', 'subdir/'])

        self.run_bzr_captured(['add', self.test_dir])

        self.run_bzr_captured(['mv', 'a', 'b'])

        self.run_bzr_captured(['mv', 'b', 'subdir'])

        self.run_bzr_captured(['mv', 'subdir/b', 'a'])

        self.run_bzr_captured(['mv', 'a', 'c', 'subdir'])

        self.run_bzr_captured(['mv', 'subdir/a', 'subdir/newa'])

    def test_main_version(self):

        """Check output from version command and master option is reasonable"""

        # output is intentionally passed through to stdout so that we

        # can see the version being tested

        output = self.runbzr('version', backtick=1)

        self.log('bzr version output:')

        self.log(output)

        self.assert_(output.startswith('bzr (bazaar-ng) '))

        self.assertNotEqual(output.index('Canonical'), -1)

        # make sure --version is consistent

        tmp_output = self.runbzr('--version', backtick=1)

        self.log('bzr --version output:')

        self.log(tmp_output)

        self.assertEquals(output, tmp_output)

    def example_branch(test):

        test.runbzr('init')

        file('hello', 'wt').write('foo')

        test.runbzr('add hello')

        test.runbzr('commit -m setup hello')

        file('goodbye', 'wt').write('baz')

        test.runbzr('add goodbye')

        test.runbzr('commit -m setup goodbye')

    def test_export(self):

        os.mkdir('branch')

        os.chdir('branch')

        self.example_branch()

        self.runbzr('export ../latest')

        self.assertEqual(file('../latest/goodbye', 'rt').read(), 'baz')

        self.runbzr('export ../first -r 1')

        assert not os.path.exists('../first/goodbye')

        self.assertEqual(file('../first/hello', 'rt').read(), 'foo')

        self.runbzr('export ../first.gz -r 1')

        self.assertEqual(file('../first.gz/hello', 'rt').read(), 'foo')

        self.runbzr('export ../first.bz2 -r 1')

        self.assertEqual(file('../first.bz2/hello', 'rt').read(), 'foo')

        self.runbzr('export ../first.tar -r 1')

        assert os.path.isfile('../first.tar')

        from tarfile import TarFile

        tf = TarFile('../first.tar')

        assert 'first/hello' in tf.getnames(), tf.getnames()

        self.assertEqual(tf.extractfile('first/hello').read(), 'foo')

        self.runbzr('export ../first.tar.gz -r 1')

        assert os.path.isfile('../first.tar.gz')

        self.runbzr('export ../first.tbz2 -r 1')

        assert os.path.isfile('../first.tbz2')

        self.runbzr('export ../first.tar.bz2 -r 1')

        assert os.path.isfile('../first.tar.bz2')

        self.runbzr('export ../first.tar.tbz2 -r 1')

        assert os.path.isfile('../first.tar.tbz2')

        from bz2 import BZ2File

        tf = TarFile('../first.tar.tbz2', 

                     fileobj=BZ2File('../first.tar.tbz2', 'r'))

        assert 'first.tar/hello' in tf.getnames(), tf.getnames()

        self.assertEqual(tf.extractfile('first.tar/hello').read(), 'foo')

        self.runbzr('export ../first2.tar -r 1 --root pizza')

        tf = TarFile('../first2.tar')

        assert 'pizza/hello' in tf.getnames(), tf.getnames()

    def test_diff(self):

        self.example_branch()

        file('hello', 'wt').write('hello world!')

        self.runbzr('commit -m fixing hello')

        output = self.runbzr('diff -r 2..3', backtick=1)

        self.assert_('\n+hello world!' in output)

        output = self.runbzr('diff -r last:3..last:1', backtick=1)

        self.assert_('\n+baz' in output)

    def test_diff_branches(self):

        self.build_tree(['branch1/', 'branch1/file', 'branch2/'])

        branch = Branch.initialize('branch1')

        branch.add(['file'])

        branch.commit('add file')

        copy_branch(branch, 'branch2')

        print >> open('branch2/file', 'w'), 'new content'

        branch2 = Branch.open('branch2')

        branch2.commit('update file')

        # should open branch1 and diff against branch2, 

        output = self.run_bzr_captured(['diff', '-r', 'branch:branch2', 'branch1'])

        self.assertEquals(("=== modified file 'file'\n"

                           "--- file\n"

                           "+++ file\n"

                           "@@ -1,1 +1,1 @@\n"

                           "-new content\n"

                           "+contents of branch1/file\n"

                           "\n", ''), output)

    def test_branch(self):

        """Branch from one branch to another."""

        os.mkdir('a')

        os.chdir('a')

        self.example_branch()

        os.chdir('..')

        self.runbzr('branch a b')

        self.runbzr('branch a c -r 1')

        os.chdir('b')

        self.runbzr('commit -m foo --unchanged')

        os.chdir('..')

        # naughty - abstraction violations RBC 20050928  

        print "test_branch used to delete the stores, how is this meant to work ?"

        #shutil.rmtree('a/.bzr/revision-store')

        #shutil.rmtree('a/.bzr/inventory-store', ignore_errors=True)

        #shutil.rmtree('a/.bzr/text-store', ignore_errors=True)

        self.runbzr('branch a d --basis b')

    def test_merge(self):

        from bzrlib.branch import Branch

        os.mkdir('a')

        os.chdir('a')

        self.example_branch()

        os.chdir('..')

        self.runbzr('branch a b')

        os.chdir('b')

        file('goodbye', 'wt').write('quux')

        self.runbzr(['commit',  '-m',  "more u's are always good"])

        os.chdir('../a')

        file('hello', 'wt').write('quuux')

        # We can't merge when there are in-tree changes

        self.runbzr('merge ../b', retcode=1)

        self.runbzr(['commit', '-m', "Like an epidemic of u's"])

        self.runbzr('merge ../b')

        self.check_file_contents('goodbye', 'quux')

        # Merging a branch pulls its revision into the tree

        a = Branch.open('.')

        b = Branch.open('../b')

        a.get_revision_xml(b.last_revision())

        self.log('pending merges: %s', a.pending_merges())

        #        assert a.pending_merges() == [b.last_revision()], "Assertion %s %s" \

        #        % (a.pending_merges(), b.last_patch())

    def test_merge_with_missing_file(self):

        """Merge handles missing file conflicts"""

        os.mkdir('a')

        os.chdir('a')

        os.mkdir('sub')

        print >> file('sub/a.txt', 'wb'), "hello"

        print >> file('b.txt', 'wb'), "hello"

        print >> file('sub/c.txt', 'wb'), "hello"

        self.runbzr('init')

        self.runbzr('add')

        self.runbzr(('commit', '-m', 'added a'))

        self.runbzr('branch . ../b')

        print >> file('sub/a.txt', 'ab'), "there"

        print >> file('b.txt', 'ab'), "there"

        print >> file('sub/c.txt', 'ab'), "there"

        self.runbzr(('commit', '-m', 'Added there'))

        os.unlink('sub/a.txt')

        os.unlink('sub/c.txt')

        os.rmdir('sub')

        os.unlink('b.txt')

        self.runbzr(('commit', '-m', 'Removed a.txt'))

        os.chdir('../b')

        print >> file('sub/a.txt', 'ab'), "something"

        print >> file('b.txt', 'ab'), "something"

        print >> file('sub/c.txt', 'ab'), "something"

        self.runbzr(('commit', '-m', 'Modified a.txt'))

        self.runbzr('merge ../a/')

        assert os.path.exists('sub/a.txt.THIS')

        assert os.path.exists('sub/a.txt.BASE')

        os.chdir('../a')

        self.runbzr('merge ../b/')

        assert os.path.exists('sub/a.txt.OTHER')

        assert os.path.exists('sub/a.txt.BASE')

    def test_pull(self):

        """Pull changes from one branch to another."""

        os.mkdir('a')

        os.chdir('a')

        self.example_branch()

        self.runbzr('pull', retcode=1)

        self.runbzr('missing', retcode=1)

        self.runbzr('missing .')

        self.runbzr('missing')

        self.runbzr('pull')

        self.runbzr('pull /', retcode=1)

        self.runbzr('pull')

        os.chdir('..')

        self.runbzr('branch a b')

        os.chdir('b')

        self.runbzr('pull')

        os.mkdir('subdir')

        self.runbzr('add subdir')

        self.runbzr('commit -m blah --unchanged')

        os.chdir('../a')

        a = Branch.open('.')

        b = Branch.open('../b')

        assert a.revision_history() == b.revision_history()[:-1]

        self.runbzr('pull ../b')

        assert a.revision_history() == b.revision_history()

        self.runbzr('commit -m blah2 --unchanged')

        os.chdir('../b')

        self.runbzr('commit -m blah3 --unchanged')

        # no clobber

        self.runbzr('pull ../a', retcode=1)

        os.chdir('..')

        self.runbzr('branch b clobberme')

        os.chdir('clobberme')

        self.runbzr('pull --clobber ../a')

        clobbered = Branch.open('.')

        self.assertEqual(clobbered.revision_history(),

                         a.revision_history())

        os.chdir('../a')

        self.runbzr('merge ../b')

        self.runbzr('commit -m blah4 --unchanged')

        os.chdir('../b/subdir')

        self.runbzr('pull ../../a')

        assert a.revision_history()[-1] == b.revision_history()[-1]

        self.runbzr('commit -m blah5 --unchanged')

        self.runbzr('commit -m blah6 --unchanged')

        os.chdir('..')

        self.runbzr('pull ../a')

        os.chdir('../a')

        self.runbzr('commit -m blah7 --unchanged')

        self.runbzr('merge ../b')

        self.runbzr('commit -m blah8 --unchanged')

        self.runbzr('pull ../b')

        self.runbzr('pull ../b')

    def test_locations(self):

        """Using and remembering different locations"""

        os.mkdir('a')

        os.chdir('a')

        self.runbzr('init')

        self.runbzr('commit -m unchanged --unchanged')

        self.runbzr('pull', retcode=1)

        self.runbzr('merge', retcode=1)

        self.runbzr('branch . ../b')

        os.chdir('../b')

        self.runbzr('pull')

        self.runbzr('branch . ../c')

        self.runbzr('pull ../c')

        self.runbzr('merge')

        os.chdir('../a')

        self.runbzr('pull ../b')

        self.runbzr('pull')

        self.runbzr('pull ../c')

        self.runbzr('branch ../c ../d')

        shutil.rmtree('../c')

        self.runbzr('pull')

        os.chdir('../b')

        self.runbzr('pull')

        os.chdir('../d')

        self.runbzr('pull', retcode=1)

        self.runbzr('pull ../a --remember')

        self.runbzr('pull')

    def test_add_reports(self):

        """add command prints the names of added files."""

        b = Branch.initialize('.')

        self.build_tree(['top.txt', 'dir/', 'dir/sub.txt'])

        out = self.run_bzr_captured(['add'], retcode = 0)[0]

        # the ordering is not defined at the moment

        results = sorted(out.rstrip('\n').split('\n'))

        self.assertEquals(['added dir',

                           'added dir'+os.sep+'sub.txt',

                           'added top.txt',],

                          results)

    def test_add_quiet_is(self):

        """add -q does not print the names of added files."""

        b = Branch.initialize('.')

        self.build_tree(['top.txt', 'dir/', 'dir/sub.txt'])

        out = self.run_bzr_captured(['add', '-q'], retcode = 0)[0]

        # the ordering is not defined at the moment

        results = sorted(out.rstrip('\n').split('\n'))

        self.assertEquals([''], results)

    def test_unknown_command(self):

        """Handling of unknown command."""

        out, err = self.run_bzr_captured(['fluffy-badger'],

                                         retcode=1)

        self.assertEquals(out, '')

        err.index('unknown command')

    def test_conflicts(self):

        """Handling of merge conflicts"""

        os.mkdir('base')

        os.chdir('base')

        file('hello', 'wb').write("hi world")

        file('answer', 'wb').write("42")

        self.runbzr('init')

        self.runbzr('add')

        self.runbzr('commit -m base')

        self.runbzr('branch . ../other')

        self.runbzr('branch . ../this')

        os.chdir('../other')

        file('hello', 'wb').write("Hello.")

        file('answer', 'wb').write("Is anyone there?")

        self.runbzr('commit -m other')

        os.chdir('../this')

        file('hello', 'wb').write("Hello, world")

        self.runbzr('mv answer question')

        file('question', 'wb').write("What do you get when you multiply six"

                                   "times nine?")

        self.runbzr('commit -m this')

        self.runbzr('merge ../other')

        result = self.runbzr('conflicts', backtick=1)

        self.assertEquals(result, "hello\nquestion\n")

        result = self.runbzr('status', backtick=1)

        assert "conflicts:\n  hello\n  question\n" in result, result

        self.runbzr('resolve hello')

        result = self.runbzr('conflicts', backtick=1)

        self.assertEquals(result, "question\n")

        self.runbzr('commit -m conflicts', retcode=1)

        self.runbzr('resolve --all')

        result = self.runbzr('conflicts', backtick=1)

        self.runbzr('commit -m conflicts')

        self.assertEquals(result, "")

    def test_resign(self):

        """Test re signing of data."""

        import bzrlib.gpg

        oldstrategy = bzrlib.gpg.GPGStrategy

        branch = Branch.initialize('.')

        branch.commit("base", allow_pointless=True, rev_id='A')

        try:

            # monkey patch gpg signing mechanism

            from bzrlib.testament import Testament

            bzrlib.gpg.GPGStrategy = bzrlib.gpg.LoopbackGPGStrategy

            self.runbzr('re-sign -r revid:A')

            self.assertEqual(Testament.from_revision(branch,'A').as_short_text(),

                             branch.revision_store.get('A', 'sig').read())

        finally:

            bzrlib.gpg.GPGStrategy = oldstrategy

def listdir_sorted(dir):

    L = os.listdir(dir)

    L.sort()

    return L

class OldTests(ExternalBase):

    """old tests moved from ./testbzr."""

    def test_bzr(self):

        from os import chdir, mkdir

        from os.path import exists

        runbzr = self.runbzr

        capture = self.capture

        progress = self.log

        progress("basic branch creation")

        mkdir('branch1')

        chdir('branch1')

        runbzr('init')

        self.assertEquals(capture('root').rstrip(),

                          os.path.join(self.test_dir, 'branch1'))

        progress("status of new file")

        f = file('test.txt', 'wt')

        f.write('hello world!\n')

        f.close()

        self.assertEquals(capture('unknowns'), 'test.txt\n')

        out = capture("status")

        assert out == 'unknown:\n  test.txt\n'

        out = capture("status --all")

        assert out == "unknown:\n  test.txt\n"

        out = capture("status test.txt --all")

        assert out == "unknown:\n  test.txt\n"

        f = file('test2.txt', 'wt')

        f.write('goodbye cruel world...\n')

        f.close()

        out = capture("status test.txt")

        assert out == "unknown:\n  test.txt\n"

        out = capture("status")

        assert out == ("unknown:\n"

                       "  test.txt\n"

                       "  test2.txt\n")