@@ -816,19 +816,23 @@ def test_can_generate_websockets_api(self, sample_websocket_app):
816816 'value' : '${aws_lambda_function.websocket_connect.arn}'
817817 },
818818 'WebsocketConnectHandlerName' : {
819- 'value' : '${aws_lambda_function.websocket_connect}'
819+ 'value' : (
820+ '${aws_lambda_function.websocket_connect.function_name}' )
820821 },
821822 'WebsocketMessageHandlerArn' : {
822823 'value' : '${aws_lambda_function.websocket_message.arn}'
823824 },
824825 'WebsocketMessageHandlerName' : {
825- 'value' : '${aws_lambda_function.websocket_message}'
826+ 'value' : (
827+ '${aws_lambda_function.websocket_message.function_name}' )
826828 },
827829 'WebsocketDisconnectHandlerArn' : {
828830 'value' : '${aws_lambda_function.websocket_disconnect.arn}'
829831 },
830832 'WebsocketDisconnectHandlerName' : {
831- 'value' : '${aws_lambda_function.websocket_disconnect}'
833+ 'value' : (
834+ '${aws_lambda_function.websocket_disconnect'
835+ '.function_name}' )
832836 },
833837 'WebsocketConnectEndpointURL' : {
834838 'value' : 'wss://${aws_apigatewayv2_api.websocket_api.id}'
@@ -984,19 +988,19 @@ def test_can_generate_custom_domain_name(self, sample_app):
984988 )
985989 template = self .generate_template (config )
986990 assert template ['resource' ]['aws_api_gateway_domain_name' ][
987- 'api_gateway_custom_domain' ] == {
988- 'domain_name' : 'example.com' ,
989- 'certificate_arn' : 'my_cert_arn' ,
990- 'security_policy' : 'TLS_1_2' ,
991- 'endpoint_configuration' : {'types' : ['EDGE' ]},
992- 'tags' : {'foo' : 'bar' },
993- }
991+ 'api_gateway_custom_domain' ] == {
992+ 'domain_name' : 'example.com' ,
993+ 'certificate_arn' : 'my_cert_arn' ,
994+ 'security_policy' : 'TLS_1_2' ,
995+ 'endpoint_configuration' : {'types' : ['EDGE' ]},
996+ 'tags' : {'foo' : 'bar' },
997+ }
994998 assert template ['resource' ]['aws_api_gateway_base_path_mapping' ][
995- 'api_gateway_custom_domain_mapping' ] == {
996- 'api_id' : '${aws_api_gateway_rest_api.rest_api.id}' ,
997- 'stage_name' : 'api' ,
998- 'domain_name' : 'example.com' ,
999- }
999+ 'api_gateway_custom_domain_mapping' ] == {
1000+ 'api_id' : '${aws_api_gateway_rest_api.rest_api.id}' ,
1001+ 'stage_name' : 'api' ,
1002+ 'domain_name' : 'example.com' ,
1003+ }
10001004 outputs = template ['output' ]
10011005 assert outputs ['AliasDomainName' ]['value' ] == (
10021006 '${aws_api_gateway_domain_name.api_gateway_custom_domain'
@@ -1020,17 +1024,17 @@ def test_can_generate_domain_for_regional_endpoint(self, sample_app):
10201024 )
10211025 template = self .generate_template (config )
10221026 assert template ['resource' ]['aws_api_gateway_domain_name' ][
1023- 'api_gateway_custom_domain' ] == {
1024- 'domain_name' : 'example.com' ,
1025- 'regional_certificate_arn' : 'my_cert_arn' ,
1026- 'endpoint_configuration' : {'types' : ['REGIONAL' ]},
1027- }
1027+ 'api_gateway_custom_domain' ] == {
1028+ 'domain_name' : 'example.com' ,
1029+ 'regional_certificate_arn' : 'my_cert_arn' ,
1030+ 'endpoint_configuration' : {'types' : ['REGIONAL' ]},
1031+ }
10281032 assert template ['resource' ]['aws_api_gateway_base_path_mapping' ][
1029- 'api_gateway_custom_domain_mapping' ] == {
1030- 'api_id' : '${aws_api_gateway_rest_api.rest_api.id}' ,
1031- 'stage_name' : 'api' ,
1032- 'domain_name' : 'example.com' ,
1033- }
1033+ 'api_gateway_custom_domain_mapping' ] == {
1034+ 'api_id' : '${aws_api_gateway_rest_api.rest_api.id}' ,
1035+ 'stage_name' : 'api' ,
1036+ 'domain_name' : 'example.com' ,
1037+ }
10341038 outputs = template ['output' ]
10351039 assert outputs ['AliasDomainName' ]['value' ] == (
10361040 '${aws_api_gateway_domain_name.api_gateway_custom_domain'
0 commit comments