@@ -84,7 +84,9 @@ def test_entrypoint_success():
8484
8585 with patch (
8686 "cloudformation_cli_python_lib.resource.ProviderLogHandler.setup"
87- ) as mock_log_delivery :
87+ ) as mock_log_delivery , patch (
88+ "cloudformation_cli_python_lib.hook._get_boto_session" , autospec = True
89+ ):
8890 event = resource .__call__ .__wrapped__ ( # pylint: disable=no-member
8991 resource , ENTRYPOINT_PAYLOAD , None
9092 )
@@ -164,7 +166,9 @@ def test_entrypoint_with_context():
164166 )
165167 mock_handler = resource .handler (Action .CREATE )(Mock (return_value = event ))
166168
167- with patch ("cloudformation_cli_python_lib.resource.ProviderLogHandler.setup" ):
169+ with patch (
170+ "cloudformation_cli_python_lib.resource.ProviderLogHandler.setup"
171+ ), patch ("cloudformation_cli_python_lib.hook._get_boto_session" , autospec = True ):
168172 resource .__call__ .__wrapped__ ( # pylint: disable=no-member
169173 resource , payload , None
170174 )
@@ -181,7 +185,9 @@ def test_entrypoint_ignore_remove_fields_from_response():
181185
182186 with patch (
183187 "cloudformation_cli_python_lib.resource.ProviderLogHandler.setup"
184- ) as mock_log_delivery :
188+ ) as mock_log_delivery , patch (
189+ "cloudformation_cli_python_lib.hook._get_boto_session" , autospec = True
190+ ):
185191 event = resource .__call__ .__wrapped__ ( # pylint: disable=no-member
186192 resource , ENTRYPOINT_PAYLOAD , None
187193 )
0 commit comments