Ticket #1743: test_wsgi-2.py

#!/usr/bin/python
import sys
import os
from cStringIO import StringIO
from nevow import wsgi, rend, loaders, stan, tags, testutil
# deprecated, but uses in wsgiref, so we test it too
from types import StringType

class NevowIndex(rend.Page):
    addSlash = True
    docFactory = loaders.stan(tags.html())
    
    def childFactory(self, context, name):
        return NevowIndex()

def test_write(data):
    assert isinstance(data, str), "Data must bet string instead of %r" % type(data)
    # deprecated, but uses in wsgiref, so we test it too
    assert type(data) is StringType, "Check str type of data in old-class style failed; it is %r type, not str" % type(data)

def test_response(status, headers, exc_info=None):
    assert isinstance(status, str), "Status must be string instead of %r" % type(status)
    # deprecated, but uses in wsgiref, so we test it too
    assert type(status) is StringType, "Check str type of status in old-class style failed; it is %r type, not str" % type(status)
    #
    assert len(status) >= 4, "Status must be at least 4 characters"
    assert int(status[:3]), "Status message mus begin with 3-digit code"
    assert status[3]==" ", "Status message must have a space after code"
    for name, val in headers:
        assert isinstance(name, str), "Header names must be strings instead of %r" % type(name)
        assert isinstance(val, str), "Header values must be string instead of %r" % type(val)
        # deprecated, but uses in wsgiref, so we test it too
        assert type(name) is StringType, "Check str type of header's name in old-class style failed; it is %r type, not str" % type(name)
        assert type(val) is StringType, "Check str type of header's val in old-class style failed; it is %r type, not str" % type(val)
        
    return test_write

def get_wsgi_env(method='GET', path='/', query='', proto='HTTP/1.1'):
    custom_env = {
            'HTTP_ACCEPT': 'text/xml,text/html,text/plain',
            'HTTP_HOST': 'localhost:8080',
            'PATH_INFO': path,
            'QUERY_STRING': query,
            'REMOTE_ADDR': '127.0.0.1',
            'REMOTE_HOST': 'localhost',
            'REQUEST_METHOD': method,
            'SERVER_NAME': 'localhost',
            'SERVER_PORT': '8080',
            'SERVER_PROTOCOL': proto,
            'SERVER_SOFTWARE': 'Nevow WSGI test',
            'wsgi.input': StringIO(' '.join((method,path,proto))),
            'wsgi.errors': sys.stderr,
            'wsgi.version': (1,0),
            'wsgi.run_once': False,
            'wsgi.url_scheme': 'http',
            'wsgi.multithread': True,
            'wsgi.multiprocess': True,
           }
    custom_env.update(os.environ.copy())
    return custom_env

class WSGITestCase(testutil.TestCase):
    
    def setUp(self):
        self.app = wsgi.createWSGIApplication(NevowIndex())
    
    def testWSGIData(self):
        result = self.app(get_wsgi_env(), test_response)
        for data in result:
            test_write(data)
        if hasattr(result, 'close'):
            result.close()
    
    def testWSGIHeader(self):
        result = self.app(get_wsgi_env(path='/a'), test_response)
        for data in result:
            test_write(data)
        if hasattr(result, 'close'):
            result.close()