dreadpiratepj-poolparty 0.0.8

data/spec/remoter_spec.rb

@@ -0,0 +1,65 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+class TestRemote
+  include Remoter
+  include Callbacks
+  attr_accessor :ip
+end
+describe "Remoter" do
+  before(:each) do
+    @instance = RemoteInstance.new
+    @master = Master.new
+    
+    @remoter = TestRemote.new    
+    @remoter.stub!(:put).and_return "true"
+    # @tempfile = Tempfile.new("/tmp") do |f|
+    #   f << "hi"
+    # end
+    Application.stub!(:ec2_dir).and_return "/Users"
+    Application.stub!(:keypair).and_return "app"
+    Application.stub!(:username).and_return "root"
+  end
+  it "should have an ssh method that corresponds to ssm with the keypair" do
+    @remoter.ssh_string.should == "ssh -i /Users/id_rsa-app -o StrictHostKeyChecking=no"
+  end
+  it "should have a list of ssh_tasks" do
+    @remoter.ssh_tasks.should == []
+  end
+  it "should have a list of scp_tasks" do
+    @remoter.scp_tasks.should == []
+  end
+  it "should reset the values to nil when calling reset" do
+    @remoter.target_hosts.should_not be_nil
+    @remoter.reset!
+    @hosts.should be_nil
+  end
+  describe "executing" do
+    it "should call set_hosts before it executes the tasks" do
+      @remoter.should_receive(:set_hosts).once
+      @remoter.execute_tasks {}
+    end
+    it "should not call set_hosts before it executes the task if it explicitly doesn't want it to" do
+      @remoter.should_not_receive(:set_hosts)
+      @remoter.execute_tasks(:dont_set_hosts => true) {}
+    end
+    
+    describe "ssh" do
+      before(:each) do
+        @arr = []
+        @arr << @a = proc{puts "hello"}
+        @arr << @b = proc{puts "world"}
+        @instance.stub!(:ip).and_return("127.0.0.1")
+      end
+      it "should run the tasks in an array with run_thread_list" do
+        @remoter.should_receive(:run_thread_list).once
+        @remoter.run_array_of_tasks(@arr)
+      end
+      
+      it "should be able to collect the list of the target hosts's ips" do
+        Master.should_receive(:new).and_return(@master)
+        @master.stub!(:nodes).and_return([@instance])
+        @remoter.target_hosts.should == %w(127.0.0.1)
+      end
+    end
+  end
+end

data/spec/remoting_spec.rb

@@ -0,0 +1,84 @@
+require File.dirname(__FILE__) + '/spec_helper'
+require File.dirname(__FILE__) + "/helpers/ec2_mock"
+
+describe "Master remoting: " do
+  before(:each) do
+    Kernel.stub!(:system).and_return true
+    Application.stub!(:environment).and_return("test") # So it doesn't daemonize
+    Application.stub!(:minimum_instances).and_return(2)
+    Application.stub!(:maximum_instances).and_return(10)
+    Application.stub!(:polling_time).and_return(0.1)
+    Application.stub!(:verbose).and_return(false) # Turn off messaging
+    
+    @master = Master.new
+    @master.launch_new_instance!
+  end
+  describe "starting" do
+    before(:each) do
+      @master.start_cloud!
+    end
+    it "should start the cloud with instances" do
+      @master.list_of_instances.should_not be_empty
+    end
+    it "should start the cloud with running instances" do
+      @master.list_of_running_instances.should_not be_empty
+    end
+    it "should start with the minimum_instances running" do
+      wait 0.5 # Give the last one time to get to running
+      @master.list_of_running_instances.size.should == Application.minimum_instances
+    end
+  end
+  describe "maintaining" do
+    it "should maintain the minimum_instances if one goes down" do
+      @master.start_cloud!
+      wait 0.2 # Give the two instances time to boot up
+      (Application.minimum_instances - @master.number_of_pending_and_running_instances).should == 0
+      
+      # Kill one off to test how it handles the response
+      @master.terminate_instance!(@master.list_of_running_instances[0][:instance_id])
+      (Application.minimum_instances - @master.number_of_pending_and_running_instances).should == 1
+      @master.launch_minimum_instances # Assume this runs in the bg process
+
+      (Application.minimum_instances - @master.number_of_pending_and_running_instances).should == 0
+      @master.number_of_pending_and_running_instances.should == Application.minimum_instances
+    end
+    it "should launch a new instance when the load gets too heavy set in the configs" do
+      @master.stub!(:expand?).and_return true
+      @master.start_cloud!
+      wait 0.2 # Give the two instances time to boot up
+      (Application.minimum_instances - @master.number_of_pending_and_running_instances).should == 0
+      @master.scale_cloud!
+      @master.nodes.size.should == Application.minimum_instances + 1
+    end
+    it "should terminate an instance when the load shows that it's too light" do
+      @master.stub!(:contract?).and_return true
+      @master.start_cloud!
+      @master.request_launch_new_instance
+      wait 0.5 # Give the two instances time to boot up
+      @master.number_of_pending_and_running_instances.should == Application.minimum_instances + 1
+      @master.scale_cloud!
+      @master.number_of_pending_and_running_instances.should == Application.minimum_instances
+    end
+  end
+  describe "configuring" do
+    it "should call configure on all of the nodes when calling reconfigure_running_instances" do
+      @master.nodes.each {|a| 
+        a.stub!(:status).and_return("running")
+        a.should_receive(:configure).and_return true 
+      }
+      @master.reconfigure_running_instances
+    end    
+    it "should call restart_with_monit on all of the nodes when calling restart_running_instances_services" do
+      @master.nodes.each {|a| a.should_receive(:restart_with_monit).and_return true }
+      @master.restart_running_instances_services
+    end
+    it "should be able to say there are no number_of_unconfigured_nodes left when all the nodes are configured" do
+      @master.nodes.each {|a| a.should_receive(:stack_installed?).and_return true }
+      @master.number_of_unconfigured_nodes.should == 0
+    end
+    it "should be able to say that there is an unconfigured node" do
+      @master.nodes[-1].should_receive(:stack_installed?).and_return false
+      @master.number_of_unconfigured_nodes.should_not == 0
+    end
+  end
+end

data/spec/scheduler_spec.rb

@@ -0,0 +1,75 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+class TestSched  
+  include Scheduler
+end
+describe "Scheduler" do
+  before(:each) do
+    @test = TestSched.new
+  end
+  it "should create a ScheduleTasks" do
+    @test._tasker.class.should == ScheduleTasks
+  end
+  describe "run_thread_loop" do
+    before(:each) do
+      @klass = Class.new
+      @klass.stub!(:pop).once.and_return true
+      @block = Proc.new {@klass.pop}
+    end
+    it "should yield the block that it is given" do
+      @test.run_thread_list &@block
+    end
+    it "should run run_threads" do
+      @test.should_receive(:run_threads).and_return true
+      @test.run_thread_list &@block
+    end
+    it "should call run on the _tasker" do
+      @test._tasker.should_receive(:run).once.and_return true
+      @test.run_thread_list &@block
+    end
+    describe "ScheduleTasks class" do
+      before(:each) do
+        @stasks = ScheduleTasks.new
+        @test.stub!(:_tasker).and_return(@stasks)
+      end
+      it "should have its tasks listed as an empty array if there are no tasks added" do
+        @stasks.tasks.should == []
+      end
+      it "should have one task listed if it is added" do
+        @test.add_task {@klass.pop}        
+        @stasks.tasks.size.should == 1
+      end
+      it "should add each task as a thread" do
+        Thread.should_receive(:new).once
+        @stasks.push proc{puts "hi"}
+      end
+    end
+    describe "when running" do
+      before(:each) do
+        @test.add_task {@klass.pop}
+      end
+      it "should not run the tasks after adding them" do
+        @klass.should_not_receive(:pop)
+        @test.add_task {@klass.pop}
+      end
+      it "should run the tasks when run_thread_list" do
+        @klass.should_receive(:pop)
+        @test.run_thread_list
+      end
+      it "should be able to add tasks and have the run_thread_list run" do
+        @test._tasker.class.should_receive(:synchronize).once        
+        @test.run_thread_list
+      end
+      it "should empty all the tasks after running them in the loop" do
+        @test.run_thread_list
+        @test._tasker.tasks.size.should == 0
+      end
+      describe "daemonizing" do
+        it "should detached the process" do
+          Process.should_receive(:detach).once
+          @test.daemonize
+        end
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,39 @@
+require File.join(File.dirname(__FILE__), *%w[.. lib poolparty])
+
+%w(test/spec).each do |library|
+  begin
+    require library
+  rescue
+    STDERR.puts "== Cannot run test without #{library}"
+  end
+end
+
+Dir["#{File.dirname(__FILE__)}/helpers/**"].each {|a| require a}
+
+include PoolParty
+extend PoolParty
+
+Application.environment = "test"
+Application.verbose = false
+
+def wait_launch(time=5)
+  pid = fork {yield}
+  wait time
+  Process.kill("INT", pid)
+  Process.wait(pid, 0)
+end
+
+module Test::Unit::AssertDifference
+  def assert_difference(object, method = nil, difference = 1)
+    initial_value = object.send(method)
+    yield
+    assert_equal initial_value + difference, object.send(method), "#{object}##{method}"
+  end
+
+  def assert_no_difference(object, method, &block)
+    assert_difference object, method, 0, &block
+  end
+end
+
+Test::Spec::Should.send(:include, Test::Unit::AssertDifference)
+Test::Spec::ShouldNot.send(:include, Test::Unit::AssertDifference)

data/spec/string_spec.rb

@@ -0,0 +1,28 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "String" do
+  before(:each) do
+    @string = "string"
+    @string.stub!(:bucket_objects).and_return([])
+  end
+  # Dumb test
+  it "should be able to call bucket_objects on itself" do
+    @string.should_receive(:bucket_objects)
+    @string.bucket_objects
+  end
+  describe "with config replacements" do
+    it "should replace those syms in the string" do
+      ("new :port" ^ {:port => 100}).should == "new 100"
+    end
+    it "should be able to detect vars" do
+      @string=<<-EOC
+listen web_proxy 127.0.0.1::client_port
+\tserver web1 127.0.0.1::port weight 1 minconn 3 maxconn 6 check inter 30000
+      EOC
+      (@string ^ {:client_port => 3000, :port => 3001}).should ==<<-EOO
+listen web_proxy 127.0.0.1:3000
+\tserver web1 127.0.0.1:3001 weight 1 minconn 3 maxconn 6 check inter 30000
+      EOO
+    end
+  end
+end

data/web/static/conf/nginx.conf

@@ -0,0 +1,22 @@
+server {
+
+    listen   80;
+    server_name  www.poolpartyrb.com;
+    rewrite ^/(.*) http://poolpartyrb.com permanent;
+   }
+
+
+server {
+
+    listen   80;
+    server_name poolpartyrb.com;
+
+    access_log /apps/poolpartyrb/logs/access.log;
+    error_log /apps/poolpartyrb/logs/error.log;
+        
+    location / {    
+      root   /apps/poolpartyrb;
+      index  index.html;
+    }
+
+  }

data/web/static/site/index.html

@@ -0,0 +1,71 @@
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
+<html>
+  <head>
+    <title>PoolParty - Cloud EC2 Computing Made Easy</title>
+    <link href="/stylesheets/reset.css" media="all" rel="stylesheet" type="text/css"/>
+    <link href="/stylesheets/application.css" media="all" rel="stylesheet" type="text/css"/>
+    <script type="text/javascript" src="/javascripts/jquery-1.2.6.pack.js"></script>
+    <script type="text/javascript" src="/javascripts/corner.js"></script>
+    <script type="text/javascript" src="/javascripts/application.js"></script>
+  </head>
+  <body>
+    <div class='container'>
+      <div class='balloon'><img alt="Balloon" src="images/balloon.png"/></div>
+      <div class='header'>
+        <h1><a href="/">PoolParty</a></h1>
+        <div class='clouds'><img alt="Clouds" src="images/clouds.png"/></div>
+        <div class='links'>
+          <a href="#">api</a>
+          <a href="http://poolparty.rubyforge.org/">docs</a>
+          <a href="misc.html">misc</a>
+          <a href="#">group</a>
+        </div>
+        <div class='clear'>
+        </div>
+      </div>
+      <div class='clear'>
+      </div>
+      <div class='content'>
+        <h2>
+          One Ruby <strong><a href="http://github.com/auser/pool-party/tree/master">Gem</a></strong> = easy EC2 computing cloud.
+        </h2>
+        <div class='tout'>
+          <p class='alt'>
+            Amazon's Elastic Compute Cloud provides pay-per-use scalability to the masses.  However, clouds do not have innate load-balancing ability, and there is no native failover solution.
+          </p>
+          <p class='alt'>
+            PoolParty is an <strong>open source</strong> tool that <strong>automates</strong> deployment, monitoring, and load balancing of EC2 instances.  Configure once, and relax by the poolside -- PoolParty will keep your site and its instances afloat.
+          </p>
+        </div>
+        <p>
+          "I'm convinced, how do I get started?"
+          <code>
+            sudo gem install auser-poolparty -s http://gems.github.com
+          </code>
+        </p>
+        <p>
+          "Okay, are there examples or documentation?"
+          <span class='alt'>
+            <a href="http://poolparty.rubyforge.org/">Docs</a>
+            &
+            <a href="#">Examples</a>
+          </span>
+        </p>
+        <p>
+          "How do I extend it?"
+          <p class='alt'>
+            More coming soon
+            <strong>
+              <a href='http://github.com/auser/pool-party-plugins/tree/master'>
+                http://github.com/auser/pool-party-plugins/tree/master
+              </a>
+            </strong>
+          </p>
+        </p>
+      </div>
+      <div class='cb'>
+        <a href="http://www.citrusbyte.com"><img alt="Cb" src="images/cb.png"/></a>
+      </div>
+    </div>
+  </body>
+</html>

data/web/static/site/javascripts/application.js

@@ -0,0 +1,3 @@
+$(document).ready(function() {
+  $(".content").corner();
+});

data/web/static/site/javascripts/corner.js

@@ -0,0 +1,178 @@
+/*
+ * jQuery corner plugin
+ *
+ * version 1.92 (12/18/2007)
+ *
+ * Dual licensed under the MIT and GPL licenses:
+ *   http://www.opensource.org/licenses/mit-license.php
+ *   http://www.gnu.org/licenses/gpl.html
+ */
+
+/**
+ * The corner() method provides a simple way of styling DOM elements.  
+ *
+ * corner() takes a single string argument:  $().corner("effect corners width")
+ *
+ *   effect:  The name of the effect to apply, such as round or bevel. 
+ *            If you don't specify an effect, rounding is used.
+ *
+ *   corners: The corners can be one or more of top, bottom, tr, tl, br, or bl. 
+ *            By default, all four corners are adorned. 
+ *
+ *   width:   The width specifies the width of the effect; in the case of rounded corners this 
+ *            will be the radius of the width. 
+ *            Specify this value using the px suffix such as 10px, and yes it must be pixels.
+ *
+ * For more details see: http://methvin.com/jquery/jq-corner.html
+ * For a full demo see:  http://malsup.com/jquery/corner/
+ *
+ *
+ * @example $('.adorn').corner();
+ * @desc Create round, 10px corners 
+ *
+ * @example $('.adorn').corner("25px");
+ * @desc Create round, 25px corners 
+ *
+ * @example $('.adorn').corner("notch bottom");
+ * @desc Create notched, 10px corners on bottom only
+ *
+ * @example $('.adorn').corner("tr dog 25px");
+ * @desc Create dogeared, 25px corner on the top-right corner only
+ *
+ * @example $('.adorn').corner("round 8px").parent().css('padding', '4px').corner("round 10px");
+ * @desc Create a rounded border effect by styling both the element and its parent
+ * 
+ * @name corner
+ * @type jQuery
+ * @param String options Options which control the corner style
+ * @cat Plugins/Corner
+ * @return jQuery
+ * @author Dave Methvin (dave.methvin@gmail.com)
+ * @author Mike Alsup (malsup@gmail.com)
+ */
+(function($) { 
+
+$.fn.corner = function(o) {
+    var ie6 = $.browser.msie && /MSIE 6.0/.test(navigator.userAgent);
+    function sz(el, p) { return parseInt($.css(el,p))||0; };
+    function hex2(s) {
+        var s = parseInt(s).toString(16);
+        return ( s.length < 2 ) ? '0'+s : s;
+    };
+    function gpc(node) {
+        for ( ; node && node.nodeName.toLowerCase() != 'html'; node = node.parentNode ) {
+            var v = $.css(node,'backgroundColor');
+            if ( v.indexOf('rgb') >= 0 ) { 
+                if ($.browser.safari && v == 'rgba(0, 0, 0, 0)')
+                    continue;
+                var rgb = v.match(/\d+/g); 
+                return '#'+ hex2(rgb[0]) + hex2(rgb[1]) + hex2(rgb[2]);
+            }
+            if ( v && v != 'transparent' )
+                return v;
+        }
+        return '#ffffff';
+    };
+    function getW(i) {
+        switch(fx) {
+        case 'round':  return Math.round(width*(1-Math.cos(Math.asin(i/width))));
+        case 'cool':   return Math.round(width*(1+Math.cos(Math.asin(i/width))));
+        case 'sharp':  return Math.round(width*(1-Math.cos(Math.acos(i/width))));
+        case 'bite':   return Math.round(width*(Math.cos(Math.asin((width-i-1)/width))));
+        case 'slide':  return Math.round(width*(Math.atan2(i,width/i)));
+        case 'jut':    return Math.round(width*(Math.atan2(width,(width-i-1))));
+        case 'curl':   return Math.round(width*(Math.atan(i)));
+        case 'tear':   return Math.round(width*(Math.cos(i)));
+        case 'wicked': return Math.round(width*(Math.tan(i)));
+        case 'long':   return Math.round(width*(Math.sqrt(i)));
+        case 'sculpt': return Math.round(width*(Math.log((width-i-1),width)));
+        case 'dog':    return (i&1) ? (i+1) : width;
+        case 'dog2':   return (i&2) ? (i+1) : width;
+        case 'dog3':   return (i&3) ? (i+1) : width;
+        case 'fray':   return (i%2)*width;
+        case 'notch':  return width; 
+        case 'bevel':  return i+1;
+        }
+    };
+    o = (o||"").toLowerCase();
+    var keep = /keep/.test(o);                       // keep borders?
+    var cc = ((o.match(/cc:(#[0-9a-f]+)/)||[])[1]);  // corner color
+    var sc = ((o.match(/sc:(#[0-9a-f]+)/)||[])[1]);  // strip color
+    var width = parseInt((o.match(/(\d+)px/)||[])[1]) || 10; // corner width
+    var re = /round|bevel|notch|bite|cool|sharp|slide|jut|curl|tear|fray|wicked|sculpt|long|dog3|dog2|dog/;
+    var fx = ((o.match(re)||['round'])[0]);
+    var edges = { T:0, B:1 };
+    var opts = {
+        TL:  /top|tl/.test(o),       TR:  /top|tr/.test(o),
+        BL:  /bottom|bl/.test(o),    BR:  /bottom|br/.test(o)
+    };
+    if ( !opts.TL && !opts.TR && !opts.BL && !opts.BR )
+        opts = { TL:1, TR:1, BL:1, BR:1 };
+    var strip = document.createElement('div');
+    strip.style.overflow = 'hidden';
+    strip.style.height = '1px';
+    strip.style.backgroundColor = sc || 'transparent';
+    strip.style.borderStyle = 'solid';
+    return this.each(function(index){
+        var pad = {
+            T: parseInt($.css(this,'paddingTop'))||0,     R: parseInt($.css(this,'paddingRight'))||0,
+            B: parseInt($.css(this,'paddingBottom'))||0,  L: parseInt($.css(this,'paddingLeft'))||0
+        };
+
+        if ($.browser.msie) this.style.zoom = 1; // force 'hasLayout' in IE
+        if (!keep) this.style.border = 'none';
+        strip.style.borderColor = cc || gpc(this.parentNode);
+        var cssHeight = $.curCSS(this, 'height');
+
+        for (var j in edges) {
+            var bot = edges[j];
+            // only add stips if needed
+            if ((bot && (opts.BL || opts.BR)) || (!bot && (opts.TL || opts.TR))) {
+                strip.style.borderStyle = 'none '+(opts[j+'R']?'solid':'none')+' none '+(opts[j+'L']?'solid':'none');
+                var d = document.createElement('div');
+                $(d).addClass('jquery-corner');
+                var ds = d.style;
+
+                bot ? this.appendChild(d) : this.insertBefore(d, this.firstChild);
+
+                if (bot && cssHeight != 'auto') {
+                    if ($.css(this,'position') == 'static')
+                        this.style.position = 'relative';
+                    ds.position = 'absolute';
+                    ds.bottom = ds.left = ds.padding = ds.margin = '0';
+                    if ($.browser.msie)
+                        ds.setExpression('width', 'this.parentNode.offsetWidth');
+                    else
+                        ds.width = '100%';
+                }
+                else if (!bot && $.browser.msie) {
+                    if ($.css(this,'position') == 'static')
+                        this.style.position = 'relative';
+                    ds.position = 'absolute';
+                    ds.top = ds.left = ds.right = ds.padding = ds.margin = '0';
+                    
+                    // fix ie6 problem when blocked element has a border width
+                    var bw = 0;
+                    if (ie6 || !$.boxModel)
+                        bw = sz(this,'borderLeftWidth') + sz(this,'borderRightWidth');
+                    ie6 ? ds.setExpression('width', 'this.parentNode.offsetWidth - '+bw+'+ "px"') : ds.width = '100%';
+                }
+                else {
+                    ds.margin = !bot ? '-'+pad.T+'px -'+pad.R+'px '+(pad.T-width)+'px -'+pad.L+'px' : 
+                                        (pad.B-width)+'px -'+pad.R+'px -'+pad.B+'px -'+pad.L+'px';                
+                }
+
+                for (var i=0; i < width; i++) {
+                    var w = Math.max(0,getW(i));
+                    var e = strip.cloneNode(false);
+                    e.style.borderWidth = '0 '+(opts[j+'R']?w:0)+'px 0 '+(opts[j+'L']?w:0)+'px';
+                    bot ? d.appendChild(e) : d.insertBefore(e, d.firstChild);
+                }
+            }
+        }
+    });
+};
+
+$.fn.uncorner = function(o) { return $('.jquery-corner', this).remove(); };
+    
+})(jQuery);