uliweb_test_client.py

import os
from uliweb import manage, models
from uliweb.manage import make_simple_application

os.chdir('demo')

manage.call('uliweb syncdb -v')
manage.call('uliweb dbinit -v')

def test_apijson_get():
    """
    >>> app = make_simple_application(project_dir='.')
    >>> from uliweb import is_in_web
    >>> from uliweb.utils.test import client
    >>> c = client('.')
    >>> data ='''{
    ... "user":{
    ...    "@role":"OWNER"
    ... }
    ... }'''
    >>> r = c.post('/apijson/get',data=data)
    >>> print(r.data)
    {}
    """

uliweb_test_handler.py

    User = get_model('user')
    
    def pre_call(request):
        user = User.get(int(user_id))
        request.user = user
    
    settings.set_var('GENERIC/X_SENDFILE', False)
    handler = application.handler()
    begin = time.time()
    r = handler.get(str(url), pre_call=pre_call, middlewares=[])
    #disable direct_passthrough change for werkzeug 0.9.1
    r.direct_passthrough = False
    
    fname = functions.filename_convert(filename, UUIDFilenameConverter)
    real_fname = os.path.join(settings.get_var('GENERIC/DOWNLOAD_DIR'), fname)
    path = os.path.dirname(real_fname)
    if not os.path.exists(path):
        os.makedirs(path)
        
    with open(real_fname, 'wb') as f:
        f.write(r.data)