Download Install Tutorial Docs FAQ Tools WikiLicense Team IRC Planet Involvement Shop Book

Changeset 477

Show
Ignore:
Timestamp:
07/18/05 01:03:12
Author:
fumanchu
Message:

Force Host request header on serverless tests.

Files:

Legend:

Unmodified
Added
Removed
Modified
Copied
Moved
  • trunk/cherrypy/test/helper.py

    r476 r477  
    8585    def _getRequest(self, url, headers, method, body): 
    8686        # Like getPage, but for serverless requests. 
     87        webtest.ServerError.on = False 
    8788        self.url = url 
     89         
    8890        requestLine = "%s %s HTTP/1.0" % (method.upper(), url) 
    89         headers = webtest.cleanHeaders(headers, method, body) 
    90          
    91         found = False 
    92         for k, v in headers: 
    93             if k.lower() == 'host': 
    94                 found = True 
    95                 break 
    96         if not found: 
    97             headers.append(("Host", "%s:%s" % (HOST, PORT))) 
    98          
     91        headers = webtest.cleanHeaders(headers, method, body, HOST, PORT) 
    9992        if body is not None: 
    10093            body = StringIO.StringIO(body) 
    10194         
    102         webtest.ServerError.on = False 
    10395        cherrypy.server.request(HOST, HOST, requestLine, headers, body, "http") 
     96         
    10497        self.status = cherrypy.response.status 
    10598        self.headers = cherrypy.response.headers 
  • trunk/cherrypy/test/webtest.py

    r476 r477  
    274274 
    275275 
    276 def cleanHeaders(headers, method, body): 
     276def cleanHeaders(headers, method, body, host, port): 
    277277    if headers is None: 
    278278        headers = [] 
     279     
     280    # Add the required Host header if not present 
     281    found = False 
     282    for k, v in headers: 
     283        if k.lower() == 'host': 
     284            found = True 
     285            break 
     286    if not found: 
     287        headers.append(("Host", "%s:%s" % (host, port))) 
    279288     
    280289    if method in ("POST", "PUT"): 
     
    295304            host="127.0.0.1", port=8000): 
    296305     
    297     headers = cleanHeaders(headers, method, body
     306    headers = cleanHeaders(headers, method, body, host, port
    298307     
    299308    # Trying 10 times is simply in case of socket errors. 

Hosted by WebFaction

Log in as guest/cpguest to create tickets