Ejemplo n.º 1
0
    def open(self, *args, **kwargs):
        if self.context_preserved:
            _request_ctx_stack.pop()
            self.context_preserved = False
        kwargs.setdefault('environ_overrides', {}) \
            ['flask._preserve_context'] = self.preserve_context

        as_tuple = kwargs.pop('as_tuple', False)
        buffered = kwargs.pop('buffered', False)
        follow_redirects = kwargs.pop('follow_redirects', False)

        builder = EnvironBuilder(*args, **kwargs)

        if self.application.config.get('SERVER_NAME'):
            server_name = self.application.config.get('SERVER_NAME')
            if ':' not in server_name:
                http_host, http_port = server_name, None
            else:
                http_host, http_port = server_name.split(':', 1)
            if builder.base_url == 'http://localhost/':
                # Default Generated Base URL
                if http_port != None:
                    builder.host = http_host + ':' + http_port
                else:
                    builder.host = http_host
        old = _request_ctx_stack.top
        try:
            return Client.open(self,
                               builder,
                               as_tuple=as_tuple,
                               buffered=buffered,
                               follow_redirects=follow_redirects)
        finally:
            self.context_preserved = _request_ctx_stack.top is not old
Ejemplo n.º 2
0
 def run(self, args):
     from IPython.Shell import IPShellEmbed
     from werkzeug import Client, EnvironBuilder
     from dextrose.context import Context
     from dextrose.http import Response
     app = load_application(args.package, args.environment)
     environ_builder = EnvironBuilder()
     environ = environ_builder.get_environ()
     request = environ_builder.get_request()
     client = Client(app, Response)
     with Context(app, environ, request, {}) as context:
         banner = "Dextrose IPython shell\n%s" % args.package
         shell = IPShellEmbed(
             banner=banner,
             argv=[
                 '-prompt_in1',
                 '%s> ' % args.package, '-prompt_in2',
                 '%s... ' % (' ' * (len(args.package) - 3)), '-prompt_out',
                 '=> '
             ])
         shell(global_ns={},
               local_ns={
                   'app': app,
                   'environ': environ,
                   'client': client,
                   'context': context
               })
Ejemplo n.º 3
0
 def test_upload(self):
     # prepare form data
     _data = {'file': (StringIO('This is a test'), 'dummy.txt')}
     # build environ
     _builder = EnvironBuilder('/upload', method='POST', data=_data)
     _environ = _builder.get_environ()
     # call
     _app_iter, _status, _headers = run_wsgi_app(upload.app, _environ)
     # parse response
     for _i in _app_iter:
         # load data
         _data = json.loads(_i)
         # get filename
         _content_type = _data['contentType']
         _file_name = _data['fileName']
     # set upload path main var for removal
     self.upload_file_path = os.path.join(c.path.uploads, _file_name)
     # check content type
     assert _content_type == 'text/plain',\
             'Found: %s' % _content_type
     # check was uploaded
     assert os.path.exists(self.upload_file_path),\
             'Found: %s' % self.upload_file_path
     # open file
     _f = open(self.upload_file_path)
     # check content
     assert _f.read() == 'This is a test'
     _f.close()
Ejemplo n.º 4
0
def main(verbose=False):
    if (verbose): print "%s: start testing" % __name__
    
    builder = EnvironBuilder(path='/', method='GET', query_string=None)
    (iter, status, headers) = werkzeug.run_wsgi_app(gdispatch.get_application(), builder.get_environ())
    assert status == '200 OK', status
    assert ('Content-Length', '4') in headers
    assert "".join(iter) == "root"
    builder = EnvironBuilder(path='/sub', method='GET', query_string="foo=bar")
    (iter, status, headers) = werkzeug.run_wsgi_app(gdispatch.get_application(), builder.get_environ())
    assert status == '200 OK', status
    assert "".join(iter) == "get sub(foo=bar)"
    builder = EnvironBuilder(path='/sub', method='POST', query_string=None)
    (iter, status, headers) = werkzeug.run_wsgi_app(gdispatch.get_application(), builder.get_environ())
    assert status == '200 OK', status
    assert "".join(iter) == "post sub"
    builder = EnvironBuilder(path='/api/f1', method='GET', query_string=None)
    (iter, status, headers) = werkzeug.run_wsgi_app(gdispatch.get_application(), builder.get_environ())
    assert status == '200 OK', status
    assert "".join(iter) == "api(/api/f1)"
    builder = EnvironBuilder(path='/error', method='GET', query_string=None)
    (iter, status, headers) = werkzeug.run_wsgi_app(gdispatch.get_application(), builder.get_environ())
    assert status == '404 Not Found', status

    if (verbose): print "%s: testing complete" % __name__
Ejemplo n.º 5
0
    def open(self, *args, **kwargs):
        if self.context_preserved:
            _request_ctx_stack.pop()
            self.context_preserved = False
        kwargs.setdefault('environ_overrides', {}) \
            ['flask._preserve_context'] = self.preserve_context

        as_tuple = kwargs.pop('as_tuple', False)
        buffered = kwargs.pop('buffered', False)
        follow_redirects = kwargs.pop('follow_redirects', False)

        builder = EnvironBuilder(*args, **kwargs)

        if self.application.config.get('SERVER_NAME'):
            server_name = self.application.config.get('SERVER_NAME')
            if ':' not in server_name:
                http_host, http_port = server_name, None
            else:
                http_host, http_port = server_name.split(':', 1)
            if builder.base_url == 'http://localhost/':
                # Default Generated Base URL
                if http_port != None:
                    builder.host = http_host + ':' + http_port
                else:
                    builder.host = http_host
        old = _request_ctx_stack.top
        try:
            return Client.open(self, builder,
                               as_tuple=as_tuple,
                               buffered=buffered,
                               follow_redirects=follow_redirects)
        finally:
            self.context_preserved = _request_ctx_stack.top is not old
Ejemplo n.º 6
0
  def open(self, *args, **kw):
    # Copy the kwargs here since we will
    # modify them for passing to EnvironBuilder().
    kwargs = kw.copy()

    # Pop off extra options that open() supports but that
    # will cause EnvironBuilder() to choke.
    as_tuple = kwargs.pop('as_tuple', False)
    buffered = kwargs.pop('buffered', False)
    follow_redirects = kwargs.pop('follow_redirects', False)
    environ = None
    if not kwargs and len(args) == 1:
        if isinstance(args[0], EnvironBuilder):
            environ = args[0].get_environ()
        elif isinstance(args[0], dict):
            environ = args[0]
    if environ is None:
        builder = EnvironBuilder(*args, **kwargs)
        try:
            environ = builder.get_environ()
        finally:
            builder.close()
    
    old_env = os.environ.copy()
    old_cwd = os.getcwd()

    stripped_environ = {}
    for key in environ.keys():
      if isinstance(environ[key], basestring):
        stripped_environ[key] = environ[key]
    try:
        # Add the variables for the current environment to
        # the OS environment to mimic the appengine CGI environment.
        # Don't clear the os.environ since we have set up a number
        # of variables for the environment already. We only want to
        # add variables pertaining to the current request.
        os.environ.update(stripped_environ)
        # Add a dummy value for the appengine specific CGI variable.
        os.environ["PATH_TRANSLATED"] = "main.py"

        # Add additional environment variables like the app id and
        # version number.
        #setup_env()
        # Add the AUTH_DOMAIN
        #os.environ["AUTH_DOMAIN"]

        return super(Client, self).open(*args, **kw)
    finally:
        # Make sure we reset the os environment to the old value.
        os.environ.clear()
        os.environ.update(old_env)
Ejemplo n.º 7
0
Archivo: test.py Proyecto: IanLewis/kay
 def __init__(self, application, response_wrapper=None, use_cookies=True):
   super(Client, self).__init__(application,
                                response_wrapper=response_wrapper,
                                use_cookies=True)
   builder = EnvironBuilder()
   try:
     env = builder.get_environ()
   finally:
     builder.close()
   self.application.app._prepare(env)
   for key, submount_app in self.application.mounts.iteritems():
     if not hasattr(submount_app, 'app_settings') or key == "/_kay":
       continue
     submount_app._prepare(env)
Ejemplo n.º 8
0
 def __init__(self, application, response_wrapper=None, use_cookies=True):
     super(Client, self).__init__(application,
                                  response_wrapper=response_wrapper,
                                  use_cookies=True)
     builder = EnvironBuilder()
     try:
         env = builder.get_environ()
     finally:
         builder.close()
     self.application.app._prepare(env)
     for key, submount_app in self.application.mounts.iteritems():
         if not hasattr(submount_app, 'app_settings') or key == "/_kay":
             continue
         submount_app._prepare(env)
Ejemplo n.º 9
0
 def run(self, args):
     from IPython.Shell import IPShellEmbed
     from werkzeug import Client, EnvironBuilder
     from dextrose.context import Context
     from dextrose.http import Response
     app = load_application(args.package, args.environment)
     environ_builder = EnvironBuilder()
     environ = environ_builder.get_environ()
     request = environ_builder.get_request()
     client = Client(app, Response)
     with Context(app, environ, request, {}) as context:
         banner="Dextrose IPython shell\n%s" % args.package
         shell = IPShellEmbed(banner=banner, argv=[
             '-prompt_in1', '%s> ' % args.package,
             '-prompt_in2', '%s... ' % (' ' * (len(args.package) - 3)),
             '-prompt_out', '=> '
         ])
         shell(global_ns={}, local_ns={
             'app': app,
             'environ': environ,
             'client': client,
             'context': context
         })