catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add set-instance-health support

Browse source
This commit is contained in:
Paul Carleton 2017-10-20 11:25:09 -07:00
commit 685e9163a8
3 changed files with 56 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
moto/autoscaling/models.py
View file

@ -433,6 +433,12 @@ class AutoScalingBackend(BaseBackend):
group.instance_states.extend(new_instances)
self.update_attached_elbs(group.name)
def set_instance_health(self, instance_id, health_status, should_respect_grace_period):
instance = self.ec2_backend.get_instance(instance_id)
instance_state = next(instance_state for group in self.autoscaling_groups.values()
for instance_state in group.instance_states if instance_state.instance.id == instance.id)
instance_state.health_status = health_status
def detach_instances(self, group_name, instance_ids, should_decrement):
group = self.autoscaling_groups[group_name]
original_size = len(group.instance_states)