Skip to content

Commit 6a9027b

Browse files
authored
chore: Only do request logging when internal flag is provided (#2514)
* Only do logging when it's internal * Black reformatting * Fix typo
1 parent 2acbc24 commit 6a9027b

File tree

2 files changed

+19
-11
lines changed

2 files changed

+19
-11
lines changed

integration/combination/test_api_with_authorizers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -436,10 +436,10 @@ def verify_authorized_request(
436436
header_value=None,
437437
):
438438
if not header_key or not header_value:
439-
response = BaseTest.do_get_request_with_logging(url)
439+
response = self.do_get_request_with_logging(url)
440440
else:
441441
headers = {header_key: header_value}
442-
response = BaseTest.do_get_request_with_logging(url, headers)
442+
response = self.do_get_request_with_logging(url, headers)
443443
status = response.status_code
444444

445445
if status != expected_status_code:

integration/helpers/base_test.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,14 @@ def s3_bucket(self, get_s3):
6464
def case_name(self, request):
6565
self.testcase = request.node.name
6666

67+
@pytest.fixture(autouse=True)
68+
def internal_check(self, check_internal):
69+
self.internal = check_internal
70+
6771
@classmethod
68-
@pytest.mark.usefixtures("get_prefix", "get_stage", "check_internal", "parameter_values", "get_s3")
72+
@pytest.mark.usefixtures(
73+
"get_prefix", "get_stage", "check_internal", "parameter_values", "get_s3", "check_internal"
74+
)
6975
def setUpClass(cls):
7076
cls.FUNCTION_OUTPUT = "hello"
7177
cls.tests_integ_dir = Path(__file__).resolve().parents[1]
@@ -560,10 +566,11 @@ def do_get_request_with_logging(self, url, headers=None):
560566
"""
561567
response = requests.get(url, headers=headers) if headers else requests.get(url)
562568
amazon_headers = RequestUtils(response).get_amazon_headers()
563-
REQUEST_LOGGER.info(
564-
"Request made to " + url,
565-
extra={"test": self.testcase, "status": response.status_code, "headers": amazon_headers},
566-
)
569+
if self.internal:
570+
REQUEST_LOGGER.info(
571+
"Request made to " + url,
572+
extra={"test": self.testcase, "status": response.status_code, "headers": amazon_headers},
573+
)
567574
return response
568575

569576
def do_options_request_with_logging(self, url, headers=None):
@@ -578,8 +585,9 @@ def do_options_request_with_logging(self, url, headers=None):
578585
"""
579586
response = requests.options(url, headers=headers) if headers else requests.options(url)
580587
amazon_headers = RequestUtils(response).get_amazon_headers()
581-
REQUEST_LOGGER.info(
582-
"Request made to " + url,
583-
extra={"test": self.testcase, "status": response.status_code, "headers": amazon_headers},
584-
)
588+
if self.internal:
589+
REQUEST_LOGGER.info(
590+
"Request made to " + url,
591+
extra={"test": self.testcase, "status": response.status_code, "headers": amazon_headers},
592+
)
585593
return response

0 commit comments

Comments
 (0)