1
# Copyright (C) 2006 by Canonical Ltd
1
# Copyright (C) 2006 Canonical Ltd
3
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.
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.
7
8
# This program is distributed in the hope that it will be useful,
8
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
25
26
from bzrlib.transform import TreeTransform
26
27
from bzrlib.workingtree import WorkingTree
28
30
class LinesDone(Exception):
31
"""Raised when `LineConsumer` reaches the required number of lines."""
31
35
class LineConsumer(object):
36
"""Count lines that are produced.
38
When the required number of lines have been reached, raise `LinesDone`.
33
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
34
47
self.required_lines = required_lines
36
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.
37
58
self.required_lines -= text.count('\n')
38
59
if self.required_lines < 0:
42
63
class LogBenchmark(Benchmark):
64
"""Benchmarks for ``'bzr log'`` performance."""
44
66
def test_log(self):
45
"""Run log in a many-commit tree."""
67
"""Run log in a many-commit tree."""
46
68
tree = self.make_many_commit_tree(hardlink=True)
47
69
lf = log_formatter('long', to_file=StringIO())
48
70
self.time(show_log, tree.branch, lf, direction='reverse')