1
# Copyright (C) 2006 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Tests for tree transform performance"""
22
from bzrlib.benchmarks import Benchmark
23
from bzrlib.log import log_formatter, show_log
24
from bzrlib.osutils import pathjoin
25
from cStringIO import StringIO
26
from bzrlib.transform import TreeTransform
27
from bzrlib.workingtree import WorkingTree
30
class LinesDone(Exception):
31
"""Raised when `LineConsumer` reaches the required number of lines."""
35
class LineConsumer(object):
36
"""Count lines that are produced.
38
When the required number of lines have been reached, raise `LinesDone`.
41
def __init__(self, required_lines):
42
"""Create a new consumer.
44
:param required_lines: How many lines must be produced.
45
:type required_lines: integer
47
self.required_lines = required_lines
49
def write(self, text):
50
"""Write some text to the black hole.
52
But count how many lines have been written.
54
:param text: A string that would be written.
55
:raise LinesDone: when the required number of lines has been reached.
58
self.required_lines -= text.count('\n')
59
if self.required_lines < 0:
63
class LogBenchmark(Benchmark):
64
"""Benchmarks for ``'bzr log'`` performance."""
67
"""Run log in a many-commit tree."""
68
tree = self.make_many_commit_tree(hardlink=True)
69
lf = log_formatter('long', to_file=StringIO())
70
self.time(show_log, tree.branch, lf, direction='reverse')
72
def test_merge_log(self):
73
"""Run log in a tree with many merges"""
74
tree = self.make_heavily_merged_tree(hardlink=True)
75
lf = log_formatter('short', to_file=StringIO())
76
self.time(show_log, tree.branch, lf, direction='reverse')
78
def test_log_screenful(self):
79
"""Simulate log --long|less"""
80
self.screenful_tester('long')
82
def test_log_screenful_line(self):
83
"""Simulate log --line|less"""
84
self.screenful_tester('line')
86
def test_log_screenful_short(self):
87
"""Simulate log --short|less"""
88
self.screenful_tester('short')
90
def screenful_tester(self, formatter):
91
"""Run show_log, but stop after 25 lines are generated"""
92
tree = self.make_many_commit_tree(hardlink=True)
94
lf = log_formatter(formatter, to_file=LineConsumer(25))
96
show_log(tree.branch, lf, direction='reverse')
100
raise Exception, "LinesDone not raised"
101
self.time(log_screenful)
103
def test_cmd_log(self):
104
"""Test execution of the log command."""
105
tree = self.make_many_commit_tree(hardlink=True)
106
self.time(self.run_bzr, ['log', '-r', '-4..'])
108
def test_cmd_log_subprocess(self):
109
"""Text startup and execution of the log command."""
110
tree = self.make_many_commit_tree(hardlink=True)
111
self.time(self.run_bzr_subprocess, 'log', '-r', '-4..')
113
def test_log_verbose(self):
114
"""'verbose' log -- shows file changes"""
115
tree = self.make_many_commit_tree(hardlink=True)
116
lf = log_formatter('long', to_file=StringIO())
117
self.time(show_log, tree.branch, lf, direction='reverse', verbose=True)
added
removed
bzrlib/benchmarks/bench_log.py
