~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/benchmarks/bench_log.py

  • Committer: Andrew Bennetts
  • Date: 2007-03-26 06:24:01 UTC
  • mto: This revision was merged to the branch mainline in revision 2376.
  • Revision ID: andrew.bennetts@canonical.com-20070326062401-k3nbefzje5332jaf
Deal with review comments from Robert:

  * Add my name to the NEWS file
  * Move the test case to a new module in branch_implementations
  * Remove revision_history cruft from identitymap and test_identitymap
  * Improve some docstrings

Also, this fixes a bug where revision_history was not returning a copy of the
cached data, allowing the cache to be corrupted.

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')
103
82
    def test_cmd_log(self):
104
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..')
107
86
 
108
87
    def test_cmd_log_subprocess(self):
109
88
        """Text startup and execution of the log command."""