def test_request_protocol_by_header(self):
func_request = self._generate_func_request(
headers={"x-forwarded-port": "8081"})
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['SERVER_PORT'], str(8081))
self.assertEqual(environ['wsgi.url_scheme'], 'https')
def test_request_has_no_query_param(self):
func_request = self._generate_func_request(
url="https://function.azurewebsites.net",
params=None)
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['QUERY_STRING'], '')
def test_response_no_headers(self):
app = self._generate_wsgi_app(response_headers=None)
func_request = self._generate_func_request()
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
wsgi_response: WsgiResponse = WsgiResponse.from_app(app, environ)
func_response: func.HttpResponse = wsgi_response.to_func_response()
self.assertEqual(func_response.headers, {})
def test_request_has_no_body(self):
func_request = self._generate_func_request(body=None)
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['CONTENT_LENGTH'], str(0))
self.assertIsInstance(environ['wsgi.input'], BytesIO)
bytes_io: BytesIO = environ['wsgi.input']
bytes_io.seek(0)
self.assertEqual(bytes_io.read(), b'')
def test_request_parse_function_context(self):
func_request = self._generate_func_request()
func_context = self._generate_func_context()
error_buffer = StringIO()
environ = WsgiRequest(func_request,
func_context).to_environ(error_buffer)
self.assertEqual(environ['azure_functions.invocation_id'],
'123e4567-e89b-12d3-a456-426655440000')
self.assertEqual(environ['azure_functions.function_name'],
'httptrigger')
self.assertEqual(environ['azure_functions.function_directory'],
'/home/roger/wwwroot/httptrigger')
def test_response_with_exception(self):
app = self._generate_wsgi_app(exception=WsgiException(
message='wsgi excpt'))
func_request = self._generate_func_request()
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
with self.assertRaises(WsgiException) as e:
wsgi_response = WsgiResponse.from_app(app, environ)
wsgi_response.to_func_response()
self.assertEqual(e.exception.message, 'wsgi excpt')
def test_request_general_environ_conversion(self):
func_request = self._generate_func_request()
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['REQUEST_METHOD'], 'POST')
self.assertEqual(environ['SCRIPT_NAME'], '')
self.assertEqual(environ['PATH_INFO'], '/api/http')
self.assertEqual(environ['QUERY_STRING'], 'firstname=rt')
self.assertEqual(environ['CONTENT_TYPE'], 'application/json')
self.assertEqual(environ['CONTENT_LENGTH'],
str(len(b'{ "lastname": "tsang" }')))
self.assertEqual(environ['SERVER_NAME'], 'function.azurewebsites.net')
self.assertEqual(environ['SERVER_PORT'], '443')
self.assertEqual(environ['SERVER_PROTOCOL'], 'HTTP/1.1')
def test_response_from_wsgi_app(self):
app = self._generate_wsgi_app()
func_request = self._generate_func_request()
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
wsgi_response: WsgiResponse = WsgiResponse.from_app(app, environ)
func_response: func.HttpResponse = wsgi_response.to_func_response()
self.assertEqual(func_response.mimetype, 'text/plain')
self.assertEqual(func_response.charset, 'utf-8')
self.assertEqual(func_response.headers['Content-Type'], 'text/plain')
self.assertEqual(func_response.status_code, 200)
self.assertEqual(func_response.get_body(), b'sample string')
def test_request_wsgi_environ_conversion(self):
func_request = self._generate_func_request()
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['wsgi.version'], (1, 0))
self.assertEqual(environ['wsgi.url_scheme'], 'https')
self.assertIsInstance(environ['wsgi.input'], BytesIO)
bytes_io: BytesIO = environ['wsgi.input']
bytes_io.seek(0)
self.assertEqual(bytes_io.read(), b'{ "lastname": "tsang" }')
self.assertIsInstance(environ['wsgi.errors'], StringIO)
string_io: StringIO = environ['wsgi.errors']
string_io.seek(0)
self.assertEqual(string_io.read(), '')
self.assertEqual(environ['wsgi.multithread'], False)
self.assertEqual(environ['wsgi.multiprocess'], False)
self.assertEqual(environ['wsgi.run_once'], False)
def test_request_protocol_by_scheme(self):
func_request = self._generate_func_request(url="http://a.b.com")
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['SERVER_PORT'], str(80))
self.assertEqual(environ['wsgi.url_scheme'], 'http')
def test_request_has_no_headers(self):
func_request = self._generate_func_request(headers=None)
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertNotIn('CONTENT_TYPE', environ)
def test_request_http_environ_conversion(self):
func_request = self._generate_func_request()
error_buffer = StringIO()
environ = WsgiRequest(func_request).to_environ(error_buffer)
self.assertEqual(environ['HTTP_X_MS_SITE_RESTRICTED_TOKEN'], 'xmsrt')