def test_base_proxy_event_get_header_value_case_insensitive():
default_value = "default"
set_value = "value"
event = BaseProxyEvent({"headers": {}})
event._data["headers"] = {"Test": set_value}
value = event.get_header_value("test", case_sensitive=True)
assert value is None
value = event.get_header_value("test",
default_value=default_value,
case_sensitive=True)
assert value == default_value
value = event.get_header_value("Test", case_sensitive=True)
assert value == set_value
value = event.get_header_value("unknown",
default_value,
case_sensitive=True)
assert value == default_value
value = event.get_header_value("unknown", case_sensitive=True)
assert value is None
def _route(self, event: BaseProxyEvent, cors: Optional[CORSConfig]):
"""Optionally handle any of the route's configure response handling"""
if self.route is None:
return
if self.route.cors:
self._add_cors(cors or CORSConfig())
if self.route.cache_control:
self._add_cache_control(self.route.cache_control)
if self.route.compress and "gzip" in (event.get_header_value(
"accept-encoding", "") or ""):
self._compress()
def test_base_proxy_event_get_query_string_value():
default_value = "default"
set_value = "value"
event = BaseProxyEvent({})
value = event.get_query_string_value("test", default_value)
assert value == default_value
event._data["queryStringParameters"] = {"test": set_value}
value = event.get_query_string_value("test", default_value)
assert value == set_value
value = event.get_query_string_value("unknown", default_value)
assert value == default_value
value = event.get_query_string_value("unknown")
assert value is None
def test_base_proxy_event_get_header_value():
default_value = "default"
set_value = "value"
event = BaseProxyEvent({"headers": {}})
value = event.get_header_value("test", default_value)
assert value == default_value
event._data["headers"] = {"test": set_value}
value = event.get_header_value("test", default_value)
assert value == set_value
value = event.get_header_value("unknown", default_value)
assert value == default_value
value = event.get_header_value("unknown")
assert value is None
def test_base_proxy_event_get_header_value():
default_value = "default"
set_value = "value"
event = BaseProxyEvent({"headers": {}})
value = event.get_header_value("test", default_value)
assert value == default_value
event._data["headers"] = {"test": set_value}
value = event.get_header_value("test", default_value)
assert value == set_value
# Verify that the default look is case insensitive
value = event.get_header_value("Test")
assert value == set_value
value = event.get_header_value("unknown", default_value)
assert value == default_value
value = event.get_header_value("unknown")
assert value is None