/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/benchmarks/__init__.py

Switch build_tree to use non_atomic_put()

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/benchmarks/__init__.py
 
1
# Copyright (C) 2006 by Canonical Ltd
 
2
#
 
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.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""Benchmark test suite for bzr."""
 
18
 
 
19
from bzrlib import (
 
20
    plugin,
 
21
    )
 
22
from bzrlib.tests.TestUtil import TestLoader
 
23
from bzrlib.tests.blackbox import ExternalBase
 
24
 
 
25
 
 
26
class Benchmark(ExternalBase):
 
27
 
 
28
    def make_kernel_like_tree(self, url=None, root='.',
 
29
                              link_working=False):
 
30
        """Setup a temporary tree roughly like a kernel tree.
 
31
        
 
32
        :param url: Creat the kernel like tree as a lightweight checkout
 
33
        of a new branch created at url.
 
34
        :param link_working: instead of creating a new copy of all files
 
35
            just hardlink the working tree. Tests must request this, because
 
36
            they must break links if they want to change the files
 
37
        """
 
38
        from bzrlib.benchmarks.tree_creator.kernel_like import (
 
39
            KernelLikeTreeCreator,
 
40
            )
 
41
        creator = KernelLikeTreeCreator(self, link_working=link_working,
 
42
                                        url=url)
 
43
        return creator.create(root=root)
 
44
 
 
45
    def make_kernel_like_added_tree(self, root='.',
 
46
                                    link_working=True,
 
47
                                    hot_cache=True):
 
48
        """Make a kernel like tree, with all files added
 
49
 
 
50
        :param root: Where to create the files
 
51
        :param link_working: Instead of copying all of the working tree
 
52
            files, just hardlink them to the cached files. Tests can unlink
 
53
            files that they will change.
 
54
        :param hot_cache: Run through the newly created tree and make sure
 
55
            the stat-cache is correct. The old way of creating a freshly
 
56
            added tree always had a hot cache.
 
57
        """
 
58
        from bzrlib.benchmarks.tree_creator.kernel_like import (
 
59
            KernelLikeAddedTreeCreator,
 
60
            )
 
61
        creator = KernelLikeAddedTreeCreator(self, link_working=link_working,
 
62
                                             hot_cache=hot_cache)
 
63
        return creator.create(root=root)
 
64
 
 
65
    def make_kernel_like_committed_tree(self, root='.',
 
66
                                    link_working=True,
 
67
                                    link_bzr=False,
 
68
                                    hot_cache=True):
 
69
        """Make a kernel like tree, with all files added and committed
 
70
 
 
71
        :param root: Where to create the files
 
72
        :param link_working: Instead of copying all of the working tree
 
73
            files, just hardlink them to the cached files. Tests can unlink
 
74
            files that they will change.
 
75
        :param link_bzr: Hardlink the .bzr directory. For readonly 
 
76
            operations this is safe, and shaves off a lot of setup time
 
77
        """
 
78
        from bzrlib.benchmarks.tree_creator.kernel_like import (
 
79
            KernelLikeCommittedTreeCreator,
 
80
            )
 
81
        creator = KernelLikeCommittedTreeCreator(self,
 
82
                                                 link_working=link_working,
 
83
                                                 link_bzr=link_bzr,
 
84
                                                 hot_cache=hot_cache)
 
85
        return creator.create(root=root)
 
86
 
 
87
    def make_kernel_like_inventory(self):
 
88
        """Create an inventory with the properties of a kernel-like tree
 
89
 
 
90
        This should be equivalent to a committed kernel like tree, not
 
91
        just a working tree.
 
92
        """
 
93
        from bzrlib.benchmarks.tree_creator.kernel_like import (
 
94
            KernelLikeInventoryCreator,
 
95
            )
 
96
        creator = KernelLikeInventoryCreator(self)
 
97
        return creator.create()
 
98
 
 
99
    def make_many_commit_tree(self, directory_name='.',
 
100
                              hardlink=False):
 
101
        """Create a tree with many commits.
 
102
        
 
103
        No file changes are included. Not hardlinking the working tree, 
 
104
        because there are no working tree files.
 
105
        """
 
106
        from bzrlib.benchmarks.tree_creator.simple_many_commit import (
 
107
            SimpleManyCommitTreeCreator,
 
108
            )
 
109
        creator = SimpleManyCommitTreeCreator(self, link_bzr=hardlink)
 
110
        return creator.create(root=directory_name)
 
111
 
 
112
    def make_heavily_merged_tree(self, directory_name='.',
 
113
                                 hardlink=False):
 
114
        """Create a tree in which almost every commit is a merge.
 
115
       
 
116
        No file changes are included.  This produces two trees, 
 
117
        one of which is returned.  Except for the first commit, every
 
118
        commit in its revision-history is a merge another commit in the other
 
119
        tree.  Not hardlinking the working tree, because there are no working 
 
120
        tree files.
 
121
        """
 
122
        from bzrlib.benchmarks.tree_creator.heavily_merged import (
 
123
            HeavilyMergedTreeCreator,
 
124
            )
 
125
        creator = HeavilyMergedTreeCreator(self, link_bzr=hardlink)
 
126
        return creator.create(root=directory_name)
 
127
 
 
128
 
 
129
def test_suite():
 
130
    """Build and return a TestSuite which contains benchmark tests only."""
 
131
    testmod_names = [ \
 
132
                   'bzrlib.benchmarks.bench_add',
 
133
                   'bzrlib.benchmarks.bench_bench',
 
134
                   'bzrlib.benchmarks.bench_cache_utf8',
 
135
                   'bzrlib.benchmarks.bench_checkout',
 
136
                   'bzrlib.benchmarks.bench_commit',
 
137
                   'bzrlib.benchmarks.bench_inventory',
 
138
                   'bzrlib.benchmarks.bench_log',
 
139
                   'bzrlib.benchmarks.bench_osutils',
 
140
                   'bzrlib.benchmarks.bench_rocks',
 
141
                   'bzrlib.benchmarks.bench_status',
 
142
                   'bzrlib.benchmarks.bench_transform',
 
143
                   'bzrlib.benchmarks.bench_workingtree',
 
144
                   'bzrlib.benchmarks.bench_xml',
 
145
                   ]
 
146
    suite = TestLoader().loadTestsFromModuleNames(testmod_names) 
 
147
 
 
148
    # Load any benchmarks from plugins
 
149
    for name, module in plugin.all_plugins().items():
 
150
        if getattr(module, 'bench_suite', None) is not None:
 
151
            suite.addTest(module.bench_suite())
 
152
 
 
153
    return suite