/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

  • Committer: Carl Friedrich Bolz
  • Date: 2006-08-19 02:17:16 UTC
  • mto: (1908.3.21 usecases-benchmarks)
  • mto: This revision was merged to the branch mainline in revision 2068.
  • Revision ID: cfbolz@gmx.de-20060819021716-4a2cdd443e83721b
Refactor the bundle benchmarks to use the existing helper functions.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/benchmarks/__init__.py
127
127
        creator = HeavilyMergedTreeCreator(self, link_bzr=hardlink)
128
128
        return creator.create(root=directory_name)
129
129
 
130
 
    def create_with_commits(self, num_files, num_commits, directory_name='.'):
 
130
    def create_with_commits(self, num_files, num_commits, directory_name='.',
 
131
                            hardlink=False):
131
132
        """Create a tree with many files and many commits. Every commit changes
132
133
        exactly one file.
133
134
        
134
135
        :param num_files: number of files to be created
135
136
        :param num_commits: number of commits in the newly created tree
136
137
        """
137
 
        files = ["%s/%s" % (directory_name, i) for i in range(num_files)]
138
 
        for fn in files:
139
 
            f = open(fn, "wb")
140
 
            try:
141
 
                f.write("some content\n")
142
 
            finally:
143
 
                f.close()
144
 
        tree = bzrdir.BzrDir.create_standalone_workingtree(directory_name)
145
 
        tree.add([str(i) for i in range(num_files)])
146
 
        tree.lock_write()
147
 
        try:
148
 
            tree.commit('initial commit')
149
 
            for i in range(num_commits):
150
 
                fn = files[i % len(files)]
151
 
                content = range(i) + [i, i, i, ""]
152
 
                f = open(fn, "wb")
153
 
                try:
154
 
                    f.write("\n".join([str(i) for i in content]))
155
 
                finally:
156
 
                    f.close()
157
 
                tree.commit("changing file %s" % fn)
158
 
        finally:
159
 
            tree.unlock()
 
138
        from bzrlib.benchmarks.tree_creator.many_commit import (
 
139
            ManyCommitTreeCreator,
 
140
            )
 
141
        creator = ManyCommitTreeCreator(self, link_bzr=hardlink,
 
142
                                        num_files=num_files,
 
143
                                        num_commits=num_commits)
 
144
        tree = creator.create(root=directory_name)
 
145
        files = ["%s/%s" % (directory_name, fn) for fn in creator.files]
160
146
        return tree, files
161
147
 
162
148
    def commit_some_revisions(self, tree, files, num_commits,