kubernetes
/
kops
mirror of https://github.com/kubernetes/kops.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support multiple lifecycle hooks for the same ASG in our mocks

This commit is contained in:
Ole Markus With 2022-08-18 19:47:39 +02:00
parent 3ace7fd47d
commit cdea136e64
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cloudmock/aws/mockautoscaling/group.go
View File

@ -358,7 +358,8 @@ func (m *MockAutoscaling) PutLifecycleHook(input *autoscaling.PutLifecycleHookIn
if m.LifecycleHooks == nil { if m.LifecycleHooks == nil {
m.LifecycleHooks = make(map[string]*autoscaling.LifecycleHook) m.LifecycleHooks = make(map[string]*autoscaling.LifecycleHook)
} }
m.LifecycleHooks[*hook.AutoScalingGroupName] = hook name := *input.AutoScalingGroupName + "::" + *input.LifecycleHookName
m.LifecycleHooks[name] = hook
return &autoscaling.PutLifecycleHookOutput{}, nil return &autoscaling.PutLifecycleHookOutput{}, nil
} }
@ -367,13 +368,14 @@ func (m *MockAutoscaling) DescribeLifecycleHooks(input *autoscaling.DescribeLife
m.mutex.Lock() m.mutex.Lock()
defer m.mutex.Unlock() defer m.mutex.Unlock()
name := *input.AutoScalingGroupName
response := &autoscaling.DescribeLifecycleHooksOutput{} response := &autoscaling.DescribeLifecycleHooksOutput{}
for _, lifecycleHookName := range input.LifecycleHookNames {
name := *input.AutoScalingGroupName + "::" + *lifecycleHookName
hook := m.LifecycleHooks[name] hook := m.LifecycleHooks[name]
if hook == nil { if hook != nil {
return response, nil response.LifecycleHooks = append(response.LifecycleHooks, hook)
}
} }
response.LifecycleHooks = []*autoscaling.LifecycleHook{hook}
return response, nil return response, nil
} }