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

Add GetAtt support to elbv2 LoadBalancer

Browse Source
This commit is contained in:
Hugo Lopes Tavares 2017-10-27 15:32:16 -04:00
parent 9804d7a963
commit 63c33211ee
2 changed files with 26 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
moto/elbv2/models.py
View File

@ -274,6 +274,12 @@ class FakeLoadBalancer(BaseModel):
load_balancer = elbv2_backend.create_load_balancer(name, security_groups, subnet_ids, scheme=scheme) load_balancer = elbv2_backend.create_load_balancer(name, security_groups, subnet_ids, scheme=scheme)
return load_balancer return load_balancer
def get_cfn_attribute(self, attribute_name):
attributes = {
'DNSName': self.dns_name,
'LoadBalancerName': self.name,
}
return attributes[attribute_name]
class ELBv2Backend(BaseBackend): class ELBv2Backend(BaseBackend):

22
tests/test_cloudformation/test_cloudformation_stack_integration.py
View File

@ -2119,6 +2119,16 @@ def test_stack_spot_fleet():
def test_stack_elbv2_resources_integration(): def test_stack_elbv2_resources_integration():
alb_template = { alb_template = {
"AWSTemplateFormatVersion": "2010-09-09", "AWSTemplateFormatVersion": "2010-09-09",
"Outputs": {
"albdns": {
"Description": "Load balanacer DNS",
"Value": {"Fn::GetAtt": ["alb", "DNSName"]},
},
"albname": {
"Description": "Load balancer name",
"Value": {"Fn::GetAtt": ["alb", "LoadBalancerName"]},
},
},
"Resources": { "Resources": {
"alb": { "alb": {
"Type": "AWS::ElasticLoadBalancingV2::LoadBalancer", "Type": "AWS::ElasticLoadBalancingV2::LoadBalancer",
@ -2207,8 +2217,8 @@ def test_stack_elbv2_resources_integration():
} }
alb_template_json = json.dumps(alb_template) alb_template_json = json.dumps(alb_template)
conn = boto3.client("cloudformation", "us-west-1") cfn_conn = boto3.client("cloudformation", "us-west-1")
conn.create_stack( cfn_conn.create_stack(
StackName="elb_stack", StackName="elb_stack",
TemplateBody=alb_template_json, TemplateBody=alb_template_json,
) )
@ -2246,3 +2256,11 @@ def test_stack_elbv2_resources_integration():
"Type": "forward", "Type": "forward",
"TargetGroupArn": target_groups[0]['TargetGroupArn'] "TargetGroupArn": target_groups[0]['TargetGroupArn']
}]) }])
# test outputs
stacks = cfn_conn.describe_stacks(StackName='elb_stack')['Stacks']
len(stacks).should.equal(1)
stacks[0]['Outputs'].should.equal([
{'OutputKey': 'albdns', 'OutputValue': load_balancers[0]['DNSName']},
{'OutputKey': 'albname', 'OutputValue': load_balancers[0]['LoadBalancerName']},
])