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

Allow ELB policies to be set via Cloudformation. (#809)

Browse Source 
Also more closely mirror AWS by not returning ELB
BackendServerDescriptions that don't have policies.

Signed-off-by: Michael Nussbaum <michael.nussbaum@getbraintree.com>
This commit is contained in:
Michael Nussbaum 2017-01-18 20:02:04 -08:00 committed by Steve Pulec
parent e1260bca06
commit 51129f6ef3
2 changed files with 29 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
moto/elb/models.py
View File

@ -8,7 +8,10 @@ from boto.ec2.elb.attributes import (
AccessLogAttribute,
CrossZoneLoadBalancingAttribute,
)
from boto.ec2.elb.policies import Policies
from boto.ec2.elb.policies import (
Policies,
OtherPolicy,
)
from moto.core import BaseBackend
from moto.ec2.models import ec2_backends
from .exceptions import (
@ -104,6 +107,21 @@ class FakeLoadBalancer(object):
for instance_id in instance_ids:
elb_backend.register_instances(new_elb.name, [instance_id])
policies = properties.get('Policies', [])
port_policies = {}
for policy in policies:
policy_name = policy["PolicyName"]
other_policy = OtherPolicy()
other_policy.policy_name = policy_name
elb_backend.create_lb_other_policy(new_elb.name, other_policy)
for port in policy.get("InstancePorts", []):
policies_for_port = port_policies.get(port, set())
policies_for_port.add(policy_name)
port_policies[port] = policies_for_port
for port, policies in port_policies.items():
elb_backend.set_load_balancer_policies_of_backend_server(new_elb.name, port, list(policies))
health_check = properties.get('HealthCheck')
if health_check:
elb_backend.configure_health_check(
@ -307,7 +325,9 @@ class ELBBackend(BaseBackend):
def create_lb_other_policy(self, load_balancer_name, other_policy):
load_balancer = self.get_load_balancer(load_balancer_name)
load_balancer.policies.other_policies.append(other_policy)
if other_policy.policy_name not in [p.policy_name for p in load_balancer.policies.other_policies]:
load_balancer.policies.other_policies.append(other_policy)
return load_balancer
def create_app_cookie_stickiness_policy(self, load_balancer_name, policy):

16
moto/elb/responses.py
View File

@ -445,18 +445,16 @@ DESCRIBE_LOAD_BALANCERS_TEMPLATE = """<DescribeLoadBalancersResponse xmlns="http
<DNSName>{{ load_balancer.dns_name }}</DNSName>
<BackendServerDescriptions>
{% for backend in load_balancer.backends %}
{% if backend.policy_names %}
<member>
{% if backend.instance_port %}
<InstancePort>{{ backend.instance_port }}</InstancePort>
{% endif %}
{% if backend.policy_names %}
<PolicyNames>
{% for policy in backend.policy_names %}
<member>{{ policy }}</member>
{% endfor %}
</PolicyNames>
{% endif %}
<PolicyNames>
{% for policy in backend.policy_names %}
<member>{{ policy }}</member>
{% endfor %}
</PolicyNames>
</member>
{% endif %}
{% endfor %}
</BackendServerDescriptions>
<Subnets>