From 6017a481cf2e05b0b233e4b80752196420f51367 Mon Sep 17 00:00:00 2001 From: Chris Henry Date: Wed, 2 Dec 2015 17:46:24 -0500 Subject: [PATCH] Initial implementation of elb scheme --- moto/elb/models.py | 9 ++++++--- moto/elb/responses.py | 7 +++---- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/moto/elb/models.py b/moto/elb/models.py index b10d53cf4..7e64b2210 100644 --- a/moto/elb/models.py +++ b/moto/elb/models.py @@ -44,13 +44,14 @@ class FakeBackend(object): class FakeLoadBalancer(object): - def __init__(self, name, zones, ports): + def __init__(self, name, zones, ports, scheme='internet-facing',): self.name = name self.health_check = None self.instance_ids = [] self.zones = zones self.listeners = [] self.backends = [] + self.scheme = scheme self.attributes = FakeLoadBalancer.get_default_attributes() self.policies = Policies() self.policies.other_policies = [] @@ -135,8 +136,10 @@ class ELBBackend(BaseBackend): def __init__(self): self.load_balancers = {} - def create_load_balancer(self, name, zones, ports): - new_load_balancer = FakeLoadBalancer(name=name, zones=zones, ports=ports) + def create_load_balancer(self, name, zones, ports, scheme='internet-facing'): + print scheme + + new_load_balancer = FakeLoadBalancer(name=name, zones=zones, ports=ports, scheme=scheme) self.load_balancers[name] = new_load_balancer return new_load_balancer diff --git a/moto/elb/responses.py b/moto/elb/responses.py index 75f48d326..a0187f160 100644 --- a/moto/elb/responses.py +++ b/moto/elb/responses.py @@ -21,17 +21,16 @@ class ELBResponse(BaseResponse): return elb_backends[self.region] def create_load_balancer(self): - """ - u'Scheme': [u'internet-facing'], - """ load_balancer_name = self._get_param('LoadBalancerName') availability_zones = self._get_multi_param("AvailabilityZones.member") ports = self._get_list_prefix("Listeners.member") + scheme = self._get_param('Scheme') self.elb_backend.create_load_balancer( name=load_balancer_name, zones=availability_zones, ports=ports, + scheme=scheme ) template = self.response_template(CREATE_LOAD_BALANCER_TEMPLATE) return template.render() @@ -325,7 +324,7 @@ DESCRIBE_LOAD_BALANCERS_TEMPLATE = """