~bzr-pqm/bzr/bzr.dev

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

# TODO: Perhaps there should be an API to find out if bzr running under the

# test suite -- some plugins might want to avoid making intrusive changes if

# this is the case.  However, we want behaviour under to test to diverge as

# little as possible, so this should be used rarely if it's added at all.

# (Suggestion from j-a-meinel, 2005-11-24)

# NOTE: Some classes in here use camelCaseNaming() rather than

# underscore_naming().  That's for consistency with unittest; it's not the

# general style of bzrlib.  Please continue that consistency when adding e.g.

# new assertFoo() methods.

import codecs

from cStringIO import StringIO

import difflib

import doctest

import errno

import logging

import os

from pprint import pformat

import random

import re

import stat

from subprocess import Popen, PIPE

import sys

import tempfile

import unittest

import time

from bzrlib import (

    bzrdir,

    debug,

    errors,

    memorytree,

    osutils,

    progress,

    ui,

    urlutils,

    workingtree,

    )

import bzrlib.branch

import bzrlib.commands

import bzrlib.timestamp

import bzrlib.export

import bzrlib.inventory

import bzrlib.iterablefile

import bzrlib.lockdir

try:

    import bzrlib.lsprof

except ImportError:

    # lsprof not available

    pass

from bzrlib.merge import merge_inner

import bzrlib.merge3

import bzrlib.osutils

import bzrlib.plugin

from bzrlib.revision import common_ancestor

import bzrlib.store

from bzrlib import symbol_versioning

import bzrlib.trace

from bzrlib.transport import get_transport

import bzrlib.transport

from bzrlib.transport.local import LocalURLServer

from bzrlib.transport.memory import MemoryServer

from bzrlib.transport.readonly import ReadonlyServer

from bzrlib.trace import mutter, note

from bzrlib.tests import TestUtil

from bzrlib.tests.HttpServer import HttpServer

from bzrlib.tests.TestUtil import (

                          TestSuite,

                          TestLoader,

                          )

from bzrlib.tests.treeshape import build_tree_contents

from bzrlib.workingtree import WorkingTree, WorkingTreeFormat2

# Mark this python module as being part of the implementation

# of unittest: this gives us better tracebacks where the last

# shown frame is the test code, not our assertXYZ.

__unittest = 1

default_transport = LocalURLServer

MODULES_TO_TEST = []

MODULES_TO_DOCTEST = [

                      bzrlib.timestamp,

                      bzrlib.errors,

                      bzrlib.export,

                      bzrlib.inventory,

                      bzrlib.iterablefile,

                      bzrlib.lockdir,

                      bzrlib.merge3,

                      bzrlib.option,

                      bzrlib.store,

                      ]

def packages_to_test():

    """Return a list of packages to test.

    The packages are not globally imported so that import failures are

    triggered when running selftest, not when importing the command.

    """

    import bzrlib.doc

    import bzrlib.tests.blackbox

    import bzrlib.tests.branch_implementations

    import bzrlib.tests.bzrdir_implementations

    import bzrlib.tests.interrepository_implementations

    import bzrlib.tests.interversionedfile_implementations

    import bzrlib.tests.intertree_implementations

    import bzrlib.tests.per_lock

    import bzrlib.tests.repository_implementations

    import bzrlib.tests.revisionstore_implementations

    import bzrlib.tests.tree_implementations

    import bzrlib.tests.workingtree_implementations

    return [

            bzrlib.doc,

            bzrlib.tests.blackbox,

            bzrlib.tests.branch_implementations,

            bzrlib.tests.bzrdir_implementations,

            bzrlib.tests.interrepository_implementations,

            bzrlib.tests.interversionedfile_implementations,

            bzrlib.tests.intertree_implementations,

            bzrlib.tests.per_lock,

            bzrlib.tests.repository_implementations,

            bzrlib.tests.revisionstore_implementations,

            bzrlib.tests.tree_implementations,

            bzrlib.tests.workingtree_implementations,

            ]

class ExtendedTestResult(unittest._TextTestResult):

    """Accepts, reports and accumulates the results of running tests.

    Compared to this unittest version this class adds support for profiling,

    benchmarking, stopping as soon as a test fails,  and skipping tests.

    There are further-specialized subclasses for different types of display.

    """

    stop_early = False

    def __init__(self, stream, descriptions, verbosity,

                 bench_history=None,

                 num_tests=None,

                 use_numbered_dirs=False,

                 ):

        """Construct new TestResult.

        :param bench_history: Optionally, a writable file object to accumulate

            benchmark results.

        """

        unittest._TextTestResult.__init__(self, stream, descriptions, verbosity)

        if bench_history is not None:

            from bzrlib.version import _get_bzr_source_tree

            src_tree = _get_bzr_source_tree()

            if src_tree:

                try:

                    revision_id = src_tree.get_parent_ids()[0]

                except IndexError:

                    # XXX: if this is a brand new tree, do the same as if there

                    # is no branch.

                    revision_id = ''

            else:

                # XXX: If there's no branch, what should we do?

                revision_id = ''

            bench_history.write("--date %s %s\n" % (time.time(), revision_id))

        self._bench_history = bench_history

        self.ui = ui.ui_factory

        self.num_tests = num_tests

        self.error_count = 0

        self.failure_count = 0

        self.known_failure_count = 0

        self.skip_count = 0

        self.unsupported = {}

        self.count = 0

        self.use_numbered_dirs = use_numbered_dirs

        self._overall_start_time = time.time()

    def extractBenchmarkTime(self, testCase):

        """Add a benchmark time for the current test case."""

        self._benchmarkTime = getattr(testCase, "_benchtime", None)

    def _elapsedTestTimeString(self):

        """Return a time string for the overall time the current test has taken."""

        return self._formatTime(time.time() - self._start_time)

    def _testTimeString(self):

        if self._benchmarkTime is not None:

            return "%s/%s" % (

                self._formatTime(self._benchmarkTime),

                self._elapsedTestTimeString())

        else:

            return "           %s" % self._elapsedTestTimeString()

    def _formatTime(self, seconds):

        """Format seconds as milliseconds with leading spaces."""

        # some benchmarks can take thousands of seconds to run, so we need 8

        # places

        return "%8dms" % (1000 * seconds)

    def _shortened_test_description(self, test):

        what = test.id()

        what = re.sub(r'^bzrlib\.(tests|benchmarks)\.', '', what)

        return what

    def startTest(self, test):

        unittest.TestResult.startTest(self, test)

        self.report_test_start(test)

        test.number = self.count

        self._recordTestStartTime()

    def _recordTestStartTime(self):

        """Record that a test has started."""

        self._start_time = time.time()

    def _cleanupLogFile(self, test):

        # We can only do this if we have one of our TestCases, not if

        # we have a doctest.

        setKeepLogfile = getattr(test, 'setKeepLogfile', None)

        if setKeepLogfile is not None:

            setKeepLogfile()

    def addError(self, test, err):

        self.extractBenchmarkTime(test)

        self._cleanupLogFile(test)

        if isinstance(err[1], TestSkipped):

            return self.addSkipped(test, err)

        elif isinstance(err[1], UnavailableFeature):

            return self.addNotSupported(test, err[1].args[0])

        unittest.TestResult.addError(self, test, err)

        self.error_count += 1

        self.report_error(test, err)

        if self.stop_early:

            self.stop()

    def addFailure(self, test, err):

        self._cleanupLogFile(test)

        self.extractBenchmarkTime(test)

        if isinstance(err[1], KnownFailure):

            return self.addKnownFailure(test, err)

        unittest.TestResult.addFailure(self, test, err)

        self.failure_count += 1

        self.report_failure(test, err)

        if self.stop_early:

            self.stop()

    def addKnownFailure(self, test, err):

        self.known_failure_count += 1

        self.report_known_failure(test, err)

    def addNotSupported(self, test, feature):

        self.unsupported.setdefault(str(feature), 0)

        self.unsupported[str(feature)] += 1

        self.report_unsupported(test, feature)

    def addSuccess(self, test):

        self.extractBenchmarkTime(test)

        if self._bench_history is not None:

            if self._benchmarkTime is not None:

                self._bench_history.write("%s %s\n" % (

                    self._formatTime(self._benchmarkTime),

                    test.id()))

        self.report_success(test)

        unittest.TestResult.addSuccess(self, test)

    def addSkipped(self, test, skip_excinfo):

        self.report_skip(test, skip_excinfo)

        # seems best to treat this as success from point-of-view of unittest

        # -- it actually does nothing so it barely matters :)

        try:

            test.tearDown()

        except KeyboardInterrupt:

            raise

        except:

            self.addError(test, test.__exc_info())

        else:

            unittest.TestResult.addSuccess(self, test)

    def printErrorList(self, flavour, errors):

        for test, err in errors:

            self.stream.writeln(self.separator1)

            self.stream.write("%s: " % flavour)

            if self.use_numbered_dirs:

                self.stream.write('#%d ' % test.number)

            self.stream.writeln(self.getDescription(test))

            if getattr(test, '_get_log', None) is not None:

                print >>self.stream

                print >>self.stream, \

                        ('vvvv[log from %s]' % test.id()).ljust(78,'-')

                print >>self.stream, test._get_log()

                print >>self.stream, \

                        ('^^^^[log from %s]' % test.id()).ljust(78,'-')

            self.stream.writeln(self.separator2)

            self.stream.writeln("%s" % err)

    def finished(self):

        pass

    def report_cleaning_up(self):

        pass

    def report_success(self, test):

        pass

class TextTestResult(ExtendedTestResult):

    """Displays progress and results of tests in text form"""

    def __init__(self, stream, descriptions, verbosity,

                 bench_history=None,

                 num_tests=None,

                 pb=None,

                 use_numbered_dirs=False,

                 ):

        ExtendedTestResult.__init__(self, stream, descriptions, verbosity,

            bench_history, num_tests, use_numbered_dirs)

        if pb is None:

            self.pb = self.ui.nested_progress_bar()

            self._supplied_pb = False

        else:

            self.pb = pb

            self._supplied_pb = True

        self.pb.show_pct = False

        self.pb.show_spinner = False

        self.pb.show_eta = False,

        self.pb.show_count = False

        self.pb.show_bar = False

    def report_starting(self):

        self.pb.update('[test 0/%d] starting...' % (self.num_tests))

    def _progress_prefix_text(self):

        a = '[%d' % self.count

        if self.num_tests is not None:

            a +='/%d' % self.num_tests

        a += ' in %ds' % (time.time() - self._overall_start_time)

        if self.error_count:

            a += ', %d errors' % self.error_count

        if self.failure_count:

            a += ', %d failed' % self.failure_count

        if self.known_failure_count:

            a += ', %d known failures' % self.known_failure_count

        if self.skip_count:

            a += ', %d skipped' % self.skip_count

        if self.unsupported:

            a += ', %d missing features' % len(self.unsupported)

        a += ']'

        return a

    def report_test_start(self, test):

        self.count += 1

        self.pb.update(

                self._progress_prefix_text()

                + ' ' 

                + self._shortened_test_description(test))

    def _test_description(self, test):

        if self.use_numbered_dirs:

            return '#%d %s' % (self.count,

                               self._shortened_test_description(test))

        else:

            return self._shortened_test_description(test)

    def report_error(self, test, err):

        self.pb.note('ERROR: %s\n    %s\n', 

            self._test_description(test),

            err[1],

            )

    def report_failure(self, test, err):

        self.pb.note('FAIL: %s\n    %s\n', 

            self._test_description(test),

            err[1],

            )

    def report_known_failure(self, test, err):

        self.pb.note('XFAIL: %s\n%s\n',

            self._test_description(test), err[1])

    def report_skip(self, test, skip_excinfo):

        self.skip_count += 1

        if False:

            # at the moment these are mostly not things we can fix

            # and so they just produce stipple; use the verbose reporter

            # to see them.

            if False:

                # show test and reason for skip

                self.pb.note('SKIP: %s\n    %s\n', 

                    self._shortened_test_description(test),

                    skip_excinfo[1])

            else:

                # since the class name was left behind in the still-visible

                # progress bar...

                self.pb.note('SKIP: %s', skip_excinfo[1])

    def report_unsupported(self, test, feature):

        """test cannot be run because feature is missing."""

    def report_cleaning_up(self):

        self.pb.update('cleaning up...')

    def finished(self):

        if not self._supplied_pb:

            self.pb.finished()

class VerboseTestResult(ExtendedTestResult):

    """Produce long output, with one line per test run plus times"""

    def _ellipsize_to_right(self, a_string, final_width):

        """Truncate and pad a string, keeping the right hand side"""

        if len(a_string) > final_width:

            result = '...' + a_string[3-final_width:]

        else:

            result = a_string

        return result.ljust(final_width)

    def report_starting(self):

        self.stream.write('running %d tests...\n' % self.num_tests)

    def report_test_start(self, test):

        self.count += 1

        name = self._shortened_test_description(test)

        # width needs space for 6 char status, plus 1 for slash, plus 2 10-char

        # numbers, plus a trailing blank

        # when NUMBERED_DIRS: plus 5 chars on test number, plus 1 char on space

        if self.use_numbered_dirs:

            self.stream.write('%5d ' % self.count)

            self.stream.write(self._ellipsize_to_right(name,

                                osutils.terminal_width()-36))

        else:

            self.stream.write(self._ellipsize_to_right(name,

                                osutils.terminal_width()-30))

        self.stream.flush()

    def _error_summary(self, err):

        indent = ' ' * 4

        if self.use_numbered_dirs:

            indent += ' ' * 6

        return '%s%s' % (indent, err[1])

    def report_error(self, test, err):

        self.stream.writeln('ERROR %s\n%s'

                % (self._testTimeString(),

                   self._error_summary(err)))

    def report_failure(self, test, err):

        self.stream.writeln(' FAIL %s\n%s'

                % (self._testTimeString(),

                   self._error_summary(err)))

    def report_known_failure(self, test, err):

        self.stream.writeln('XFAIL %s\n%s'

                % (self._testTimeString(),

                   self._error_summary(err)))

    def report_success(self, test):

        self.stream.writeln('   OK %s' % self._testTimeString())

        for bench_called, stats in getattr(test, '_benchcalls', []):

            self.stream.writeln('LSProf output for %s(%s, %s)' % bench_called)

            stats.pprint(file=self.stream)

        # flush the stream so that we get smooth output. This verbose mode is

        # used to show the output in PQM.

        self.stream.flush()

    def report_skip(self, test, skip_excinfo):

        self.skip_count += 1

        self.stream.writeln(' SKIP %s\n%s'

                % (self._testTimeString(),

                   self._error_summary(skip_excinfo)))

    def report_unsupported(self, test, feature):

        """test cannot be run because feature is missing."""

        self.stream.writeln("NODEP %s\n    The feature '%s' is not available."

                %(self._testTimeString(), feature))

class TextTestRunner(object):

    stop_on_failure = False

    def __init__(self,

                 stream=sys.stderr,

                 descriptions=0,

                 verbosity=1,

                 keep_output=False,

                 bench_history=None,

                 use_numbered_dirs=False,

                 list_only=False

                 ):

        self.stream = unittest._WritelnDecorator(stream)

        self.descriptions = descriptions

        self.verbosity = verbosity

        self.keep_output = keep_output

        self._bench_history = bench_history

        self.use_numbered_dirs = use_numbered_dirs

        self.list_only = list_only

    def run(self, test):

        "Run the given test case or test suite."

        startTime = time.time()

        if self.verbosity == 1:

            result_class = TextTestResult

        elif self.verbosity >= 2:

            result_class = VerboseTestResult

        result = result_class(self.stream,

                              self.descriptions,

                              self.verbosity,

                              bench_history=self._bench_history,

                              num_tests=test.countTestCases(),

                              use_numbered_dirs=self.use_numbered_dirs,

                              )

        result.stop_early = self.stop_on_failure

        result.report_starting()

        if self.list_only:

            if self.verbosity >= 2:

                self.stream.writeln("Listing tests only ...\n")

            run = 0

            for t in iter_suite_tests(test):

                self.stream.writeln("%s" % (t.id()))

                run += 1

            actionTaken = "Listed"

        else: 

            test.run(result)

            run = result.testsRun

            actionTaken = "Ran"

        stopTime = time.time()

        timeTaken = stopTime - startTime

        result.printErrors()

        self.stream.writeln(result.separator2)

        self.stream.writeln("%s %d test%s in %.3fs" % (actionTaken,

                            run, run != 1 and "s" or "", timeTaken))

        self.stream.writeln()

        if not result.wasSuccessful():

            self.stream.write("FAILED (")

            failed, errored = map(len, (result.failures, result.errors))

            if failed:

                self.stream.write("failures=%d" % failed)

            if errored:

                if failed: self.stream.write(", ")

                self.stream.write("errors=%d" % errored)

            if result.known_failure_count:

                if failed or errored: self.stream.write(", ")

                self.stream.write("known_failure_count=%d" %

                    result.known_failure_count)

            self.stream.writeln(")")

        else:

            if result.known_failure_count:

                self.stream.writeln("OK (known_failures=%d)" %

                    result.known_failure_count)

            else:

                self.stream.writeln("OK")

        if result.skip_count > 0:

            skipped = result.skip_count

            self.stream.writeln('%d test%s skipped' %

                                (skipped, skipped != 1 and "s" or ""))

        if result.unsupported:

            for feature, count in sorted(result.unsupported.items()):

                self.stream.writeln("Missing feature '%s' skipped %d tests." %

                    (feature, count))

        result.report_cleaning_up()

        # This is still a little bogus, 

        # but only a little. Folk not using our testrunner will

        # have to delete their temp directories themselves.

        test_root = TestCaseWithMemoryTransport.TEST_ROOT

        if result.wasSuccessful() or not self.keep_output:

            if test_root is not None:

                # If LANG=C we probably have created some bogus paths

                # which rmtree(unicode) will fail to delete

                # so make sure we are using rmtree(str) to delete everything

                # except on win32, where rmtree(str) will fail

                # since it doesn't have the property of byte-stream paths

                # (they are either ascii or mbcs)

                if sys.platform == 'win32':

                    # make sure we are using the unicode win32 api

                    test_root = unicode(test_root)

                else:

                    test_root = test_root.encode(

                        sys.getfilesystemencoding())

                _rmtree_temp_dir(test_root)

        else:

            note("Failed tests working directories are in '%s'\n", test_root)

        TestCaseWithMemoryTransport.TEST_ROOT = None

        result.finished()

        return result