summaryrefslogtreecommitdiff
path: root/tests/test_profilemiddleware.py
blob: 4c189f83887fc97bec16a5bf0b92b8716a77f61b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from paste.fixture import *
try:
    from paste.debug.profile import *
    disable = False
except ImportError:
    disable = True

if not disable:
    def simple_app(environ, start_response):
        start_response('200 OK', [('content-type', 'text/html')])
        return ['all ok']

    def long_func():
        for i in range(1000):
            pass
        return 'test'

    def test_profile():
        app = TestApp(ProfileMiddleware(simple_app, {}))
        res = app.get('/')
        # The original app:
        res.mustcontain('all ok')
        # The profile information:
        res.mustcontain('<pre')

    def test_decorator():
        value = profile_decorator()(long_func)()
        assert value == 'test'