awsborn 0.7.0 → 0.8.0

data/lib/awsborn/server.rb

@@ -1,6 +1,7 @@
 module Awsborn
   class Server
-    
+    include Awsborn::AwsConstants
+
     def initialize (name, options = {})
       @name = name
       @options = options.dup
@@ -106,8 +107,8 @@ module Awsborn
 
     def launch_instance (key_pair)
       @launch_response = ec2.launch_instance(image_id,
-        :instance_type => constant(instance_type),
-        :availability_zone => constant(zone),
+        :instance_type => symbol_to_aws_instance_type(instance_type),
+        :availability_zone => symbol_to_aws_zone(zone),
         :key_name => key_pair.name,
         :group_ids => security_group,
         :monitoring_enabled => monitor
@@ -233,7 +234,7 @@ module Awsborn
         tmp.is_a?(String) ? tmp : tmp[architecture]
       end
       def architecture
-        string = constant(instance_type)
+        string = symbol_to_aws_instance_type(instance_type)
         case
         when INSTANCE_TYPES_32_BIT.include?(string) then :i386
         when INSTANCE_TYPES_64_BIT.include?(string) then :x64
@@ -296,22 +297,6 @@ module Awsborn
       end
     end
 
-    AVAILABILITY_ZONES = %w[
-      us-east-1a us-east-1b us-east-1c us-east-1d
-      us-west-1a us-west-1b
-      eu-west-1a eu-west-1b
-      ap-southeast-1a ap-southeast-1b
-      ap-northeast-1a ap-northeast-1b
-    ]
-    INSTANCE_TYPES_32_BIT = %w[m1.small c1.medium t1.micro]
-    INSTANCE_TYPES_64_BIT = %w[m1.large m1.xlarge m2.xlarge m2.2xlarge m2.4xlarge c1.xlarge cc1.4xlarge t1.micro]
-    INSTANCE_TYPES = (INSTANCE_TYPES_32_BIT + INSTANCE_TYPES_64_BIT).uniq
-    SYMBOL_CONSTANT_MAP = (AVAILABILITY_ZONES + INSTANCE_TYPES).inject({}) { |memo,str| memo[str.tr('-.','_').to_sym] = str; memo }
-
-    def constant (symbol)
-      SYMBOL_CONSTANT_MAP[symbol]
-    end
-
     def logger
       @logger ||= self.class.logger
     end

data/lib/awsborn/server_cluster.rb

@@ -34,14 +34,19 @@ module Awsborn
       @klass = klass
       @name = name.to_s
       @instances = []
+      @load_balancers = []
       self.class.clusters << self
     end
-    
+
     def domain (*args)
       @domain = args.first unless args.empty?
       @domain
     end
-    
+
+    def load_balancer (name, options={})
+      @load_balancers << Awsborn::LoadBalancer.new(name, options)
+    end
+
     def server (name, options = {})
       options = add_domain_to_ip(options)
       instance = @klass.new name, options
@@ -53,6 +58,7 @@ module Awsborn
       running, missing = requested.partition { |e| e.running? }
       refresh_running(running) if running.any?
       start_missing_instances(missing) if missing.any?
+      update_load_balancing(running) unless @load_balancers.empty?
     end
 
     def refresh_running (instances)
@@ -65,6 +71,12 @@ module Awsborn
       delete_key_pair(instances)
     end
 
+    def update_load_balancing(running)
+      @load_balancers.each do |lb|
+        lb.update_with(running).inspect
+      end
+    end
+
     def generate_key_pair (instances)
       @key_pair = instances.first.ec2.generate_key_pair
     end

data/spec/aws_constants_spec.rb

@@ -0,0 +1,82 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe Awsborn::AwsConstants do
+  include Awsborn::AwsConstants
+  describe "endpoint_for_zone_and_service" do
+    it "should have endpoints for each service and for zones in five regions" do
+      endpoint_for_zone_and_service(:eu_west_1a,  :ec2).should == 'https://eu-west-1.ec2.amazonaws.com'
+      endpoint_for_zone_and_service("eu_west_1b", :ec2).should == 'https://eu-west-1.ec2.amazonaws.com'
+      endpoint_for_zone_and_service(:us_west_1b,  :ec2).should == 'https://us-west-1.ec2.amazonaws.com'
+      endpoint_for_zone_and_service(:us_east_1b,  :ec2).should == 'https://us-east-1.ec2.amazonaws.com'
+      endpoint_for_zone_and_service(:eu_west_1a,  :elb).should == 'https://eu-west-1.elasticloadbalancing.amazonaws.com'
+      endpoint_for_zone_and_service("eu_west_1b", :elb).should == 'https://eu-west-1.elasticloadbalancing.amazonaws.com'
+      endpoint_for_zone_and_service(:us_west_1b,  :elb).should == 'https://us-west-1.elasticloadbalancing.amazonaws.com'
+      endpoint_for_zone_and_service(:us_east_1b,  :elb).should == 'https://us-east-1.elasticloadbalancing.amazonaws.com'
+    end
+  end
+
+  describe "zone_to_awz_region" do
+    it "accepts a zone symbol and returns its region" do
+      zone_to_awz_region(:eu_west_1a).should == 'eu-west-1'
+    end
+    it "accepts an aws zone symbol and returns its region" do
+      zone_to_awz_region('eu-west-1a').should == 'eu-west-1'
+    end
+    it "raise an error if no region found" do
+      expect{zone_to_awz_region('santa-northpole-2b')}.to raise_error(Awsborn::UnknownConstantError)
+    end
+    it "returns a region even when a region is given" do
+      zone_to_awz_region('eu-west-1').should == 'eu-west-1'
+      zone_to_awz_region(:eu_west_1).should == 'eu-west-1'
+    end
+  end
+
+  describe "symbol_to_aws_zone" do
+    it "accepts a zone symbol and returns its aws zone" do
+      symbol_to_aws_zone(:eu_west_1a).should == 'eu-west-1a'
+    end
+    it "raise an error if no matching aws zone found" do
+      expect{symbol_to_aws_zone(:santa_northpole_2b)}.to raise_error(Awsborn::UnknownConstantError)
+    end
+  end
+
+  describe "aws_zone_to_symbol" do
+    it "returns a symbol from aws zone" do
+      aws_zone_to_symbol('eu-west-1a').should == :eu_west_1a
+    end
+    it "raises an error if the zone is unknown" do
+      expect{aws_zone_to_symbol('santa-northpole-2a')}.to raise_error(Awsborn::UnknownConstantError)
+    end
+  end
+
+  describe "symbol_to_aws_instance_type" do
+    it "accepts an instance type  symbol and returns its aws instance type" do
+      symbol_to_aws_instance_type(:m1_small).should == 'm1.small'
+    end
+    it "raise an error if no matching aws zone found" do
+      expect{symbol_to_aws_instance_type(:xx_megalarge)}.to raise_error(Awsborn::UnknownConstantError)
+    end
+  end
+
+  describe "aws_instance_type_to_symbol" do
+    it "returns a symbol from aws instance type" do
+      aws_instance_type_to_symbol('m1.small').should == :m1_small
+    end
+    it "raises an error if the instance type is unknown" do
+      expect{aws_instance_type_to_symbol('xx.megalarge')}.to raise_error(Awsborn::UnknownConstantError)
+    end
+  end
+
+  describe "awz_constant" do
+    it "should look up an availability zone" do
+      awz_constant(:eu_west_1a).should == "eu-west-1a"
+    end
+    it "should look up an instance type" do
+      awz_constant(:m1_large).should == "m1.large"
+    end
+    it "should raise an error if the symbol is unknown" do
+      expect{awz_constant(:unknown_constant)}.to raise_error(Awsborn::UnknownConstantError)
+    end
+  end
+
+end

data/spec/ec2_spec.rb

@@ -1,12 +1,4 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
 describe Awsborn::Ec2 do
-  context ".endpoint_for_zone" do
-    it "should have endpoints for zones in five regions" do
-      Awsborn::Ec2.endpoint_for_zone(:eu_west_1a).should == 'https://eu-west-1.ec2.amazonaws.com'
-      Awsborn::Ec2.endpoint_for_zone("eu_west_1b").should == 'https://eu-west-1.ec2.amazonaws.com'
-      Awsborn::Ec2.endpoint_for_zone(:us_west_1b).should == 'https://us-west-1.ec2.amazonaws.com'
-      Awsborn::Ec2.endpoint_for_zone(:us_east_1b).should == 'https://us-east-1.ec2.amazonaws.com'
-    end
-  end
 end

data/spec/elb_spec.rb

@@ -0,0 +1,256 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe Awsborn::Elb do
+  before do
+    @mock_interface = mock(:elb_interface)
+    RightAws::ElbInterface.stub!(:new).and_return(@mock_interface)
+    Awsborn.stub!(:access_key_id).and_return('access_key_id')
+    Awsborn.stub!(:secret_access_key).and_return('secret_access_key')
+  end
+  describe "initialize" do
+    it "sets a proper endpoint" do
+      [:eu_west_1, 'eu-west-1', :eu_west_1a, "eu_west_1b"].each do |zone|
+        elb = Awsborn::Elb.new(:eu_west_1)
+        elb.region.should == 'eu-west-1'
+        elb.endpoint.should == 'https://eu-west-1.elasticloadbalancing.amazonaws.com'
+      end
+    end
+  end
+
+  describe "connection" do
+    before do
+      @elb = Awsborn::Elb.new(:eu_west_1b)
+    end
+    it "setup a connection to a proper endpoint the first time" do
+      RightAws::ElbInterface.should_receive(:new).
+                             with('access_key_id', 'secret_access_key', :logger => Awsborn.logger).
+                             exactly(:once)
+      @elb.connection
+      @elb.connection
+    end
+  end
+
+  context "for a valid elb" do
+    before do
+      @elb = Awsborn::Elb.new(:eu_west_1b)
+    end
+
+    describe "describe_load_balancer" do
+      it "forwards to ElbInterface" do
+        @mock_interface.should_receive(:describe_load_balancers).with('some-name').and_return([:description])
+        @elb.describe_load_balancer('some-name').should == :description
+      end
+    end
+
+    describe "running?" do
+      it "returns true if load balancer is running" do
+        @mock_interface.should_receive(:describe_load_balancers).with('some-name').and_return([:description])
+        @elb.running?('some-name').should be_true
+      end
+      it "returns true if load balancer is not running" do
+        @mock_interface.should_receive(:describe_load_balancers).with('some-name').and_raise(RightAws::AwsError)
+        @elb.running?('some-name').should be_false
+      end
+    end
+
+    describe "dns_name" do
+      it "extracts name from description" do
+        @mock_interface.should_receive(:describe_load_balancers).with('some-name').and_return([{:dns_name => 'dns-name'}])
+        @elb.dns_name('some-name').should == 'dns-name'
+      end
+    end
+
+    describe "instances" do
+      it "extracts instances from description" do
+        @mock_interface.should_receive(:describe_load_balancers).with('some-name').and_return([{:instances => 'instances'}])
+        @elb.instances('some-name').should == 'instances'
+      end
+    end
+
+    describe "zones" do
+      it "extracts zones from description" do
+        @mock_interface.should_receive(:describe_load_balancers).with('some-name').and_return([{:availability_zones => 'zones'}])
+        @elb.zones('some-name').should == 'zones'
+      end
+    end
+
+    describe "create_load_balancer" do
+      it "forwards to ElbInterface with a temporary zone and no listeners" do
+        @mock_interface.should_receive(:create_load_balancer).with('some-name', ['eu-west-1a'], [])
+        @elb.create_load_balancer('some-name')
+      end
+    end
+
+    describe "set_load_balancer_listeners" do
+      it "sets listeners if none where set before" do
+        description = {
+          :listeners => []
+        }
+        new_listeners = [
+          { :protocol => :http, :load_balancer_port => 80, :instance_port => 80 },
+        ]
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_not_receive(:delete_load_balancer_listeners)
+        @mock_interface.should_receive(:create_load_balancer_listeners).with('some-name', new_listeners)
+        @elb.set_load_balancer_listeners('some-name', new_listeners)
+      end
+      it "updates listeners if some where set before" do
+        description = {
+          :listeners => [
+            { :protocol => :http, :load_balancer_port => 11, :instance_port => 11 },
+            { :protocol => :http, :load_balancer_port => 22, :instance_port => 22 },
+            { :protocol => :tcp,  :load_balancer_port => 33, :instance_port => 33 }
+          ]
+        }
+        new_listeners = [
+          { :protocol => :http, :load_balancer_port => 80, :instance_port => 80 },
+        ]
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_receive(:delete_load_balancer_listeners).with('some-name', 11, 22, 33)
+        @mock_interface.should_receive(:create_load_balancer_listeners).with('some-name', new_listeners)
+        @elb.set_load_balancer_listeners('some-name', new_listeners)
+      end
+    end
+
+    describe "set_lb_cookie_policy" do
+      it "creates and sets sticky policy for each given port when the policy is missing" do
+        description = {
+          :listeners => [
+            { :protocol => :http, :load_balancer_port => 11, :instance_port => 11 },
+            { :protocol => :http, :load_balancer_port => 22, :instance_port => 22 },
+            { :protocol => :tcp,  :load_balancer_port => 33, :instance_port => 33 }
+          ],
+          :lb_cookie_stickiness_policies => []
+        }
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_receive(:create_lb_cookie_stickiness_policy).
+                        with('some-name', 'lb-some-name-300', 300)
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 11, 'lb-some-name-300')
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 22, 'lb-some-name-300')
+        @elb.set_lb_cookie_policy('some-name', [11, 22], 300)
+      end
+      it "does not create but sets sticky policy for each given port when the policy exists" do
+        description = {
+          :listeners => [
+            { :protocol => :http, :load_balancer_port => 11, :instance_port => 11 },
+            { :protocol => :http, :load_balancer_port => 22, :instance_port => 22 },
+            { :protocol => :tcp,  :load_balancer_port => 33, :instance_port => 33 }
+          ],
+          :lb_cookie_stickiness_policies => [
+            { :policy_name => 'lb-some-name-300', :expiration_period => 42 },
+          ]
+        }
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_not_receive(:create_lb_cookie_stickiness_policy)
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 11, 'lb-some-name-300')
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 22, 'lb-some-name-300')
+        @elb.set_lb_cookie_policy('some-name', [11, 22], 300)
+      end
+    end
+
+    describe "set_app_cookie_policy" do
+      it "creates and sets sticky policy for each given port when the policy is missing" do
+        description = {
+          :listeners => [
+            { :protocol => :http, :load_balancer_port => 11, :instance_port => 11 },
+            { :protocol => :http, :load_balancer_port => 22, :instance_port => 22 },
+            { :protocol => :tcp,  :load_balancer_port => 33, :instance_port => 33 }
+          ],
+          :app_cookie_stickiness_policies => []
+        }
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_receive(:create_app_cookie_stickiness_policy).
+                        with('some-name', 'app-some-name--some-cookie', "_some_cookie")
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 11, 'app-some-name--some-cookie')
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 22, 'app-some-name--some-cookie')
+        @elb.set_app_cookie_policy('some-name', [11, 22], "_some_cookie")
+      end
+      it "does not create but sets sticky policy for each given port when the policy exists" do
+        description = {
+          :listeners => [
+            { :protocol => :http, :load_balancer_port => 11, :instance_port => 11 },
+            { :protocol => :http, :load_balancer_port => 22, :instance_port => 22 },
+            { :protocol => :tcp,  :load_balancer_port => 33, :instance_port => 33 }
+          ],
+          :app_cookie_stickiness_policies => [
+            { :policy_name => 'app-some-name--some-cookie', :expiration_period => 42 },
+          ]
+        }
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_not_receive(:create_app_cookie_stickiness_policy)
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 11, 'app-some-name--some-cookie')
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).
+                        with('some-name', 22, 'app-some-name--some-cookie')
+        @elb.set_app_cookie_policy('some-name', [11, 22], "_some_cookie")
+      end
+    end
+
+    describe "remove_all_cookie_policies" do
+      it "removes all policies from listeners" do
+        description = {
+          :listeners => [
+            { :protocol => :http, :load_balancer_port => 11, :instance_port => 11 },
+            { :protocol => :http, :load_balancer_port => 22, :instance_port => 22 },
+            { :protocol => :tcp,  :load_balancer_port => 33, :instance_port => 33 }
+          ],
+          :app_cookie_stickiness_policies => []
+        }
+        @mock_interface.should_receive(:describe_load_balancers).
+                        with('some-name').
+                        and_return([description])
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).with('some-name', 11)
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).with('some-name', 22)
+        @mock_interface.should_receive(:set_load_balancer_policies_of_listener).with('some-name', 33)
+        @elb.remove_all_cookie_policies('some-name')
+      end
+    end
+
+    describe "register_instances" do
+      it "forwards to ElbInterface" do
+        @mock_interface.should_receive(:register_instances_with_load_balancer).with('some-name', 'i-00000001', 'i-00000002')
+        @elb.register_instances('some-name', ['i-00000001', 'i-00000002'])
+      end
+    end
+
+    describe "deregister_instances" do
+      it "forwards to ElbInterface" do
+        @mock_interface.should_receive(:deregister_instances_with_load_balancer).with('some-name', 'i-00000001', 'i-00000002')
+        @elb.deregister_instances('some-name', ['i-00000001', 'i-00000002'])
+      end
+    end
+
+    describe "enable_zones" do
+      it "forwards to ElbInterface" do
+        @mock_interface.should_receive(:enable_availability_zones_for_load_balancer).with('some-name', 'eu-west-1a', 'eu-west-1b')
+        @elb.enable_zones('some-name', ['eu-west-1a', 'eu-west-1b'])
+      end
+    end
+
+    describe "disable_zones" do
+      it "forwards to ElbInterface" do
+        @mock_interface.should_receive(:disable_availability_zones_for_load_balancer).with('some-name', 'eu-west-1a', 'eu-west-1b')
+        @elb.disable_zones('some-name', ['eu-west-1a', 'eu-west-1b'])
+      end
+    end
+  end
+end
+

data/spec/load_balancer_spec.rb

@@ -0,0 +1,288 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe Awsborn::LoadBalancer do
+  before do
+    @mocked_elb = mock(:elb,
+                       :running? => false,
+                       :remove_all_cookie_policies => true,
+                       :create_load_balancer => {})
+    @listener_fixture = [ { :protocol => :tcp, :load_balancer_port => 123, :instance_port => 123} ]
+    @cookies_fixture = [ { :ports => [123], :policy => :disabled } ]
+    Awsborn::Elb.stub!(:new).and_return(@mocked_elb)
+  end
+  describe "initialize" do
+    it "requires a valid region option" do
+      expect { Awsborn::LoadBalancer.new('some-name') }.to raise_error
+      expect { Awsborn::LoadBalancer.new('some-name', :region => :blabla) }.to raise_error
+    end
+    it "launches the load balancer if not running" do
+      @mocked_elb.should_receive(:running?).with('some-name').and_return(false)
+      @mocked_elb.should_receive(:create_load_balancer).with('some-name').and_return(nil)
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+    end
+    it "does not launch the load balancer if running" do
+      @mocked_elb.should_receive(:running?).with('some-name').and_return(true)
+      @mocked_elb.should_not_receive(:create_load_balancer)
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+    end
+    describe "sets all attributes properly" do
+      subject do
+        @balancer = Awsborn::LoadBalancer.new(
+          'some-name',
+          :region => :eu_west_1a,
+          :only => [:server1, :server2],
+          :except => [:server2],
+          :listeners => @listener_fixture,
+          :sticky_cookies => @cookies_fixture
+        )
+      end
+      its(:name)           { should == 'some-name' }
+      its(:region)         { should == 'eu-west-1' }
+      its(:only)           { should == [:server1, :server2] }
+      its(:except)         { should == [:server2] }
+      its(:listeners)      { should == @listener_fixture }
+      its(:sticky_cookies) { should == @cookies_fixture }
+    end
+    describe "sets proper default values" do
+      subject do
+        @balancer = Awsborn::LoadBalancer.new(
+          'some-name',
+          :region => :eu_west_1
+        )
+      end
+      its(:only)           { should == [] }
+      its(:except)         { should == [] }
+      its(:listeners)      { should == Awsborn::LoadBalancer::DEFAULT_LISTENERS }
+      its(:sticky_cookies) { should == [] }
+    end
+  end
+
+  describe "dns_name" do
+    it "delegates to elb" do
+      @mocked_elb.should_receive(:dns_name).with('some-name').and_return('dns-name')
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      ).dns_name.should == 'dns-name'
+    end
+  end
+
+
+  describe "instances" do
+    it "delegates to elb" do
+      @mocked_elb.should_receive(:instances).with('some-name').and_return('instances')
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      ).instances.should == 'instances'
+    end
+  end
+
+  describe "instances=" do
+    it "sets instances properly when the load balancer has no previous instances" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+      @balancer.stub!(:instances).and_return([])
+      @mocked_elb.should_receive(:register_instances).with('some-name', ['i-00000001', 'i-00000002'])
+      @mocked_elb.should_not_receive(:deregister_instances)
+      @balancer.instances = ['i-00000001', 'i-00000002']
+    end
+    it "does nothing if the previous instances are the same as the ones to be set" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+      @balancer.stub!(:instances).and_return(['i-00000001', 'i-00000002'])
+      @mocked_elb.should_not_receive(:register_instances)
+      @mocked_elb.should_not_receive(:deregister_instances)
+      @balancer.instances = ['i-00000001', 'i-00000002']
+    end
+    it "removes and adds instances accorind to previous and new state" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+      @balancer.stub!(:instances).and_return(['i-00000001', 'i-00000002'])
+      @mocked_elb.should_receive(:register_instances).with('some-name', ['i-00000003'])
+      @mocked_elb.should_receive(:deregister_instances).with('some-name', ['i-00000001'])
+      @balancer.instances = ['i-00000002', 'i-00000003']
+    end
+  end
+
+  describe "zones" do
+    it "delegates to elb" do
+      @mocked_elb.should_receive(:zones).with('some-name').and_return('zones')
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      ).zones.should == 'zones'
+    end
+  end
+
+  describe "zones=" do
+    it "sets zones properly when the load balancer has no previous zones" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+      @balancer.stub!(:zones).and_return([])
+      @mocked_elb.should_receive(:enable_zones).with('some-name', ['eu-west-1a', 'eu-west-1b'])
+      @mocked_elb.should_not_receive(:disable_zones)
+      @balancer.zones = ['eu-west-1a', 'eu-west-1b']
+    end
+    it "does nothing if the previous zones are the same as the ones to be set" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+      @balancer.stub!(:zones).and_return(['eu-west-1a', 'eu-west-1b'])
+      @mocked_elb.should_not_receive(:enable_zones)
+      @mocked_elb.should_not_receive(:disable_zones)
+      @balancer.zones = ['eu-west-1a', 'eu-west-1b']
+    end
+    it "removes and adds zones accordind to previous and new state" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      )
+      @balancer.stub!(:zones).and_return(['eu-west-1a', 'eu-west-1b'])
+      @mocked_elb.should_receive(:enable_zones).with('some-name', ['eu-west-1c'])
+      @mocked_elb.should_receive(:disable_zones).with('some-name', ['eu-west-1a'])
+      @balancer.zones = ['eu-west-1b', 'eu-west-1c']
+    end
+  end
+
+  describe "description" do
+    it "delegates to elb" do
+      @mocked_elb.should_receive(:describe_load_balancer).with('some-name').and_return('description')
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      ).description.should == 'description'
+    end
+  end
+
+  describe "launch" do
+    it "delegates to elb" do
+      @mocked_elb.should_receive(:create_load_balancer).with('some-name')
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1
+      ).launch
+    end
+  end
+
+  describe "update_listeners" do
+    it "delegates to elb" do
+      @mocked_elb.should_receive('set_load_balancer_listeners').with('some-name', @listener_fixture)
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1,
+        :listeners => @listener_fixture
+      ).update_listeners
+    end
+  end
+
+  describe "update_sticky_cookies" do
+    it "sets proper policies" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1,
+        :sticky_cookies => [
+          { :ports => [11,22], :policy => :disabled },
+          { :ports => [33], :policy => :app_generated, :cookie_name => 'some_cookie' },
+          { :ports => [44], :policy => :lb_generated, :expiration_period => 42 }
+        ]
+      )
+      @mocked_elb.should_receive(:remove_all_cookie_policies).with('some-name')
+      @mocked_elb.should_receive(:set_app_sticky_cookie).with('some-name', [33], 'some_cookie')
+      @mocked_elb.should_receive(:set_lb_sticky_cookie).with('some-name', [44], 42)
+
+      @balancer.update_sticky_cookies
+    end
+
+    it "raises an error if a policy is missing a ports option" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1,
+        :sticky_cookies => [
+          { :policy => :disabled },
+        ]
+      )
+      expect { @balancer.update_sticky_cookies }.to raise_error
+    end
+    it "raises an error if an app sticky policy is missing a cookie name" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1,
+        :sticky_cookies => [
+          { :ports => [33], :policy => :app_generated },
+        ]
+      )
+      expect { @balancer.update_sticky_cookies }.to raise_error
+    end
+    it "raises an error if an lb sticky policy is missing an expiration period" do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1,
+        :sticky_cookies => [
+          { :ports => [44], :policy => :lb_generated }
+        ]
+      )
+      expect { @balancer.update_sticky_cookies }.to raise_error
+    end
+  end
+
+  describe "update_with" do
+    before do
+      @balancer = Awsborn::LoadBalancer.new(
+        'some-name',
+        :region => :eu_west_1a,
+        :listeners => @listener_fixture,
+        :sticky_cookies => @cookies_fixture
+      )
+      @new_servers = [
+        mock(:server1, :name => :server1, :instance_id => 'i-00000001', :zone => :eu_west_1a),
+        mock(:server1, :name => :server2, :instance_id => 'i-00000002', :zone => :eu_west_1b)
+      ]
+    end
+    it "sets new instances, sets new zones, updates listeners and updates sticky cookies" do
+      @balancer.should_receive(:instances=).with(['i-00000001', 'i-00000002'])
+      @balancer.should_receive(:zones=).with(['eu-west-1a', 'eu-west-1b'])
+      @balancer.should_receive(:update_listeners)
+      @balancer.should_receive(:update_sticky_cookies)
+      @balancer.should_receive(:description).and_return('description')
+      @balancer.update_with(@new_servers).should == 'description'
+    end
+    it "takes into account the :only option" do
+      @balancer.only = [:server1]
+
+      @balancer.should_receive(:instances=).with(['i-00000001'])
+      @balancer.should_receive(:zones=).with(['eu-west-1a'])
+      @balancer.should_receive(:update_listeners)
+      @balancer.should_receive(:update_sticky_cookies)
+      @balancer.should_receive(:description).and_return('description')
+      @balancer.update_with(@new_servers).should == 'description'
+    end
+    it "takes into account the :except option" do
+      @balancer.except = [:server1]
+
+      @balancer.should_receive(:instances=).with(['i-00000002'])
+      @balancer.should_receive(:zones=).with(['eu-west-1b'])
+      @balancer.should_receive(:update_listeners)
+      @balancer.should_receive(:update_sticky_cookies)
+      @balancer.should_receive(:description).and_return('description')
+      @balancer.update_with(@new_servers).should == 'description'
+    end
+  end
+
+end
+