Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor Autoscaling tests (#5426)

Browse Source
This commit is contained in:
Bert Blommers 2022-08-28 14:41:11 +00:00 committed by GitHub
parent 257d775dfb
commit 067e008716
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 1153 additions and 1414 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1296
tests/test_autoscaling/test_autoscaling.py
View File

File diff suppressed because it is too large Load Diff

1058
tests/test_autoscaling/test_elb.py Normal file
View File

File diff suppressed because it is too large Load Diff

165
tests/test_autoscaling/test_elbv2.py
View File

@ -1,25 +1,29 @@
import boto3 import boto3
import sure # noqa # pylint: disable=unused-import import sure # noqa # pylint: disable=unused-import
import unittest
from moto import mock_autoscaling, mock_elbv2 from moto import mock_autoscaling, mock_elbv2
from .utils import setup_networking
from tests import EXAMPLE_AMI_ID from tests import EXAMPLE_AMI_ID
from uuid import uuid4
from .utils import setup_networking
@mock_elbv2 @mock_elbv2
@mock_autoscaling @mock_autoscaling
def test_attach_detach_target_groups(): class TestAutoscalignELBv2(unittest.TestCase):
mocked_networking = setup_networking() def setUp(self) -> None:
INSTANCE_COUNT = 2 self.mocked_networking = setup_networking()
client = boto3.client("autoscaling", region_name="us-east-1") self.instance_count = 2
elbv2_client = boto3.client("elbv2", region_name="us-east-1") self.as_client = boto3.client("autoscaling", region_name="us-east-1")
self.elbv2_client = boto3.client("elbv2", region_name="us-east-1")
response = elbv2_client.create_target_group( self.target_name = str(uuid4())[0:6]
Name="a-target", self.lc_name = str(uuid4())[0:6]
self.asg_name = str(uuid4())[0:6]
response = self.elbv2_client.create_target_group(
Name=self.target_name,
Protocol="HTTP", Protocol="HTTP",
Port=8080, Port=8080,
VpcId=mocked_networking["vpc"], VpcId=self.mocked_networking["vpc"],
HealthCheckProtocol="HTTP", HealthCheckProtocol="HTTP",
HealthCheckPort="8080", HealthCheckPort="8080",
HealthCheckPath="/", HealthCheckPath="/",
@ -29,111 +33,78 @@ def test_attach_detach_target_groups():
UnhealthyThresholdCount=2, UnhealthyThresholdCount=2,
Matcher={"HttpCode": "200"}, Matcher={"HttpCode": "200"},
) )
target_group_arn = response["TargetGroups"][0]["TargetGroupArn"] self.target_group_arn = response["TargetGroups"][0]["TargetGroupArn"]
client.create_launch_configuration( self.as_client.create_launch_configuration(
LaunchConfigurationName="test_launch_configuration", LaunchConfigurationName=self.lc_name,
ImageId=EXAMPLE_AMI_ID, ImageId=EXAMPLE_AMI_ID,
InstanceType="t2.medium", InstanceType="t2.medium",
) )
self.as_client.create_auto_scaling_group(
# create asg, attach to target group on create AutoScalingGroupName=self.asg_name,
client.create_auto_scaling_group( LaunchConfigurationName=self.lc_name,
AutoScalingGroupName="test_asg",
LaunchConfigurationName="test_launch_configuration",
MinSize=0, MinSize=0,
MaxSize=INSTANCE_COUNT, MaxSize=self.instance_count,
DesiredCapacity=INSTANCE_COUNT, DesiredCapacity=self.instance_count,
TargetGroupARNs=[target_group_arn], TargetGroupARNs=[self.target_group_arn],
VPCZoneIdentifier=mocked_networking["subnet1"], VPCZoneIdentifier=self.mocked_networking["subnet1"],
) )
response = self.as_client.describe_load_balancer_target_groups(
AutoScalingGroupName=self.asg_name
)
list(response["LoadBalancerTargetGroups"]).should.have.length_of(1)
response = self.elbv2_client.describe_target_health(
TargetGroupArn=self.target_group_arn
)
list(response["TargetHealthDescriptions"]).should.have.length_of(
self.instance_count
)
def test_attach_detach_target_groups(self):
# create asg without attaching to target group # create asg without attaching to target group
client.create_auto_scaling_group( asg_name2 = str(uuid4())
AutoScalingGroupName="test_asg2", self.as_client.create_auto_scaling_group(
LaunchConfigurationName="test_launch_configuration", AutoScalingGroupName=asg_name2,
LaunchConfigurationName=self.lc_name,
MinSize=0, MinSize=0,
MaxSize=INSTANCE_COUNT, MaxSize=self.instance_count,
DesiredCapacity=INSTANCE_COUNT, DesiredCapacity=self.instance_count,
VPCZoneIdentifier=mocked_networking["subnet2"], VPCZoneIdentifier=self.mocked_networking["subnet2"],
) )
response = client.describe_load_balancer_target_groups( self.as_client.attach_load_balancer_target_groups(
AutoScalingGroupName="test_asg" AutoScalingGroupName=asg_name2, TargetGroupARNs=[self.target_group_arn]
)
list(response["LoadBalancerTargetGroups"]).should.have.length_of(1)
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn)
list(response["TargetHealthDescriptions"]).should.have.length_of(INSTANCE_COUNT)
client.attach_load_balancer_target_groups(
AutoScalingGroupName="test_asg2", TargetGroupARNs=[target_group_arn]
) )
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn) response = self.elbv2_client.describe_target_health(
list(response["TargetHealthDescriptions"]).should.have.length_of(INSTANCE_COUNT * 2) TargetGroupArn=self.target_group_arn
response = client.detach_load_balancer_target_groups(
AutoScalingGroupName="test_asg2", TargetGroupARNs=[target_group_arn]
) )
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn) list(response["TargetHealthDescriptions"]).should.have.length_of(
list(response["TargetHealthDescriptions"]).should.have.length_of(INSTANCE_COUNT) self.instance_count * 2
@mock_elbv2
@mock_autoscaling
def test_detach_all_target_groups():
mocked_networking = setup_networking()
INSTANCE_COUNT = 2
client = boto3.client("autoscaling", region_name="us-east-1")
elbv2_client = boto3.client("elbv2", region_name="us-east-1")
response = elbv2_client.create_target_group(
Name="a-target",
Protocol="HTTP",
Port=8080,
VpcId=mocked_networking["vpc"],
HealthCheckProtocol="HTTP",
HealthCheckPort="8080",
HealthCheckPath="/",
HealthCheckIntervalSeconds=5,
HealthCheckTimeoutSeconds=5,
HealthyThresholdCount=5,
UnhealthyThresholdCount=2,
Matcher={"HttpCode": "200"},
)
target_group_arn = response["TargetGroups"][0]["TargetGroupArn"]
client.create_launch_configuration(
LaunchConfigurationName="test_launch_configuration",
ImageId=EXAMPLE_AMI_ID,
InstanceType="t2.medium",
) )
client.create_auto_scaling_group( response = self.as_client.detach_load_balancer_target_groups(
AutoScalingGroupName="test_asg", AutoScalingGroupName=asg_name2, TargetGroupARNs=[self.target_group_arn]
LaunchConfigurationName="test_launch_configuration", )
MinSize=0, response = self.elbv2_client.describe_target_health(
MaxSize=INSTANCE_COUNT, TargetGroupArn=self.target_group_arn
DesiredCapacity=INSTANCE_COUNT, )
TargetGroupARNs=[target_group_arn], list(response["TargetHealthDescriptions"]).should.have.length_of(
VPCZoneIdentifier=mocked_networking["subnet1"], self.instance_count
) )
response = client.describe_load_balancer_target_groups( def test_detach_all_target_groups(self):
AutoScalingGroupName="test_asg" response = self.as_client.detach_load_balancer_target_groups(
) AutoScalingGroupName=self.asg_name, TargetGroupARNs=[self.target_group_arn]
list(response["LoadBalancerTargetGroups"]).should.have.length_of(1)
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn)
list(response["TargetHealthDescriptions"]).should.have.length_of(INSTANCE_COUNT)
response = client.detach_load_balancer_target_groups(
AutoScalingGroupName="test_asg", TargetGroupARNs=[target_group_arn]
) )
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn) response = self.elbv2_client.describe_target_health(
TargetGroupArn=self.target_group_arn
)
list(response["TargetHealthDescriptions"]).should.have.length_of(0) list(response["TargetHealthDescriptions"]).should.have.length_of(0)
response = client.describe_load_balancer_target_groups( response = self.as_client.describe_load_balancer_target_groups(
AutoScalingGroupName="test_asg" AutoScalingGroupName=self.asg_name
) )
list(response["LoadBalancerTargetGroups"]).should.have.length_of(0) list(response["LoadBalancerTargetGroups"]).should.have.length_of(0)