← 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: Andrew Bennetts
  • Date: 2010-10-08 04:25:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5472.
  • Revision ID: andrew.bennetts@canonical.com-20101008042510-sg9vdhmnggilzxsk
Fix stray TAB in source.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_lsprof.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2007, 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
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):
92
93
        self.stats.save(f)
93
94
        data1 = cPickle.load(open(f))
94
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)