Skip to content

Commit c1e487d

Browse files
committed
chore: var names cleanup
1 parent a861309 commit c1e487d

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

tests/e2e/utils/infrastructure.py

+9-11
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def create_lambda_functions(self, function_props: Optional[Dict] = None):
6767
Parameters
6868
----------
6969
function_props: Optional[Dict]
70-
CDK Lambda FunctionProps as dictionary to override defaults
70+
Dictionary representing CDK Lambda FunctionProps to override defaults
7171
7272
Examples
7373
--------
@@ -88,16 +88,14 @@ def create_lambda_functions(self, function_props: Optional[Dict] = None):
8888
"""
8989
handlers = list(self.handlers_dir.rglob("*.py"))
9090
source = Code.from_asset(f"{self.handlers_dir}")
91-
props_override = function_props or {}
9291
logger.debug(f"Creating functions for handlers: {handlers}")
9392
if not self.layer_arn:
9493
raise ValueError(
9594
"""Lambda Layer ARN cannot be empty when creating Lambda functions.
9695
Make sure to inject `lambda_layer_arn` fixture and pass at the constructor level"""
9796
)
9897

99-
layer = LayerVersion.from_layer_version_arn(self.stack, "layer-arn", layer_version_arn=self.layer_arn)
100-
98+
function_settings_override = function_props or {}
10199
for fn in handlers:
102100
fn_name = fn.stem
103101
fn_name_pascal_case = fn_name.title().replace("_", "") # basic_handler -> BasicHandler
@@ -108,24 +106,24 @@ def create_lambda_functions(self, function_props: Optional[Dict] = None):
108106
"handler": f"{fn_name}.lambda_handler",
109107
"tracing": Tracing.ACTIVE,
110108
"runtime": Runtime.PYTHON_3_9,
111-
"layers": [layer],
112-
**props_override,
109+
"layers": [
110+
LayerVersion.from_layer_version_arn(self.stack, "layer-arn", layer_version_arn=self.layer_arn)
111+
],
112+
**function_settings_override,
113113
}
114114

115-
function_python = Function(self.stack, **function_settings)
115+
function = Function(self.stack, **function_settings)
116116

117117
aws_logs.LogGroup(
118118
self.stack,
119119
id=f"{fn_name}-lg",
120-
log_group_name=f"/aws/lambda/{function_python.function_name}",
120+
log_group_name=f"/aws/lambda/{function.function_name}",
121121
retention=aws_logs.RetentionDays.ONE_DAY,
122122
removal_policy=RemovalPolicy.DESTROY,
123123
)
124124

125125
# CFN Outputs only support hyphen hence pascal case
126-
self.add_cfn_output(
127-
name=fn_name_pascal_case, value=function_python.function_name, arn=function_python.function_arn
128-
)
126+
self.add_cfn_output(name=fn_name_pascal_case, value=function.function_name, arn=function.function_arn)
129127

130128
def deploy(self) -> Dict[str, str]:
131129
"""Creates CloudFormation Stack and return stack outputs as dict

0 commit comments

Comments
 (0)