← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_lsprof.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-07-07 21:30:06 UTC
  • mfrom: (5333.1.2 better_pyqt_include)
  • Revision ID: pqm@pqm.ubuntu.com-20100707213006-lriphkkbzwwrl7ne
(jameinel) Use a better list of PyQt includes and excludes. (Gary van der
 Merwe)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_lsprof.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Tests for profiling data collection."""
18
18
 
19
19
 
20
20
import cPickle
21
21
import os
 
22
import threading
22
23
 
23
24
import bzrlib
24
 
from bzrlib import tests
 
25
from bzrlib import errors, tests
25
26
 
26
27
 
27
28
class _LSProfFeature(tests.Feature):
64
65
 
65
66
    def _tempfile(self, ext):
66
67
        dir = self.test_dir
67
 
        return os.path.join(dir, "tmp_profile_data." + ext)
 
68
        return bzrlib.osutils.pathjoin(dir, "tmp_profile_data." + ext)
68
69
 
69
70
    def test_stats_save_to_txt(self):
70
71
        f = self._tempfile("txt")
77
78
        self.stats.save(f)
78
79
        lines = open(f).readlines()
79
80
        self.assertEqual(lines[0], "events: Ticks\n")
 
81
        f = bzrlib.osutils.pathjoin(self.test_dir, "callgrind.out.foo")
 
82
        self.stats.save(f)
 
83
        lines = open(f).readlines()
 
84
        self.assertEqual(lines[0], "events: Ticks\n")
80
85
        # Test explicit format nommination
81
86
        f2 = self._tempfile("txt")
82
87
        self.stats.save(f2, format="callgrind")
88
93
        self.stats.save(f)
89
94
        data1 = cPickle.load(open(f))
90
95
        self.assertEqual(type(data1), bzrlib.lsprof.Stats)
 
96
 
 
97
 
 
98
class TestBzrProfiler(tests.TestCase):
 
99
 
 
100
    _test_needs_features = [LSProfFeature]
 
101
 
 
102
    def test_start_call_stuff_stop(self):
 
103
        profiler = bzrlib.lsprof.BzrProfiler()
 
104
        profiler.start()
 
105
        try:
 
106
            def a_function():
 
107
                pass
 
108
            a_function()
 
109
        finally:
 
110
            stats = profiler.stop()
 
111
        stats.freeze()
 
112
        lines = [str(data) for data in stats.data]
 
113
        lines = [line for line in lines if 'a_function' in line]
 
114
        self.assertLength(1, lines)
 
115
 
 
116
    def test_block_0(self):
 
117
        # When profiler_block is 0, reentrant profile requests fail.
 
118
        self.overrideAttr(bzrlib.lsprof.BzrProfiler, 'profiler_block', 0)
 
119
        inner_calls = []
 
120
        def inner():
 
121
            profiler = bzrlib.lsprof.BzrProfiler()
 
122
            self.assertRaises(errors.BzrError, profiler.start)
 
123
            inner_calls.append(True)
 
124
        bzrlib.lsprof.profile(inner)
 
125
        self.assertLength(1, inner_calls)
 
126
 
 
127
    def test_block_1(self):
 
128
        # When profiler_block is 1, concurrent profiles serialise.
 
129
        # This is tested by manually acquiring the profiler lock, then
 
130
        # starting a thread that tries to profile, and releasing the lock. 
 
131
        # We know due to test_block_0 that two profiles at once hit the lock,
 
132
        # so while this isn't perfect (we'd want a callback on the lock being
 
133
        # entered to allow lockstep evaluation of the actions), its good enough
 
134
        # to be confident regressions would be caught. Alternatively, if this
 
135
        # is flakey, a fake Lock object can be used to trace the calls made.
 
136
        calls = []
 
137
        def profiled():
 
138
            calls.append('profiled')
 
139
        def do_profile():
 
140
            bzrlib.lsprof.profile(profiled)
 
141
            calls.append('after_profiled')
 
142
        thread = threading.Thread(target=do_profile)
 
143
        bzrlib.lsprof.BzrProfiler.profiler_lock.acquire()
 
144
        try:
 
145
            try:
 
146
                thread.start()
 
147
            finally:
 
148
                bzrlib.lsprof.BzrProfiler.profiler_lock.release()
 
149
        finally:
 
150
            thread.join()
 
151
        self.assertLength(2, calls)