Refactor Autoscaling tests (#5426)
This commit is contained in:
Bert Blommers
GitHub
parent
257d775dfb
commit
067e008716
File diff suppressed because it is too large
Load Diff
1058
tests/test_autoscaling/test_elb.py
Normal file
1058
tests/test_autoscaling/test_elb.py
Normal file
File diff suppressed because it is too large
Load Diff
@ -1,139 +1,110 @@
|
||||
import boto3
|
||||
|
||||
import sure # noqa # pylint: disable=unused-import
|
||||
import unittest
|
||||
from moto import mock_autoscaling, mock_elbv2
|
||||
|
||||
from .utils import setup_networking
|
||||
from tests import EXAMPLE_AMI_ID
|
||||
from uuid import uuid4
|
||||
from .utils import setup_networking
|
||||
|
||||
|
||||
@mock_elbv2
|
||||
@mock_autoscaling
|
||||
def test_attach_detach_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")
|
||||
class TestAutoscalignELBv2(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.mocked_networking = setup_networking()
|
||||
self.instance_count = 2
|
||||
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(
|
||||
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"]
|
||||
self.target_name = str(uuid4())[0:6]
|
||||
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",
|
||||
Port=8080,
|
||||
VpcId=self.mocked_networking["vpc"],
|
||||
HealthCheckProtocol="HTTP",
|
||||
HealthCheckPort="8080",
|
||||
HealthCheckPath="/",
|
||||
HealthCheckIntervalSeconds=5,
|
||||
HealthCheckTimeoutSeconds=5,
|
||||
HealthyThresholdCount=5,
|
||||
UnhealthyThresholdCount=2,
|
||||
Matcher={"HttpCode": "200"},
|
||||
)
|
||||
self.target_group_arn = response["TargetGroups"][0]["TargetGroupArn"]
|
||||
|
||||
client.create_launch_configuration(
|
||||
LaunchConfigurationName="test_launch_configuration",
|
||||
ImageId=EXAMPLE_AMI_ID,
|
||||
InstanceType="t2.medium",
|
||||
)
|
||||
self.as_client.create_launch_configuration(
|
||||
LaunchConfigurationName=self.lc_name,
|
||||
ImageId=EXAMPLE_AMI_ID,
|
||||
InstanceType="t2.medium",
|
||||
)
|
||||
self.as_client.create_auto_scaling_group(
|
||||
AutoScalingGroupName=self.asg_name,
|
||||
LaunchConfigurationName=self.lc_name,
|
||||
MinSize=0,
|
||||
MaxSize=self.instance_count,
|
||||
DesiredCapacity=self.instance_count,
|
||||
TargetGroupARNs=[self.target_group_arn],
|
||||
VPCZoneIdentifier=self.mocked_networking["subnet1"],
|
||||
)
|
||||
|
||||
# create asg, attach to target group on create
|
||||
client.create_auto_scaling_group(
|
||||
AutoScalingGroupName="test_asg",
|
||||
LaunchConfigurationName="test_launch_configuration",
|
||||
MinSize=0,
|
||||
MaxSize=INSTANCE_COUNT,
|
||||
DesiredCapacity=INSTANCE_COUNT,
|
||||
TargetGroupARNs=[target_group_arn],
|
||||
VPCZoneIdentifier=mocked_networking["subnet1"],
|
||||
)
|
||||
# create asg without attaching to target group
|
||||
client.create_auto_scaling_group(
|
||||
AutoScalingGroupName="test_asg2",
|
||||
LaunchConfigurationName="test_launch_configuration",
|
||||
MinSize=0,
|
||||
MaxSize=INSTANCE_COUNT,
|
||||
DesiredCapacity=INSTANCE_COUNT,
|
||||
VPCZoneIdentifier=mocked_networking["subnet2"],
|
||||
)
|
||||
response = self.as_client.describe_load_balancer_target_groups(
|
||||
AutoScalingGroupName=self.asg_name
|
||||
)
|
||||
list(response["LoadBalancerTargetGroups"]).should.have.length_of(1)
|
||||
|
||||
response = client.describe_load_balancer_target_groups(
|
||||
AutoScalingGroupName="test_asg"
|
||||
)
|
||||
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
|
||||
)
|
||||
|
||||
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn)
|
||||
list(response["TargetHealthDescriptions"]).should.have.length_of(INSTANCE_COUNT)
|
||||
def test_attach_detach_target_groups(self):
|
||||
# create asg without attaching to target group
|
||||
asg_name2 = str(uuid4())
|
||||
self.as_client.create_auto_scaling_group(
|
||||
AutoScalingGroupName=asg_name2,
|
||||
LaunchConfigurationName=self.lc_name,
|
||||
MinSize=0,
|
||||
MaxSize=self.instance_count,
|
||||
DesiredCapacity=self.instance_count,
|
||||
VPCZoneIdentifier=self.mocked_networking["subnet2"],
|
||||
)
|
||||
|
||||
client.attach_load_balancer_target_groups(
|
||||
AutoScalingGroupName="test_asg2", TargetGroupARNs=[target_group_arn]
|
||||
)
|
||||
self.as_client.attach_load_balancer_target_groups(
|
||||
AutoScalingGroupName=asg_name2, TargetGroupARNs=[self.target_group_arn]
|
||||
)
|
||||
|
||||
response = elbv2_client.describe_target_health(TargetGroupArn=target_group_arn)
|
||||
list(response["TargetHealthDescriptions"]).should.have.length_of(INSTANCE_COUNT * 2)
|
||||
response = self.elbv2_client.describe_target_health(
|
||||
TargetGroupArn=self.target_group_arn
|
||||
)
|
||||
list(response["TargetHealthDescriptions"]).should.have.length_of(
|
||||
self.instance_count * 2
|
||||
)
|
||||
|
||||
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(INSTANCE_COUNT)
|
||||
response = self.as_client.detach_load_balancer_target_groups(
|
||||
AutoScalingGroupName=asg_name2, TargetGroupARNs=[self.target_group_arn]
|
||||
)
|
||||
response = self.elbv2_client.describe_target_health(
|
||||
TargetGroupArn=self.target_group_arn
|
||||
)
|
||||
list(response["TargetHealthDescriptions"]).should.have.length_of(
|
||||
self.instance_count
|
||||
)
|
||||
|
||||
def test_detach_all_target_groups(self):
|
||||
response = self.as_client.detach_load_balancer_target_groups(
|
||||
AutoScalingGroupName=self.asg_name, TargetGroupARNs=[self.target_group_arn]
|
||||
)
|
||||
|
||||
@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(
|
||||
AutoScalingGroupName="test_asg",
|
||||
LaunchConfigurationName="test_launch_configuration",
|
||||
MinSize=0,
|
||||
MaxSize=INSTANCE_COUNT,
|
||||
DesiredCapacity=INSTANCE_COUNT,
|
||||
TargetGroupARNs=[target_group_arn],
|
||||
VPCZoneIdentifier=mocked_networking["subnet1"],
|
||||
)
|
||||
|
||||
response = client.describe_load_balancer_target_groups(
|
||||
AutoScalingGroupName="test_asg"
|
||||
)
|
||||
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)
|
||||
list(response["TargetHealthDescriptions"]).should.have.length_of(0)
|
||||
response = client.describe_load_balancer_target_groups(
|
||||
AutoScalingGroupName="test_asg"
|
||||
)
|
||||
list(response["LoadBalancerTargetGroups"]).should.have.length_of(0)
|
||||
response = self.elbv2_client.describe_target_health(
|
||||
TargetGroupArn=self.target_group_arn
|
||||
)
|
||||
list(response["TargetHealthDescriptions"]).should.have.length_of(0)
|
||||
response = self.as_client.describe_load_balancer_target_groups(
|
||||
AutoScalingGroupName=self.asg_name
|
||||
)
|
||||
list(response["LoadBalancerTargetGroups"]).should.have.length_of(0)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user