chore: Add missing import and add partition into base test for integration test (#139)

* Manual sync from public

* Formatting

Author: qingchm

integration/combination/test_api_with_authorizer_apikey.py

@@ -19,12 +19,9 @@ def test_authorizer_apikey(self):
         apigw_client = self.client_provider.api_client
 
         authorizers = apigw_client.get_authorizers(restApiId=rest_api_id)["items"]
-        lambda_authorizer_uri = (
-            "arn:aws:apigateway:"
-            + self.my_region
-            + ":lambda:path/2015-03-31/functions/"
-            + stack_outputs["AuthorizerFunctionArn"]
-            + "/invocations"
+
+        lambda_authorizer_uri = "arn:{}:apigateway:{}:lambda:path/2015-03-31/functions/{}/invocations".format(
+            self.partition, self.my_region, stack_outputs["AuthorizerFunctionArn"]
         )
 
         lambda_token_authorizer = get_authorizer_by_name(authorizers, "MyLambdaTokenAuth")

integration/combination/test_function_with_self_managed_kafka.py

@@ -1,3 +1,4 @@
+from unittest.case import skipIf
 import pytest
 
 from integration.helpers.base_test import BaseTest

integration/helpers/base_test.py

@@ -72,6 +72,7 @@ def setUpClass(cls):
         cls.code_dir = Path(cls.resources_dir, "code")
         cls.session = boto3.session.Session()
         cls.my_region = cls.session.region_name
+        cls.partition = cls.session.get_partition_for_region(cls.my_region)
         cls.client_provider = ClientProvider()
         cls.file_to_s3_uri_map = read_test_config_file("file_to_s3_map_modified.json")
         cls.code_key_to_file = read_test_config_file("code_key_to_file_map.json")