@@ -173,7 +173,7 @@ def find_change_set_v2(
173
173
state : CloudFormationStore , change_set_name : str , stack_name : str | None = None
174
174
) -> ChangeSet | None :
175
175
if is_changeset_arn (change_set_name ):
176
- return state .change_sets [ change_set_name ]
176
+ return state .change_sets . get ( change_set_name )
177
177
else :
178
178
if stack_name is not None :
179
179
stack = find_stack_v2 (state , stack_name )
@@ -185,7 +185,9 @@ def find_change_set_v2(
185
185
if change_set_candidate .change_set_name == change_set_name :
186
186
return change_set_candidate
187
187
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
+ )
189
191
190
192
191
193
def find_stack_set_v2 (state : CloudFormationStore , stack_set_name : str ) -> StackSet | None :
@@ -618,16 +620,7 @@ def delete_change_set(
618
620
** kwargs ,
619
621
) -> DeleteChangeSetOutput :
620
622
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 )
631
624
if not change_set :
632
625
return DeleteChangeSetOutput ()
633
626
0 commit comments