13
13
# You should have received a copy of the GNU General Public License
14
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
17
"""Tests for tree transform performance"""
26
26
from bzrlib.transform import TreeTransform
27
27
from bzrlib.workingtree import WorkingTree
30
29
class LinesDone(Exception):
31
"""Raised when `LineConsumer` reaches the required number of lines."""
35
32
class LineConsumer(object):
36
"""Count lines that are produced.
38
When the required number of lines have been reached, raise `LinesDone`.
41
34
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
35
self.required_lines = required_lines
49
37
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
38
self.required_lines -= text.count('\n')
59
39
if self.required_lines < 0:
63
43
class LogBenchmark(Benchmark):
64
"""Benchmarks for ``'bzr log'`` performance."""
66
45
def test_log(self):
67
"""Run log in a many-commit tree."""
46
"""Run log in a many-commit tree."""
68
47
tree = self.make_many_commit_tree(hardlink=True)
69
48
lf = log_formatter('long', to_file=StringIO())
70
49
self.time(show_log, tree.branch, lf, direction='reverse')
101
80
self.time(log_screenful)
103
82
def test_cmd_log(self):
104
"""Test execution of the log command."""
83
"""Test execution of the log command."""
105
84
tree = self.make_many_commit_tree(hardlink=True)
106
self.time(self.run_bzr, ['log', '-r', '-4..'])
85
self.time(self.run_bzr, 'log', '-r', '-4..')
108
87
def test_cmd_log_subprocess(self):
109
"""Text startup and execution of the log command."""
88
"""Text startup and execution of the log command."""
110
89
tree = self.make_many_commit_tree(hardlink=True)
111
90
self.time(self.run_bzr_subprocess, 'log', '-r', '-4..')