~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: 2006-08-17 07:52:09 UTC
  • mfrom: (1910.3.4 trivial)
  • Revision ID: pqm@pqm.ubuntu.com-20060817075209-e85a1f9e05ff8b87
(andrew) Trivial fixes to NotImplemented errors.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006 by Canonical Ltd
2
2
#
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 as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
 
4
# it under the terms of the GNU General Public License version 2 as published by
 
5
# the Free Software Foundation.
7
6
#
8
7
# This program is distributed in the hope that it will be useful,
9
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
26
25
from bzrlib.transform import TreeTransform
27
26
from bzrlib.workingtree import WorkingTree
28
27
 
29
 
 
30
28
class LinesDone(Exception):
31
 
    """Raised when `LineConsumer` reaches the required number of lines."""
32
29
    pass
33
30
 
34
 
 
35
31
class LineConsumer(object):
36
 
    """Count lines that are produced.
37
 
 
38
 
    When the required number of lines have been reached, raise `LinesDone`.
39
 
    """
40
32
 
41
33
    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
34
        self.required_lines = required_lines
48
35
 
49
36
    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
37
        self.required_lines -= text.count('\n')
59
38
        if self.required_lines < 0:
60
39
            raise LinesDone()
61
 
 
 
40
        
62
41
 
63
42
class LogBenchmark(Benchmark):
64
 
    """Benchmarks for ``'bzr log'`` performance."""
65
43
 
66
44
    def test_log(self):
67
 
        """Run log in a many-commit tree."""
 
45
        """Run log in a many-commit tree.""" 
68
46
        tree = self.make_many_commit_tree(hardlink=True)
69
47
        lf = log_formatter('long', to_file=StringIO())
70
48
        self.time(show_log, tree.branch, lf, direction='reverse')
103
81
    def test_cmd_log(self):
104
82
        """Test execution of the log command.""" 
105
83
        tree = self.make_many_commit_tree(hardlink=True)
106
 
        self.time(self.run_bzr, ['log', '-r', '-4..'])
 
84
        self.time(self.run_bzr, 'log', '-r', '-4..')
107
85
 
108
86
    def test_cmd_log_subprocess(self):
109
87
        """Text startup and execution of the log command."""