Add set-instance-health support

tests/test_autoscaling/test_autoscaling.py

@@ -982,3 +982,34 @@ def test_describe_instance_health():
 
     instance1 = response['AutoScalingGroups'][0]['Instances'][0]
     instance1['HealthStatus'].should.equal('Healthy')
+
+@mock_autoscaling
+@mock_ec2
+def test_set_instance_health():
+    client = boto3.client('autoscaling', region_name='us-east-1')
+    _ = client.create_launch_configuration(
+        LaunchConfigurationName='test_launch_configuration'
+    )
+    client.create_auto_scaling_group(
+        AutoScalingGroupName='test_asg',
+        LaunchConfigurationName='test_launch_configuration',
+        MinSize=2,
+        MaxSize=4,
+        DesiredCapacity=2,
+    )
+
+    response = client.describe_auto_scaling_groups(
+        AutoScalingGroupNames=['test_asg']
+    )
+
+    instance1 = response['AutoScalingGroups'][0]['Instances'][0]
+    instance1['HealthStatus'].should.equal('Healthy')
+
+    client.set_instance_health(InstanceId=instance1['InstanceId'], HealthStatus='Unhealthy')
+
+    response = client.describe_auto_scaling_groups(
+        AutoScalingGroupNames=['test_asg']
+    )
+
+    instance1 = response['AutoScalingGroups'][0]['Instances'][0]
+    instance1['HealthStatus'].should.equal('Unhealthy')