56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import re
|
|
import sys
|
|
|
|
import falcon
|
|
import falcon.testing as testing
|
|
|
|
|
|
class TestWSGIInterface(object):
|
|
|
|
def test_srmock(self):
|
|
mock = testing.StartResponseMock()
|
|
mock(falcon.HTTP_200, ())
|
|
|
|
assert mock.status == falcon.HTTP_200
|
|
assert mock.exc_info is None
|
|
|
|
mock = testing.StartResponseMock()
|
|
exc_info = sys.exc_info()
|
|
mock(falcon.HTTP_200, (), exc_info)
|
|
|
|
assert mock.exc_info == exc_info
|
|
|
|
def test_pep3333(self):
|
|
api = falcon.API()
|
|
mock = testing.StartResponseMock()
|
|
|
|
# Simulate a web request (normally done though a WSGI server)
|
|
response = api(testing.create_environ(), mock)
|
|
|
|
# Verify that the response is iterable
|
|
assert _is_iterable(response)
|
|
|
|
# Make sure start_response was passed a valid status string
|
|
assert mock.call_count == 1
|
|
assert isinstance(mock.status, str)
|
|
assert re.match('^\d+[a-zA-Z\s]+$', mock.status)
|
|
|
|
# Verify headers is a list of tuples, each containing a pair of strings
|
|
assert isinstance(mock.headers, list)
|
|
if len(mock.headers) != 0:
|
|
header = mock.headers[0]
|
|
assert isinstance(header, tuple)
|
|
assert len(header) == 2
|
|
assert isinstance(header[0], str)
|
|
assert isinstance(header[1], str)
|
|
|
|
|
|
def _is_iterable(thing):
|
|
try:
|
|
for i in thing:
|
|
break
|
|
|
|
return True
|
|
except:
|
|
return False
|