def test_if_resp_headers_are_correct(self, agent_ar_process_perclass,
valid_user_header, path,
caching_headers_test,
expect_http_status, vhost):
headers_present = {}
headers_absent = []
headers_present['Server'] = "openresty"
if caching_headers_test is True:
headers_present[
'Cache-Control'] = "no-cache, no-store, must-revalidate"
headers_present['Pragma'] = "no-cache"
headers_present['Expires'] = "0"
elif caching_headers_test is False:
headers_absent.append("Cache-Control")
headers_absent.append("Pragma")
headers_absent.append("Expires")
# caching_headers_test == "skip", do nothing
req_headers = copy.deepcopy(valid_user_header)
if vhost is not None:
req_headers["Host"] = vhost
generic_verify_response_test(
agent_ar_process_perclass,
req_headers,
path,
assert_headers=headers_present,
assert_headers_absent=headers_absent,
assert_status=expect_http_status,
)
def test_if_xaccel_header_is_passed_to_client_by_ar(
self, master_ar_process_perclass, valid_user_header, mocker):
accel_buff_header = {"X-Accel-Buffering": "TEST"}
mocker.send_command(
endpoint_id='http:///run/dcos/dcos-log.sock',
func_name='set_response_headers',
aux_data=accel_buff_header,
)
generic_verify_response_test(master_ar_process_perclass,
valid_user_header,
'/system/v1/logs/foo/bar',
assert_headers=accel_buff_header)
def test_if_xaccel_header_is_passed_to_client_by_ar(
self,
master_ar_process_perclass,
valid_user_header,
mocker):
accel_buff_header = {"X-Accel-Buffering": "TEST"}
mocker.send_command(
endpoint_id='http:///run/dcos/dcos-log.sock',
func_name='set_response_headers',
aux_data=accel_buff_header,
)
generic_verify_response_test(
master_ar_process_perclass,
valid_user_header,
'/system/v1/logs/foo/bar',
assert_headers=accel_buff_header)
