@@ -173,7 +173,7 @@ def find_change_set_v2(
173173 state : CloudFormationStore , change_set_name : str , stack_name : str | None = None
174174) -> ChangeSet | None :
175175 if is_changeset_arn (change_set_name ):
176- return state .change_sets [ change_set_name ]
176+ return state .change_sets . get ( change_set_name )
177177 else :
178178 if stack_name is not None :
179179 stack = find_stack_v2 (state , stack_name )
@@ -185,7 +185,9 @@ def find_change_set_v2(
185185 if change_set_candidate .change_set_name == change_set_name :
186186 return change_set_candidate
187187 else :
188- raise ValueError ("No stack name specified when finding change set" )
188+ raise ValidationError (
189+ "StackName must be specified if ChangeSetName is not specified as an ARN."
190+ )
189191
190192
191193def find_stack_set_v2 (state : CloudFormationStore , stack_set_name : str ) -> StackSet | None :
@@ -618,16 +620,7 @@ def delete_change_set(
618620 ** kwargs ,
619621 ) -> DeleteChangeSetOutput :
620622 state = get_cloudformation_store (context .account_id , context .region )
621-
622- if is_changeset_arn (change_set_name ):
623- change_set = state .change_sets .get (change_set_name )
624- elif not is_changeset_arn (change_set_name ) and stack_name :
625- change_set = find_change_set_v2 (state , change_set_name , stack_name )
626- else :
627- raise ValidationError (
628- "StackName must be specified if ChangeSetName is not specified as an ARN."
629- )
630-
623+ change_set = find_change_set_v2 (state , change_set_name , stack_name )
631624 if not change_set :
632625 return DeleteChangeSetOutput ()
633626
0 commit comments