summaryrefslogtreecommitdiff
path: root/tests/test_profilemiddleware.py
blob: 944243db151d8f11be8b902f4e73a67e2fc8ae62 (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
from paste.fixture import *
from paste.debug.profile import *

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'