add validation for target name
This commit is contained in:
parent
d4c6111c40
commit
775b8a953a
@ -170,7 +170,7 @@ class DuplicatePriorityError(ELBClientError):
|
|||||||
|
|
||||||
class InvalidTargetGroupNameError(ELBClientError):
|
class InvalidTargetGroupNameError(ELBClientError):
|
||||||
|
|
||||||
def __init__(self, invalid_name):
|
def __init__(self, msg):
|
||||||
super(InvalidTargetGroupNameError, self).__init__(
|
super(InvalidTargetGroupNameError, self).__init__(
|
||||||
"ValidationError",
|
"ValidationError", msg
|
||||||
"Target group name '%s' cannot be longer than '32' characters" % invalid_name)
|
)
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
from __future__ import unicode_literals
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
import datetime
|
import datetime
|
||||||
|
import re
|
||||||
from moto.compat import OrderedDict
|
from moto.compat import OrderedDict
|
||||||
from moto.core import BaseBackend, BaseModel
|
from moto.core import BaseBackend, BaseModel
|
||||||
from moto.ec2.models import ec2_backends
|
from moto.ec2.models import ec2_backends
|
||||||
@ -266,7 +267,24 @@ class ELBv2Backend(BaseBackend):
|
|||||||
|
|
||||||
def create_target_group(self, name, **kwargs):
|
def create_target_group(self, name, **kwargs):
|
||||||
if len(name) > 32:
|
if len(name) > 32:
|
||||||
raise InvalidTargetGroupNameError(name)
|
raise InvalidTargetGroupNameError(
|
||||||
|
"Target group name '%s' cannot be longer than '32' characters" % name
|
||||||
|
)
|
||||||
|
if not re.match('^[a-zA-Z0-9\-]+$', name):
|
||||||
|
raise InvalidTargetGroupNameError(
|
||||||
|
"Target group name '%s' can only contain characters that are alphanumeric characters or hyphens(-)" % name
|
||||||
|
)
|
||||||
|
|
||||||
|
# undocumented validation
|
||||||
|
if not re.match('(?!.*--)(?!^-)(?!.*-$)^[A-Za-z0-9-]+$', name):
|
||||||
|
raise InvalidTargetGroupNameError(
|
||||||
|
"1 validation error detected: Value '%s' at 'targetGroup.targetGroupArn.targetGroupName' failed to satisfy constraint: Member must satisfy regular expression pattern: (?!.*--)(?!^-)(?!.*-$)^[A-Za-z0-9-]+$" % name
|
||||||
|
)
|
||||||
|
|
||||||
|
if name.startswith('-') or name.endswith('-'):
|
||||||
|
raise InvalidTargetGroupNameError(
|
||||||
|
"Target group name '%s' cannot begin or end with '-'" % name
|
||||||
|
)
|
||||||
for target_group in self.target_groups.values():
|
for target_group in self.target_groups.values():
|
||||||
if target_group.name == name:
|
if target_group.name == name:
|
||||||
raise DuplicateTargetGroupName()
|
raise DuplicateTargetGroupName()
|
||||||
|
@ -327,6 +327,15 @@ def test_create_target_group_and_listeners():
|
|||||||
response = conn.describe_target_groups()
|
response = conn.describe_target_groups()
|
||||||
response.get('TargetGroups').should.have.length_of(0)
|
response.get('TargetGroups').should.have.length_of(0)
|
||||||
|
|
||||||
|
|
||||||
|
@mock_elbv2
|
||||||
|
@mock_ec2
|
||||||
|
def test_create_invalid_target_group():
|
||||||
|
conn = boto3.client('elbv2', region_name='us-east-1')
|
||||||
|
ec2 = boto3.resource('ec2', region_name='us-east-1')
|
||||||
|
|
||||||
|
vpc = ec2.create_vpc(CidrBlock='172.28.7.0/24', InstanceTenancy='default')
|
||||||
|
|
||||||
# Fail to create target group with name which length is 33
|
# Fail to create target group with name which length is 33
|
||||||
long_name = 'A' * 33
|
long_name = 'A' * 33
|
||||||
with assert_raises(ClientError):
|
with assert_raises(ClientError):
|
||||||
@ -344,6 +353,39 @@ def test_create_target_group_and_listeners():
|
|||||||
UnhealthyThresholdCount=2,
|
UnhealthyThresholdCount=2,
|
||||||
Matcher={'HttpCode': '200'})
|
Matcher={'HttpCode': '200'})
|
||||||
|
|
||||||
|
invalid_names = ['-name', 'name-', '-name-', 'example.com', 'test@test', 'Na--me']
|
||||||
|
for name in invalid_names:
|
||||||
|
with assert_raises(ClientError):
|
||||||
|
conn.create_target_group(
|
||||||
|
Name=name,
|
||||||
|
Protocol='HTTP',
|
||||||
|
Port=8080,
|
||||||
|
VpcId=vpc.id,
|
||||||
|
HealthCheckProtocol='HTTP',
|
||||||
|
HealthCheckPort='8080',
|
||||||
|
HealthCheckPath='/',
|
||||||
|
HealthCheckIntervalSeconds=5,
|
||||||
|
HealthCheckTimeoutSeconds=5,
|
||||||
|
HealthyThresholdCount=5,
|
||||||
|
UnhealthyThresholdCount=2,
|
||||||
|
Matcher={'HttpCode': '200'})
|
||||||
|
|
||||||
|
valid_names = ['name', 'Name', '000']
|
||||||
|
for name in valid_names:
|
||||||
|
conn.create_target_group(
|
||||||
|
Name=name,
|
||||||
|
Protocol='HTTP',
|
||||||
|
Port=8080,
|
||||||
|
VpcId=vpc.id,
|
||||||
|
HealthCheckProtocol='HTTP',
|
||||||
|
HealthCheckPort='8080',
|
||||||
|
HealthCheckPath='/',
|
||||||
|
HealthCheckIntervalSeconds=5,
|
||||||
|
HealthCheckTimeoutSeconds=5,
|
||||||
|
HealthyThresholdCount=5,
|
||||||
|
UnhealthyThresholdCount=2,
|
||||||
|
Matcher={'HttpCode': '200'})
|
||||||
|
|
||||||
|
|
||||||
@mock_elbv2
|
@mock_elbv2
|
||||||
@mock_ec2
|
@mock_ec2
|
||||||
|
Loading…
x
Reference in New Issue
Block a user