# Copyright (C) 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""Tests for bzr status performance."""


from bzrlib.benchmarks import Benchmark


class StatusBenchmark(Benchmark):
    """Benchmarks for ``'bzr status'``."""

    def test_no_ignored_unknown_kernel_like_tree(self):
        """Status in a kernel sized tree with no ignored or unknowns.

        This should be bearable (<2secs) fast.
        """
        self.make_kernel_like_added_tree()
        # on robertc's machine the first sample of this took 1687ms/15994ms
        self.time(self.run_bzr, 'status')

    def test_no_changes_known_kernel_like_tree(self):
        """Status in a kernel sized tree with no ignored, unknowns, or added."""
        self.make_kernel_like_committed_tree(link_bzr=True)
        self.time(self.run_bzr, 'status')

    def test_single_file_no_changes_known_kernel_like_tree(self):
        """Status in a kernel sized tree with no ignored, unknowns, or added
        of a single file."""
        self.make_kernel_like_committed_tree()
        #XXX depends on the filenames generated by make_kernel_like_tree
        self.time(self.run_bzr, ['status', '7/4/0/16'])

    def test_status_one_added_file_kernel_like_tree(self):
        """Status of a single added file in our stock large tree."""
        self.make_kernel_like_tree()
        self.run_bzr('add')
        self.run_bzr(['commit', '-m', 'initial import'])
        self.time(self.run_bzr, ['status', '3/3/3/10'])