1
# Copyright (C) 2006 by 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 version 2 as published by
5
# the Free Software Foundation.
7
# This program is distributed in the hope that it will be useful,
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
9
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10
# GNU General Public License for more details.
12
# You should have received a copy of the GNU General Public License
13
# along with this program; if not, write to the Free Software
14
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16
"""Tests for tree transform performance"""
21
from bzrlib.benchmarks import Benchmark
22
from bzrlib.log import log_formatter, show_log
23
from bzrlib.osutils import pathjoin
24
from cStringIO import StringIO
25
from bzrlib.transform import TreeTransform
26
from bzrlib.workingtree import WorkingTree
28
class LinesDone(Exception):
31
class LineConsumer(object):
33
def __init__(self, required_lines):
34
self.required_lines = required_lines
36
def write(self, text):
37
self.required_lines -= text.count('\n')
38
if self.required_lines < 0:
42
class LogBenchmark(Benchmark):
45
"""Run log in a many-commit tree."""
46
tree = self.make_many_commit_tree()
47
lf = log_formatter('long', to_file=StringIO())
48
self.time(show_log, tree.branch, lf, direction='reverse')
50
def test_merge_log(self):
51
"""Run log in a tree with many merges"""
52
tree = self.make_heavily_merged_tree()
53
lf = log_formatter('short', to_file=StringIO())
54
self.time(show_log, tree.branch, lf, direction='reverse')
56
def test_log_screenful(self):
57
"""Simulate log --long|less"""
58
self.screenful_tester('long')
60
def test_log_screenful_line(self):
61
"""Simulate log --line|less"""
62
self.screenful_tester('line')
64
def test_log_screenful_short(self):
65
"""Simulate log --short|less"""
66
self.screenful_tester('short')
68
def screenful_tester(self, formatter):
69
"""Run show_log, but stop after 25 lines are generated"""
70
tree = self.make_many_commit_tree()
72
lf = log_formatter(formatter, to_file=LineConsumer(25))
74
show_log(tree.branch, lf, direction='reverse')
78
raise Exception, "LinesDone not raised"
79
self.time(log_screenful)
81
def test_cmd_log(self):
82
"""Test execution of the log command."""
83
tree = self.make_many_commit_tree()
84
self.time(self.run_bzr, 'log', '-r', '-4..')
86
def test_cmd_log_subprocess(self):
87
"""Text startup and execution of the log command."""
88
tree = self.make_many_commit_tree()
89
self.time(self.run_bzr_subprocess, 'log', '-r', '-4..')
91
def test_log_verbose(self):
92
"""'verbose' log -- shows file changes"""
93
tree = self.make_many_commit_tree()
94
lf = log_formatter('long', to_file=StringIO())
95
self.time(show_log, tree.branch, lf, direction='reverse', verbose=True)
added
removed
bzrlib/benchmarks/bench_log.py
