def test_shared_data_middleware(): """Shared data middleware""" def null_application(environ, start_response): start_response('404 NOT FOUND', [('Content-Type', 'text/plain')]) yield 'NOT FOUND' app = SharedDataMiddleware( null_application, { '/': path.join(path.dirname(__file__), 'res'), '/sources': path.join(path.dirname(__file__), 'res'), '/pkg': ('werkzeug.debug', 'shared') }) for p in '/test.txt', '/sources/test.txt': app_iter, status, headers = run_wsgi_app(app, create_environ(p)) assert status == '200 OK' assert ''.join(app_iter).strip() == 'FOUND' app_iter, status, headers = run_wsgi_app(app, create_environ('/pkg/body.tmpl')) contents = ''.join(app_iter) assert 'Werkzeug Debugger' in contents app_iter, status, headers = run_wsgi_app(app, create_environ('/missing')) assert status == '404 NOT FOUND' assert ''.join(app_iter).strip() == 'NOT FOUND'
def test_get_host(): """Host lookup""" env = {'HTTP_X_FORWARDED_HOST': 'example.org', 'SERVER_NAME': 'bullshit', 'HOST_NAME': 'ignore me dammit'} assert get_host(env) == 'example.org' assert get_host(create_environ('/', 'http://example.org')) \ == 'example.org'
def initdb(): # A request_context is required to use these helper functions with new_app(app).request_context(create_environ()): db.drop_all() db.create_all() readaction = schema.Action("read") insertaction = schema.Action("insert") deleteaction = schema.Action("delete") editaction = schema.Action("edit") db.session.add(readaction) db.session.add(insertaction) db.session.add(deleteaction) db.session.add(editaction) administrator = schema.User(username="******", password="******", firstname="admin", lastname="admin", email="*****@*****.**") administrator.actions.append(readaction) administrator.actions.append(insertaction) administrator.actions.append(deleteaction) administrator.actions.append(editaction) db.session.add(administrator) db.session.commit()
def test_path_info_extraction(): """PATH INFO extraction feature""" x = extract_path_info('http://example.com/app', '/app/hello') assert x == u'/hello' x = extract_path_info('http://example.com/app', 'https://example.com/app/hello') assert x == u'/hello' x = extract_path_info('http://example.com/app/', 'https://example.com/app/hello') assert x == u'/hello' x = extract_path_info('http://example.com/app/', 'https://example.com/app') assert x == u'/' x = extract_path_info(u'http://☃.net/', u'/fööbär') assert x == u'/fööbär' x = extract_path_info(u'http://☃.net/x', u'http://☃.net/x/fööbär') assert x == u'/fööbär' env = create_environ(u'/fööbär', u'http://☃.net/x/') x = extract_path_info(env, u'http://☃.net/x/fööbär') assert x == u'/fööbär' x = extract_path_info('http://example.com/app/', 'https://example.com/a/hello') assert x is None x = extract_path_info('http://example.com/app/', 'https://example.com/app/hello', collapse_http_schemes=False) assert x is None
def test_request_context(self, *args, **kwargs): """Creates a WSGI environment from the given values (see :func:`werkzeug.create_environ` for more information, this function accepts the same arguments). """ from werkzeug import create_environ return self.request_context(create_environ(*args, **kwargs))
def __init__(self, path="/", query_string=None, method='GET', content_type=None, content_length=0, form_data=None, environ_overrides=None): """ For parameter reference see the documentation of the werkzeug package, especially the functions `url_encode` and `create_environ`. """ input_stream = None if form_data is not None: form_data = url_encode(form_data) content_type = 'application/x-www-form-urlencoded' content_length = len(form_data) input_stream = StringIO(form_data) environ = create_environ(path=path, query_string=query_string, method=method, input_stream=input_stream, content_type=content_type, content_length=content_length) environ['HTTP_USER_AGENT'] = 'MoinMoin/TestRequest' # must have reverse lookup or tests will be extremely slow: environ['REMOTE_ADDR'] = '127.0.0.1' if environ_overrides: environ.update(environ_overrides) super(TestRequest, self).__init__(environ)
def test_lighttpd_cgi_root_fix(): """Test the LighttpdCGIRootFix fixer""" app = fixers.LighttpdCGIRootFix(path_check_app) response = Response.from_app(app, dict(create_environ(), SCRIPT_NAME='/foo', PATH_INFO='/bar' )) assert response.data == 'PATH_INFO: /foo/bar\nSCRIPT_NAME: '
def test_path_info_from_request_uri_fix(): """Test the PathInfoFromRequestUriFix fixer""" app = fixers.PathInfoFromRequestUriFix(path_check_app) for key in 'REQUEST_URI', 'REQUEST_URL', 'UNENCODED_URL': env = dict(create_environ(), SCRIPT_NAME='/test', PATH_INFO='/?????') env[key] = '/test/foo%25bar?drop=this' response = Response.from_app(app, env) assert response.data == 'PATH_INFO: /foo%bar\nSCRIPT_NAME: /test'
def test_broken_multipart(): """Broken multipart does not break the applicaiton""" data = ( '--foo\r\n' 'Content-Disposition: form-data; name="test"; filename="test.txt"\r\n' 'Content-Transfer-Encoding: base64\r\n' 'Content-Type: text/plain\r\n\r\n' 'broken base 64' '--foo--' ) _, form, files = parse_form_data(create_environ(data=data, method='POST', content_type='multipart/form-data; boundary=foo')) assert not files assert not form assert_raises(ValueError, parse_form_data, create_environ(data=data, method='POST', content_type='multipart/form-data; boundary=foo'), silent=False)
def test_parse_form_data_get_without_content(): """GET requests without data, content type and length returns no data""" env = create_environ('/foo', 'http://example.org/', method='GET') del env['CONTENT_TYPE'] del env['CONTENT_LENGTH'] stream, form, files = parse_form_data(env) assert stream.read() == "" assert len(form) == 0 assert len(files) == 0
def test_get_host(): """Host lookup""" env = { 'HTTP_X_FORWARDED_HOST': 'example.org', 'SERVER_NAME': 'bullshit', 'HOST_NAME': 'ignore me dammit' } assert get_host(env) == 'example.org' assert get_host(create_environ('/', 'http://example.org')) \ == 'example.org'
def __init__(self, url=None, pagename=''): if url is None: url = 'http://localhost:0/' # just some somehow valid dummy URL environ = create_environ(base_url=url) # XXX not sure about base_url, but makes "make underlay" work environ['HTTP_USER_AGENT'] = 'CLI/Script' environ['wsgi.input'] = sys.stdin request = Request(environ) super(ScriptContext, self).__init__(request) from MoinMoin import wsgiapp wsgiapp.init(self)
def test_wrapper_internals(): """Test internals of the wrappers""" from werkzeug import Request req = Request.from_values(data={'foo': 'bar'}, method='POST') req._load_form_data() assert req.form.to_dict() == {'foo': 'bar'} # second call does not break req._load_form_data() assert req.form.to_dict() == {'foo': 'bar'} # check reprs assert repr(req) == "<Request 'http://localhost/' [POST]>" resp = Response() assert repr(resp) == '<Response 0 bytes [200 OK]>' resp.data = 'Hello World!' assert repr(resp) == '<Response 12 bytes [200 OK]>' resp.response = iter(['Test']) assert repr(resp) == '<Response streamed [200 OK]>' # unicode data does not set content length response = Response([u'Hällo Wörld']) headers = response.get_wsgi_headers(create_environ()) assert 'Content-Length' not in headers response = Response(['Hällo Wörld']) headers = response.get_wsgi_headers(create_environ()) assert 'Content-Length' in headers # check for internal warnings print 'start' filterwarnings('error', category=Warning) response = Response() environ = create_environ() response.response = 'What the...?' assert_raises(Warning, lambda: list(response.iter_encoded())) assert_raises(Warning, lambda: list(response.get_app_iter(environ))) response.direct_passthrough = True assert_raises(Warning, lambda: list(response.iter_encoded())) assert_raises(Warning, lambda: list(response.get_app_iter(environ))) resetwarnings()
def test_header_rewriter_fix(): """Test the HeaderRewriterFix fixer""" @Request.application def application(request): return Response("", headers=[ ('X-Foo', 'bar') ]) application = fixers.HeaderRewriterFix(application, ('X-Foo',), (('X-Bar', '42'),)) response = Response.from_app(application, create_environ()) assert response.headers['Content-Type'] == 'text/plain; charset=utf-8' assert 'X-Foo' not in response.headers assert response.headers['X-Bar'] == '42'
def test_header_rewriter_fix(): """Test the HeaderRewriterFix fixer""" @Request.application def application(request): return Response("", headers=[('X-Foo', 'bar')]) application = fixers.HeaderRewriterFix(application, ('X-Foo', ), (('X-Bar', '42'), )) response = Response.from_app(application, create_environ()) assert response.headers['Content-Type'] == 'text/plain; charset=utf-8' assert 'X-Foo' not in response.headers assert response.headers['X-Bar'] == '42'
def _create_environ(self, url, method, data, refer, content_type=None): """Return an environ to request *url*, including cookies.""" environ_args = dict(self._wsgi_server, method=method) base_url = self._referrer if refer else self._base_url environ_args.update(self._canonicalize_url(url, base_url)) environ_args.update(self._prep_input(method, data, content_type)) environ = create_environ(**environ_args) if refer and self._referrer: environ['HTTP_REFERER'] = self._referrer environ.setdefault('REMOTE_ADDR', '127.0.0.1') self._cookie_jar.export_to_environ(environ) return environ
def request_context(self, *args, **kw): """Create a request context for use in tests. The arguments passed will be used to create a WSGI environment to create a request instance (see :func:`werkzeug.create_environ` for more information). This method must be used with the ``with`` statement. For example:: with self.request_context(): do_something_with(request) """ from werkzeug import create_environ return self.test_app.request_context(create_environ(*args, **kw))
def test_shared_data_middleware(): """Shared data middleware""" def null_application(environ, start_response): start_response('404 NOT FOUND', [('Content-Type', 'text/plain')]) yield 'NOT FOUND' app = SharedDataMiddleware(null_application, { '/': path.join(path.dirname(__file__), 'res'), '/sources': path.join(path.dirname(__file__), 'res'), '/pkg': ('werkzeug.debug', 'shared') }) for p in '/test.txt', '/sources/test.txt': app_iter, status, headers = run_wsgi_app(app, create_environ(p)) assert status == '200 OK' assert ''.join(app_iter).strip() == 'FOUND' app_iter, status, headers = run_wsgi_app(app, create_environ('/pkg/body.tmpl')) contents = ''.join(app_iter) assert 'Werkzeug Debugger' in contents app_iter, status, headers = run_wsgi_app(app, create_environ('/missing')) assert status == '404 NOT FOUND' assert ''.join(app_iter).strip() == 'NOT FOUND'
def test_broken_multipart(): """Broken multipart does not break the applicaiton""" data = ( '--foo\r\n' 'Content-Disposition: form-data; name="test"; filename="test.txt"\r\n' 'Content-Transfer-Encoding: base64\r\n' 'Content-Type: text/plain\r\n\r\n' 'broken base 64' '--foo--') _, form, files = parse_form_data( create_environ(data=data, method='POST', content_type='multipart/form-data; boundary=foo')) assert not files assert not form assert_raises(ValueError, parse_form_data, create_environ( data=data, method='POST', content_type='multipart/form-data; boundary=foo'), silent=False)
def fetch_body(app, path): environ = create_environ(path=path) def start_response(status, headers): start_response.code = int(status.split()[0]) start_response.code = None content = ''.join(list(app(environ, start_response))) if start_response.code == 200: return content elif start_response.code // 100 == 3: abort(404) else: abort(start_response.code)
def open(self, path='/', base_url=None, query_string=None, method='GET', data=None, input_stream=None, content_type=None, content_length=0, errors_stream=None, multithread=False, multiprocess=False, run_once=False, environ_overrides=None, buffered=True): parsed = urlparse(path) if parsed.scheme: if base_url is None: base_url = parsed.scheme + '://' + parsed.netloc if query_string is None: query_string = parsed.query path = parsed.path if (input_stream is None and data is not None and method in ('PUT', 'POST')): input_stream, content_length, content_type = \ self._prep_input(input_stream, data, content_type) if base_url is None: base_url = self.base_url or self.state.base_url environ = create_environ(path, base_url, query_string, method, input_stream, content_type, content_length, errors_stream, multithread, multiprocess, run_once) current_state = self.state current_state.prepare_environ(environ) if environ_overrides: environ.update(environ_overrides) logger.info("%s %s" % (method, request_uri(environ))) rv = run_wsgi_app(self.application, environ, buffered=buffered) response = _APIClientResponse(*rv) response.state = new_state = current_state.copy() new_state.process_response(response, environ) return response
def test_parse_form_data_put_without_content(): """A PUT without a Content-Type header returns empty data Both rfc1945 and rfc2616 (1.0 and 1.1) say "Any HTTP/[1.0/1.1] message containing an entity-body SHOULD include a Content-Type header field defining the media type of that body." In the case where either headers are omitted, parse_form_data should still work. """ env = create_environ('/foo', 'http://example.org/', method='PUT') del env['CONTENT_TYPE'] del env['CONTENT_LENGTH'] stream, form, files = parse_form_data(env) assert stream.read() == "" assert len(form) == 0 assert len(files) == 0
def test_proxy_fix(): """Test the ProxyFix fixer""" @fixers.ProxyFix @Request.application def app(request): return Response('%s|%s' % ( request.remote_addr, # do not use request.host as this fixes too :) request.environ['HTTP_HOST'] )) response = Response.from_app(app, dict(create_environ(), HTTP_X_FORWARDED_HOST='example.com', HTTP_X_FORWARDED_FOR='1.2.3.4, 5.6.7.8', REMOTE_ADDR='127.0.0.1', HTTP_HOST='fake' )) assert response.data == '1.2.3.4|example.com'
def test_proxy_fix(): """Test the ProxyFix fixer""" @fixers.ProxyFix @Request.application def app(request): return Response('%s|%s' % ( request.remote_addr, # do not use request.host as this fixes too :) request.environ['HTTP_HOST'])) response = Response.from_app( app, dict(create_environ(), HTTP_X_FORWARDED_HOST='example.com', HTTP_X_FORWARDED_FOR='1.2.3.4, 5.6.7.8', REMOTE_ADDR='127.0.0.1', HTTP_HOST='fake')) assert response.data == '1.2.3.4|example.com'
def test_request_context(self, *args, **kwargs): """Creates a WSGI environment from the given values (see :func:`werkzeug.create_environ` for more information, this function accepts the same arguments). """ from werkzeug import create_environ environ_overrides = kwargs.setdefault('environ_overrides', {}) if self.config.get('SERVER_NAME'): server_name = self.config.get('SERVER_NAME') if ':' not in server_name: http_host, http_port = server_name, '80' else: http_host, http_port = server_name.split(':', 1) environ_overrides.setdefault('SERVER_NAME', server_name) environ_overrides.setdefault('HTTP_HOST', server_name) environ_overrides.setdefault('SERVER_PORT', http_port) return self.request_context(create_environ(*args, **kwargs))
def run_app(app, path='/'): env = create_environ(path, SECRET_KEY) return run_wsgi_app(app, env)
def initdb(): # A request_context is required to use these helper functions with new_app(app).request_context(create_environ()): db.drop_all() db.create_all()
__author__ = 'Robin Wittler' __contact__ = '*****@*****.**' __version__ = '0.0.1' __license__ = 'GPL3' import re import sys import blog import models import getpass import werkzeug import argparse import textwrap env = werkzeug.create_environ('/blog', 'http://localhost:5000/') ctx = blog.app.request_context(env) tag_splitter = re.compile('\s*,\s*') # now we have a valid request context and are able to use flask-sqlalchemy class CreateDB(argparse.Action): def __call__(self, parser, namespace, values, option_string=None): ctx.push() models.create_all() ctx.pop() parser.exit(0) class AddUser(argparse.Action): def __call__(self, parser, namespace, values, option_string=None): ctx.push()
def run_app(app, path='/'): config = config_generate() env = create_environ(path, config["secret_key"]) return run_wsgi_app(app, env)
def test_lighttpd_cgi_root_fix(): """Test the LighttpdCGIRootFix fixer""" app = fixers.LighttpdCGIRootFix(path_check_app) response = Response.from_app( app, dict(create_environ(), SCRIPT_NAME='/foo', PATH_INFO='/bar')) assert response.data == 'PATH_INFO: /foo/bar\nSCRIPT_NAME: '
def test_request_context(self, *args, **kwargs): return self.request_context(create_environ(*args, **kwargs))
def build_app(cls, import_name, modified_only=False, import_stash=False, logfile=None): """Create a Tango application object from a Python import name. This function accepts three kinds of import names: 1. a single-module name where the module is itself a stash 2. a package name which has a submodule or sub-package named 'stash'. 3. a dotted module name referring to a module inside a package's stash. In case #1, the module is a self-contained application in one .py file. In case #2, the package is arbitrarily laid out, but the stash module inside it is one or more modules conforming to Tango's stash conventions. In case #3, the module is inside a package matching case #2, but the import name refers to a module which by itself would otherwise match case #1. Case #3 is treated like case #1 with one important exception. Since the module is inside a package, that package is used as the application object's import name for the purposes of loading configuration directives, as stash modules are allowed to and encouraged to use their projects config.py. This is essential to shelving modules in isolation when working with a stash package with more than one stash module. Example, using a single module called simplest.py (case #1): >>> app = Tango.build_app('simplest') >>> sorted(app.url_map.iter_rules(), key=lambda rule: rule.rule) ... # doctest:+NORMALIZE_WHITESPACE [<Rule '/' (HEAD, OPTIONS, GET) -> />, <Rule '/static/<filename>' (HEAD, OPTIONS, GET) -> static>] >>> Example, using the simplesite module in this project (case #2): >>> app = Tango.build_app('simplesite') >>> sorted(app.url_map.iter_rules(), key=lambda rule: rule.rule) ... # doctest:+NORMALIZE_WHITESPACE [<Rule '/' (HEAD, OPTIONS, GET) -> />, <Rule '/static/<filename>' (HEAD, OPTIONS, GET) -> static>] >>> Example, using submodule of stash in a package with config (case #3): >>> app = Tango.build_app('simplesite.stash.index') >>> sorted(app.url_map.iter_rules(), key=lambda rule: rule.rule) ... # doctest:+NORMALIZE_WHITESPACE [<Rule '/' (HEAD, OPTIONS, GET) -> />, <Rule '/static/<filename>' (HEAD, OPTIONS, GET) -> static>] >>> Example, using submodule in the stash in a package without config (case #3 but identical to case #1): >>> app = Tango.build_app('testsite.stash.package.module') >>> sorted(app.url_map.iter_rules(), key=lambda rule: rule.rule) ... # doctest:+NORMALIZE_WHITESPACE [<Rule '/index.json' (HEAD, OPTIONS, GET) -> /index.json>, <Rule '/static/<filename>' (HEAD, OPTIONS, GET) -> static>] >>> For convenience in development, particularly with shell tab completion, a .py module name is also accepted: >>> Tango.build_app('simplest.py') # doctest:+ELLIPSIS <tango.app.Tango object at 0x...> >>> Avoids import side effects: >>> Tango.build_app('importerror.py') # doctest:+ELLIPSIS <tango.app.Tango object at 0x...> >>> Tango.build_app('importerror') # doctest:+ELLIPSIS <tango.app.Tango object at 0x...> >>> """ import_name = fix_import_name_if_pyfile(import_name) # Initialize application. See docstring above for construction logic. app = None package_name, module_name = package_submodule(import_name) if package_name and module_name: # import_name points to submodule, look for package config. root_package_name = namespace_segments(import_name)[0] if module_exists(root_package_name + '.config'): app = cls(root_package_name) app.config.from_object(root_package_name + '.config') if app is None: app = cls(import_name) # Check for a site config. if module_exists(import_name + '.config'): app.config.from_object(import_name + '.config') # Push app context onto request stack for use in initialization. # Use `with` or try-finally, ensure context is popped in case of error. with app.request_context(create_environ()): # Load Tango filters. Only needed for packages; single modules do # not have implicit templates. if module_is_package(import_name): tango.filters.init_app(app) build_options = {'import_stash': import_stash} build_options['logfile'] = logfile build_options['modified_only'] = modified_only if module_exists(import_name + '.stash'): module = __import__(import_name, fromlist=['stash']).stash app.routes = build_module_routes(module, **build_options) else: app.routes = build_module_routes(import_name, **build_options) # Stitch together context, template, and path. for route in app.routes: app.build_view(route) @app.context_processor def process_view_args(): """Put view args into template context for tango template writers. The Flask Request object has a dictionary of the request handler function's arguments, but this is None when there are no view arguments. A Flask context processor must always return a dictionary. http://flask.pocoo.org/docs/api/#flask.Request.view_args """ if request.view_args is None: return {} return request.view_args return app
def test_request_context(self, *args, **kwargs): """从给定的值创建一个WSGI环境(更多信息请参见werkzeug.create_environ, 这个函数接受相同的参数)。 """ return self.request_context(create_environ(*args, **kwargs))