dreadpiratepj-poolparty 0.0.8

data/spec/monitors/cpu_monitor_spec.rb

@@ -0,0 +1,38 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require "lib/poolparty/monitors/cpu"
+
+describe "monitors" do
+  describe "when included" do
+    before(:each) do      
+      @master = Master.new
+      @instance = RemoteInstance.new
+    end
+    it "should include them in the Monitors module" do
+      @master.methods.include?("cpu").should == true
+    end
+    it "should also include the new methods in the remote model" do
+      RemoteInstance.new.methods.include?("cpu").should == true
+    end
+    describe "master" do
+      before(:each) do
+        @master.stub!(:list_of_nonterminated_instances).and_return(
+        [{:instance_id => "i-abcdde1"}]
+        )
+      end
+      it "should try to collect the cpu for the entire set of remote instances when calling cpu" do
+        @master.nodes.should_receive(:inject).once.and_return 0.0
+        @master.cpu
+      end
+    end
+    describe "remote instance" do
+      it "should try to ssh into the remote instance" do
+        @instance.should_receive(:ssh).once.with("uptime")
+        @instance.cpu
+      end
+      it "should be able to find the exact amount of time the processor has been up" do
+        @instance.stub!(:ssh).once.with("uptime").and_return("18:55:31 up 5 min,  1 user,  load average: 0.32, 0.03, 0.00")
+        @instance.cpu.should == 0.32
+      end
+    end
+  end
+end

data/spec/monitors/memory_spec.rb

@@ -0,0 +1,50 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require "lib/poolparty/monitors/memory"
+
+describe "monitors" do
+  describe "when included" do
+    before(:each) do      
+      @master = Master.new
+      @instance = RemoteInstance.new
+    end
+    it "should include them in the Monitors module" do
+      @master.methods.include?("memory").should == true
+    end
+    it "should also include the new methods in the remote model" do
+      @instance.methods.include?("memory").should == true
+    end
+    describe "master" do
+      before(:each) do
+        @master.stub!(:list_of_nonterminated_instances).and_return(
+        [{:instance_id => "i-abcdde1"}]
+        )
+      end
+      it "should try to collect the cpu for the entire set of remote instances when calling cpu" do
+        @master.nodes.should_receive(:inject).once.and_return 0.0
+        @master.memory
+      end
+    end
+    describe "remote instance" do
+      it "should try to ssh into the remote instance" do
+        @instance.should_receive(:ssh).once.with("free -m | grep -i mem")
+        @instance.memory
+      end
+      it "should be able to find the exact amount of time the processor has been up" do
+        @instance.stub!(:ssh).once.with("free -m | grep -i mem").and_return("Mem:          1700         546       1644          0          2         18")
+        @instance.memory.round_to(2).should == 0.32
+      end
+    end
+    # it "should web monitor should be able to extract the amount of the requests it can handle" do
+    #   str = "Request rate: 1.5 req/s (649.9 ms/req)"
+    #   # Monitors::Web.monitor_from_string(str).should == 1.5
+    # end
+    # it "should be able to monitor the percentage of memory available on the server" do
+    #   str = "Mem:          1700         56       1644          0          2         18"
+    #   # Monitors::Memory.monitor_from_string(str).to_s.should =~ /0.032/
+    # end
+    # it "should be able to show the load on the cpu available on the server" do
+    #   str = "18:55:31 up 5 min,  1 user,  load average: 0.32, 0.03, 0.00"
+    #   # Monitors::Cpu.monitor_from_string(str).should == 0.32
+    # end
+  end
+end

data/spec/monitors/misc_monitor_spec.rb

@@ -0,0 +1,50 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+module Database
+  module Master
+    def db
+      nodes.inject(0) {|i,inst| i * inst.db} / nodes.size
+    end
+  end
+  module Remote
+    def db
+      5.0
+    end
+  end
+end
+
+PoolParty.register_monitor Database
+
+describe "monitors (random, to spec the inclusion)" do
+  describe "when included" do
+    before(:each) do      
+      @master = Master.new
+      @instance = RemoteInstance.new
+    end
+    it "should include them in the Monitors module" do
+      @master.methods.include?("db").should == true
+    end
+    it "should also include the new methods in the remote model" do
+      @instance.methods.include?("db").should == true
+    end
+    describe "master" do
+      before(:each) do
+        @master.stub!(:list_of_nonterminated_instances).and_return(
+        [{:instance_id => "i-abcdde1"}]
+        )
+      end
+      it "should try to collect the cpu for the entire set of remote instances when calling cpu" do
+        @master.nodes.should_receive(:inject).once.and_return 5.0
+        @master.db.should == 5.0
+      end
+    end
+    describe "remote instance" do
+      it "should try to ssh into the remote instance" do
+        @instance.db.should == 5.0
+      end
+      it "should be able to find the exact amount of time the processor has been up" do
+        @instance.db.round_to(2).should == 5.0
+      end
+    end
+  end
+end

data/spec/monitors/web_spec.rb

@@ -0,0 +1,39 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require "lib/poolparty/monitors/web"
+
+describe "monitors" do
+  describe "when included" do
+    before(:each) do      
+      Application.stub!(:client_port).and_return 8001
+      @master = Master.new
+      @instance = RemoteInstance.new
+    end
+    it "should include them in the Monitors module" do
+      @master.methods.include?("web").should == true
+    end
+    it "should also include the new methods in the remote model" do
+      @instance.methods.include?("web").should == true
+    end
+    describe "master" do
+      before(:each) do
+        @master.stub!(:list_of_nonterminated_instances).and_return(
+        [{:instance_id => "i-abcdde1"}]
+        )
+      end
+      it "should try to collect the cpu for the entire set of remote instances when calling cpu" do
+        @master.nodes.should_receive(:inject).once.and_return 0.0
+        @master.web
+      end
+    end
+    describe "remote instance" do
+      it "should try to ssh into the remote instance" do
+        @instance.should_receive(:ssh).once.with("httperf --server localhost --port #{Application.client_port} --num-conn 3 --timeout 5 | grep 'Request rate'")
+        @instance.web
+      end
+      it "should be able to find the exact amount of time the processor has been up" do
+        @instance.stub!(:ssh).once.and_return("Request rate: 1.5 req/s (649.9 ms/req)")
+        @instance.web.round_to(2).should == 1.5
+      end
+    end
+  end
+end

data/spec/optioner_spec.rb

@@ -0,0 +1,22 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "Optioner with options" do
+  it "should be able to pull out the lonely arguments without any switches" do
+    Optioner.parse("hello".split(" ")).should == ["hello"]
+  end
+  it "should be able to pull out the lonely arguments with switches" do
+    Optioner.parse("-s 30.seconds -m hello world".split(" ")).should == ["world"]
+  end
+  it "should be able to pull out start from the the string" do
+    Optioner.parse("-c 'config/config.yml' -A 'Who' -S 'DarkwingDuck' list".split(" ")).should == ["list"]
+  end
+  it "should be able to pull out the lonely arguments with optional argument switches" do
+    Optioner.parse("-s 30 -q -n start".split(" "), %w(-q -n)).should == ["start"]
+  end
+  it "should pull out the lonely arguments if none are there" do
+    Optioner.parse("-s 30 -q".split(" ")).should == []
+  end
+  it "should pull out empty array if there are no lonely arguments" do
+    Optioner.parse("-s 30".split(" ")).should == []
+  end
+end

data/spec/plugin_manager_spec.rb

@@ -0,0 +1,31 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "Plugin manager" do
+  before(:each) do
+    FileUtils.stub!(:mkdir_p).and_return true
+    Dir["./spec/../lib/../vendor/*"].each {|a| FileUtils.rm_rf a}
+  end
+  it "should git clone the directory when it is installing a plugin" do
+    File.stub!(:directory?).and_return false
+    Git.should_receive(:clone).with("git@github.com:auser/pool-party.git", "/Users/auser/Sites/work/citrusbyte/internal/gems/pool-party/pool/vendor/pool-party").and_return true
+    PluginManager.install_plugin "git@github.com:auser/pool-party.git"
+  end
+  it "should keep a list of the installed plugin locations" do
+    PluginManager.install_plugin "git@github.com:auser/pool-party-plugins.git"
+    PoolParty.installed_plugins.should == ["git@github.com:auser/pool-party-plugins.git"]
+  end
+  it "should be able to rescan the plugin directory and tell which plugins are installed" do
+    PluginManager.install_plugin "git@github.com:auser/pool-party-plugins.git"
+    PluginManager.scan.should == %w(pool-party-plugins)
+  end
+  it "should be able to remove a plugin based on the name" do
+    PluginManager.install_plugin "git@github.com:auser/pool-party-plugins.git"
+    PluginManager.remove_plugin "pool-party-plugins"
+    PluginManager.scan.should == %w()
+  end
+  it "should be able to extract the git repos from the .git/config file" do
+    PluginManager.install_plugin "git@github.com:auser/pool-party-plugins.git"
+    PoolParty.reset!
+    PoolParty.installed_plugins.should == ["git@github.com:auser/pool-party-plugins.git"]
+  end
+end

data/spec/plugin_spec.rb

@@ -0,0 +1,101 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+class TestPlugin < PoolParty::Plugin
+  after_define_tasks :takss
+  after_install :echo_hosts, :email_updates
+  before_configure :echo_hosts  
+  after_start :echo_start
+  
+  def echo_start(master)
+    "start"
+  end
+  def echo_hosts(caller)
+    "hosts"
+  end
+  def email_updates(caller)
+    "email"
+  end
+  def takss(tasks)
+    "tasks"
+  end
+end
+
+describe "Plugin" do
+  it "should define run_before method" do
+    TestPlugin.methods.include?("before_install").should == true
+  end
+  it "should define run_after method" do
+    TestPlugin.methods.include?("after_install").should == true
+  end
+  it "should define a singleton method on the plugin'ed class" do
+    Master.new.methods.include?("testplugin").should == true
+  end
+  describe "usage" do
+    before(:each) do
+      @instance = RemoteInstance.new
+      @master = Master.new
+      
+      @test = TestPlugin.new
+      @test.stub!(:echo_hosts).and_return("true")
+      @test.stub!(:email_updates).and_return("true")
+      @test.stub!(:echo_start).and_return("true")
+      TestPlugin.stub!(:new).and_return(@test)
+      Kernel.stub!(:wait).and_return true
+      
+      @master.stub!(:launch_minimum_instances).and_return true
+      @master.stub!(:number_of_pending_instances).and_return 0
+      @master.stub!(:get_node).with(0).and_return @instance
+      
+      @instance.stub!(:ssh).and_return "true"
+      @instance.stub!(:scp).and_return "true"
+      Kernel.stub!(:system).and_return "true"
+    end
+    it "should should call echo_hosts after calling configure" do      
+      @test.should_receive(:echo_hosts).at_least(1)
+      @instance.install
+    end
+    it "should call email_updates after calling install" do
+      @test.should_receive(:email_updates).at_least(1)
+      @instance.install
+    end
+    it "should call echo_hosts before it calls configure" do
+      @test.should_receive(:echo_hosts).at_least(1).and_return "hi"
+      @instance.configure
+    end
+    it "should not call echo_hosts after if configures" do
+      @test.should_not_receive(:email_updates)
+      @instance.configure
+    end
+    it "should say that it started on the master" do
+      @test.should_receive(:echo_start).at_least(1).and_return "hi"
+      @master.start
+    end
+    describe "instance methods" do
+      before(:each) do
+        @str = "filename"
+        @str.stub!(:read).and_return "filename"
+        @test.stub!(:open).and_return @str
+      end
+      it "should try to open the file with the given filename" do
+        @test.should_receive(:open).with("filename").and_return @str
+        @test.read_config_file("filename")
+      end
+      it "should open a yaml file" do
+        YAML.should_receive(:load).with("filename").and_return ""
+        @test.read_config_file("filename")
+      end
+      describe "when reading the yaml file" do
+        before(:each) do
+          @str.stub!(:read).and_return ":username: eddie\n:password: eddie"
+        end
+        it "should parse the yaml file to a Hash" do
+          @str.should_receive(:read).and_return ":username: eddie\n:password: eddie"
+          @test.read_config_file("filename").class.should == Hash
+        end
+        it "should parse the yaml file into the proper hash" do
+          @test.read_config_file("filename").should == {:username => "eddie", :password => "eddie"}
+        end
+      end
+    end
+  end
+end

data/spec/pool_binary_spec.rb

@@ -0,0 +1,10 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "Pool binary" do
+  describe "running" do
+    it "should call PoolParty.options" do
+      options = PoolParty.options(:optsparse => {:banner => "Usage: pool [OPTIONS] {start | stop | list | maintain | restart}"  })
+    end
+    
+  end
+end

data/spec/poolparty_spec.rb

@@ -0,0 +1,15 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "Application options" do
+  it "should parse and use a config file if it is given for the options" do
+    YAML.should_receive(:load).and_return({:config_file => "config/sample-config.yml"})
+    Application.make_options(:config_file => "config/sample-config.yml")
+  end
+  it "should require all the plugin's init files in the plugin directory" do
+    PoolParty.should_receive(:load_plugins).once
+    Application.options
+  end
+  it "should be able to say that the plugin directory is the current directory" do
+    File.basename(PoolParty.plugin_dir).should == "vendor"
+  end
+end

data/spec/provider_spec.rb

@@ -0,0 +1,17 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe "Provider" do
+  before(:each) do
+    @ips = ["127.0.0.1"]
+  end
+  it "should be able to make a roles from the ips" do
+    Provider.string_roles_from_ips(@ips).should == "role :app, '127.0.0.1'"
+  end
+  describe "running" do
+    it "should be able to run with the provided packages" do
+      Sprinkle::Script.should_receive(:sprinkle).once.and_return true
+      Provider.should_receive(:string_roles_from_ips).with(@ips).and_return ""
+      Provider.install_poolparty(@ips)
+    end
+  end
+end

data/spec/remote_instance_spec.rb

@@ -0,0 +1,149 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+class RemoteInstance
+  def scp(src="", dest="", opts={})
+    "true"
+  end
+  # Ssh into the instance or run a command, if the cmd is set
+  def ssh(cmd="")
+    "true"
+  end
+end
+describe "remote instance" do
+  before(:each) do
+    @instance = RemoteInstance.new({:ip => "127.0.0.1", :instance_id => "i-abcdef1"})
+    @instance.stub!(:ssh).and_return true
+    @instance.stub!(:scp).and_return true
+    Kernel.stub!(:system).and_return true
+    
+    @master = Master.new
+  end
+  
+  describe "scping" do
+  end
+  describe "ssh'ing" do
+  end
+  
+  describe "in general" do
+    it "should set the ip upon creation" do
+      @instance.ip.should == "127.0.0.1"
+    end
+    it "should be able to tell if it is the master" do
+      @instance.master?.should == true
+    end
+    it "should be able to say that it isn't secondary?" do
+      @instance.secondary?.should_not be_true
+    end
+    it "should be able to build itself a haproxy_resources_entry" do
+      @instance.haproxy_resources_entry.should =~ /node0/
+    end
+    it "should be able to build a list of the heartbeat nodes" do
+      @instance.node_entry.should =~ /node0/
+    end
+    it "should be able to build a hosts_entry for self" do
+      @instance.hosts_entry.should =~ /node0/
+    end
+    it "should be able to have local_hosts_entry with 127.0.0.01" do
+      @instance.local_hosts_entry.should =~ /127\.0\.0\.1/
+    end
+    it "should have a heartbeat_entry" do
+      @instance.heartbeat_entry.should =~ /127\.0\.0\.1/
+      @instance.heartbeat_entry.should =~ /#{Application.managed_services}/
+    end
+    it "should be able to build a haproxy_entry" do
+      @instance.haproxy_entry.should =~ /server/
+    end
+    describe "callbacks" do
+      it "should call configure after it calls install"
+    end
+  end
+    describe "new configuration style (build scripts)" do
+      before(:each) do
+        @tempfile = Tempfile.new("/tmp")
+        Kernel.stub!(:system).and_return true
+      end
+      # it "should try to run the scp build file" do
+      #   Master.should_receive(:build_scp_instances_script_for).with(@instance).and_return @tempfile
+      #   @instance.configure
+      # end
+      # it "should try to run and build the reconfigure script for the node" do
+      #   Master.should_receive(:build_reconfigure_instances_script_for).with(@instance).and_return @tempfile
+      #   @instance.configure
+      # end
+      # it "should scp the reconfigure file to the remote instance" do
+      #   @instance.should_receive(:scp).once.and_return true
+      #   @instance.configure
+      # end
+      # it "should ssh and execute the reconfigure file on the remote instance" do
+      #   @instance.should_receive(:ssh).once.with("chmod +x /usr/local/src/reconfigure.sh && /bin/sh /usr/local/src/reconfigure.sh").and_return true
+      #   @instance.configure
+      # end
+      describe "with a public ip" do
+        before(:each) do
+          Application.stub!(:public_ip).and_return "127.0.0.1"
+        end
+        it "should run associate_address if there is a public_ip set in the Application.options" do
+          @instance.should_receive(:associate_address_with).with(Application.public_ip, @instance.instance_id).and_return true
+          @instance.configure
+        end
+        it "should not run associate_address_with if the public_ip is empty" do
+          Application.stub!(:public_ip).and_return ""
+          @instance.should_not_receive(:associate_address_with)
+          @instance.configure
+        end
+
+    end
+  end  
+  describe "in failover" do
+    it "should be able to become master " do
+      @instance.stub!(:configure).and_return true
+      @instance.number = 1
+      @instance.become_master
+      @instance.number.should == 0
+    end
+    it "should reconfigure after becoming master" do
+      @instance.should_receive(:configure).and_return true
+      @instance.become_master
+    end
+    it "should say that it is the master after becoming master" do
+      @instance.stub!(:configure).and_return true
+      @instance.become_master
+      @instance.master?.should == true
+    end
+    it "should be able to detect is_not_master_and_master_is_not_running? and return false when the server is the master" do
+      @instance.is_not_master_and_master_is_not_running?.should == false
+    end
+    it "should be able to detect is_not_master_and_master_is_not_running? and return false when the master server is responding" do
+      Master.stub!(:is_master_responding?).and_return true
+      @instance.is_not_master_and_master_is_not_running?.should == false
+    end
+    it "should be able to detect is_not_master_and_master_is_not_running? and return false when the master server is responding" do
+      @instance.stub!(:master?).and_return false
+      Master.stub!(:is_master_responding?).and_return false
+      @instance.is_not_master_and_master_is_not_running?.should == true
+    end
+    
+    describe "when installing the poolparty software" do
+      it "should be able to detect if the stack_installed? == false" do
+        @instance.stack_installed?.should == false
+      end
+      it "should set the stack_installed? once installed" do
+        @instance.install
+        @instance.stack_installed?.should == true
+      end
+    end
+    
+    describe "when installing plugins" do
+      it "should call update_plugins after become master" do
+        @instance.should_receive(:update_plugins).at_least(1)
+        @instance.configure
+      end
+      it "should try to install the plugins from the git repos of the installed plugins" do
+        PluginManager.remove_plugin "pool-party-plugins"
+        PluginManager.install_plugin "git@github.com:auser/pool-party-plugins.git"
+        @instance.update_plugin_string.should == ["git@github.com:auser/pool-party-plugins.git"]
+      end
+    end
+    
+  end
+end