~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/benchmarks/bench_log.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2007-04-17 00:59:30 UTC
  • mfrom: (1551.15.4 Aaron's mergeable stuff)
  • Revision ID: pqm@pqm.ubuntu.com-20070417005930-rofskshyjsfzrahh
Fix ftp transport with servers that don't support atomic rename

Show diffs side-by-side

added added

removed removed

Lines of Context:
26
26
from bzrlib.transform import TreeTransform
27
27
from bzrlib.workingtree import WorkingTree
28
28
 
29
 
 
30
29
class LinesDone(Exception):
31
 
    """Raised when `LineConsumer` reaches the required number of lines."""
32
30
    pass
33
31
 
34
 
 
35
32
class LineConsumer(object):
36
 
    """Count lines that are produced.
37
 
 
38
 
    When the required number of lines have been reached, raise `LinesDone`.
39
 
    """
40
33
 
41
34
    def __init__(self, required_lines):
42
 
        """Create a new consumer.
43
 
 
44
 
        :param required_lines: How many lines must be produced.
45
 
        :type required_lines: integer
46
 
        """
47
35
        self.required_lines = required_lines
48
36
 
49
37
    def write(self, text):
50
 
        """Write some text to the black hole.
51
 
 
52
 
        But count how many lines have been written.
53
 
 
54
 
        :param text: A string that would be written.
55
 
        :raise LinesDone: when the required number of lines has been reached.
56
 
        :return: None
57
 
        """
58
38
        self.required_lines -= text.count('\n')
59
39
        if self.required_lines < 0:
60
40
            raise LinesDone()
61
 
 
 
41
        
62
42
 
63
43
class LogBenchmark(Benchmark):
64
 
    """Benchmarks for ``'bzr log'`` performance."""
65
44
 
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')