def setUp(self):
super(Tests, self).setUp()
zope.component.provideAdapter(HTTPCharsets, [IHTTPRequest],
IUserPreferredCharsets, '')
obj = tested_object()
obj.folder = tested_object()
obj.folder.item = tested_object()
obj._protected = tested_object()
pub = PublicationWithConflict(obj)
def application(environ, start_response):
request = BrowserRequest(environ['wsgi.input'], environ)
request.setPublication(pub)
request = publish(request)
response = request.response
start_response(response.getStatusString(), response.getHeaders())
return response.consumeBodyIter()
td.setThreadCount(4)
# Bind to any port on localhost.
self.server = WSGIHTTPServer(application, 'Browser',
LOCALHOST, 0, task_dispatcher=td)
self.port = self.server.socket.getsockname()[1]
self.run_loop = 1
self.thread = Thread(target=self.loop)
self.thread.start()
sleep(0.1) # Give the thread some time to start.
def test_secure_environment(self):
from zope.server.http.wsgihttpserver import WSGIHTTPServer
class Task(object):
def __init__(self, env):
self.env = env
self.request_data = self
def getCGIEnvironment(self):
return self.env
def getBodyStream(self):
return None
env = WSGIHTTPServer._constructWSGIEnvironment(Task({}))
self.assertEqual("http", env['wsgi.url_scheme'])
env = WSGIHTTPServer._constructWSGIEnvironment(Task({'HTTPS': 'on'}))
self.assertEqual("https", env['wsgi.url_scheme'])
env = WSGIHTTPServer._constructWSGIEnvironment(
Task({'SERVER_PORT_SECURE': "1"}))
self.assertEqual("https", env['wsgi.url_scheme'])
def test_secure_environment(self):
from zope.server.http.wsgihttpserver import WSGIHTTPServer
class Task(object):
def __init__(self, env):
self.env = env
self.request_data = self
def getCGIEnvironment(self):
return self.env
def getBodyStream(self):
return None
env = WSGIHTTPServer._constructWSGIEnvironment(Task({}))
self.assertEqual("http", env['wsgi.url_scheme'])
env = WSGIHTTPServer._constructWSGIEnvironment(Task({'HTTPS': 'on'}))
self.assertEqual("https", env['wsgi.url_scheme'])
env = WSGIHTTPServer._constructWSGIEnvironment(Task({'SERVER_PORT_SECURE': "1"}))
self.assertEqual("https", env['wsgi.url_scheme'])
class Tests(PlacelessSetup, unittest.TestCase):
def setUp(self):
super(Tests, self).setUp()
zope.component.provideAdapter(HTTPCharsets, [IHTTPRequest],
IUserPreferredCharsets, '')
obj = tested_object()
obj.folder = tested_object()
obj.folder.item = tested_object()
obj._protected = tested_object()
pub = PublicationWithConflict(obj)
def application(environ, start_response):
request = BrowserRequest(environ['wsgi.input'], environ)
request.setPublication(pub)
request = publish(request)
response = request.response
start_response(response.getStatusString(), response.getHeaders())
return response.consumeBodyIter()
td.setThreadCount(4)
# Bind to any port on localhost.
self.server = WSGIHTTPServer(application, 'Browser',
LOCALHOST, 0, task_dispatcher=td)
self.port = self.server.socket.getsockname()[1]
self.run_loop = 1
self.thread = Thread(target=self.loop)
self.thread.start()
sleep(0.1) # Give the thread some time to start.
def tearDown(self):
self.run_loop = 0
self.thread.join()
td.shutdown()
self.server.close()
def loop(self):
while self.run_loop:
poll(0.1, socket_map)
def testResponse(self, path='/', status_expected=200,
add_headers=None, request_body=''):
h = HTTPConnection(LOCALHOST, self.port)
h.putrequest('GET', path)
h.putheader('Accept', 'text/plain')
if add_headers:
for k, v in add_headers.items():
h.putheader(k, v)
if request_body:
h.putheader('Content-Length', str(int(len(request_body))))
h.endheaders()
if request_body:
h.send(request_body)
response = h.getresponse()
length = int(response.getheader('Content-Length', '0'))
if length:
response_body = response.read(length)
else:
response_body = ''
# Please do not disable the status code check. It must work.
self.failUnlessEqual(int(response.status), status_expected)
self.failUnlessEqual(length, len(response_body))
if (status_expected == 200):
if path == '/': path = ''
expect_response = 'URL invoked: http://%s:%d%s' % (LOCALHOST,
self.port, path)
self.failUnlessEqual(response_body, expect_response)
def testDeeperPath(self):
self.testResponse(path='/folder/item')
def testNotFound(self):
self.testResponse(path='/foo/bar', status_expected=404)
def testUnauthorized(self):
self.testResponse(path='/_protected', status_expected=401)
def testRedirectMethod(self):
self.testResponse(path='/redirect_method', status_expected=303)
def testRedirectException(self):
self.testResponse(path='/redirect_exception', status_expected=303)
self.testResponse(path='/folder/redirect_exception',
status_expected=303)
def testConflictRetry(self):
# Expect the "Accepted" response since the retries will succeed.
self.testResponse(path='/conflict?wait_tries=2', status_expected=202)
def testFailedConflictRetry(self):
# Expect a "Conflict" response since there will be too many
# conflicts.
self.testResponse(path='/conflict?wait_tries=10', status_expected=409)
def test_server_uses_iterable(self):
# Make sure that the task write method isn't called with a
# str or non iterable
class FakeTask:
getCGIEnvironment = lambda _: {}
class request_data:
getBodyStream = lambda _: StringIO.StringIO()
request_data = request_data()
setResponseStatus = appendResponseHeaders = lambda *_: None
def write(self, v):
if isinstance(v, str):
raise TypeError("Should only write iterables")
list(v)
self.server.executeRequest(FakeTask())
