← 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: John Arbash Meinel
  • Date: 2011-05-11 11:35:28 UTC
  • mto: This revision was merged to the branch mainline in revision 5851.
  • Revision ID: john@arbash-meinel.com-20110511113528-qepibuwxicjrbb2h
Break compatibility with python <2.6.

This includes auditing the code for places where we were doing
explicit 'sys.version' checks and removing them as appropriate.

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, 2011 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
18
18
 
19
19
 
20
20
import cPickle
21
 
import os
 
21
import threading
22
22
 
23
23
import bzrlib
24
 
from bzrlib import tests
 
24
from bzrlib import errors, tests
25
25
 
26
26
 
27
27
class _LSProfFeature(tests.Feature):
111
111
        lines = [str(data) for data in stats.data]
112
112
        lines = [line for line in lines if 'a_function' in line]
113
113
        self.assertLength(1, lines)
 
114
 
 
115
    def test_block_0(self):
 
116
        # When profiler_block is 0, reentrant profile requests fail.
 
117
        self.overrideAttr(bzrlib.lsprof.BzrProfiler, 'profiler_block', 0)
 
118
        inner_calls = []
 
119
        def inner():
 
120
            profiler = bzrlib.lsprof.BzrProfiler()
 
121
            self.assertRaises(errors.BzrError, profiler.start)
 
122
            inner_calls.append(True)
 
123
        bzrlib.lsprof.profile(inner)
 
124
        self.assertLength(1, inner_calls)
 
125
 
 
126
    def test_block_1(self):
 
127
        # When profiler_block is 1, concurrent profiles serialise.
 
128
        # This is tested by manually acquiring the profiler lock, then
 
129
        # starting a thread that tries to profile, and releasing the lock. 
 
130
        # We know due to test_block_0 that two profiles at once hit the lock,
 
131
        # so while this isn't perfect (we'd want a callback on the lock being
 
132
        # entered to allow lockstep evaluation of the actions), its good enough
 
133
        # to be confident regressions would be caught. Alternatively, if this
 
134
        # is flakey, a fake Lock object can be used to trace the calls made.
 
135
        calls = []
 
136
        def profiled():
 
137
            calls.append('profiled')
 
138
        def do_profile():
 
139
            bzrlib.lsprof.profile(profiled)
 
140
            calls.append('after_profiled')
 
141
        thread = threading.Thread(target=do_profile)
 
142
        bzrlib.lsprof.BzrProfiler.profiler_lock.acquire()
 
143
        try:
 
144
            try:
 
145
                thread.start()
 
146
            finally:
 
147
                bzrlib.lsprof.BzrProfiler.profiler_lock.release()
 
148
        finally:
 
149
            thread.join()
 
150
        self.assertLength(2, calls)