poolparty 0.0.4 → 0.2.6

data/spec/poolparty/core/time_spec.rb

@@ -0,0 +1,52 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Time" do
+  describe "parsing" do
+    it "should be able to parse seconds" do
+      10.seconds.should == 10
+    end
+    it "should be able to parse minutes" do
+      80.minutes.should == 4800
+    end
+    it "should be able to parse hours" do
+      2.hours.should == 7200
+    end
+    it "should be able to parse days" do
+      1.days.should == 86400
+    end
+    it "should be able to parse weeks" do
+      2.weeks.should == 1209600
+    end
+    it "should be able to parse months" do
+      1.month.should == 2678400
+    end
+  end
+  describe "from" do
+    it "should be able to find minutes ago" do
+      10.minutes.ago.to_s.should == (Time.new - 10.minutes).to_s
+    end
+    it "should be able to find from now" do
+      10.minutes.from_now.to_s.should == (Time.now + 10.minutes).to_s
+    end
+  end
+  describe "time_ago string" do
+    it "should be able to turn 10.minutes.ago into a string" do
+      10.minutes.time_ago.should == "10 minutes ago"
+    end
+    it "should be able to turn 30.seconds into Less than a minute ago" do
+      30.seconds.time_ago.should == "Less than a minute ago"
+    end
+    it "should turn 1.year.ago into 1 year ago" do
+      2.years.time_ago.should == "2 years ago"
+    end
+    it "should turn 35.days into 35 days ago" do
+      29.days.time_ago.should == "4 weeks ago"
+    end
+    it "should turn 45.days into 1 month ago" do
+      45.days.time_ago.should == "1 month ago"
+    end
+    it "should turn 70.days into 2 months ago" do
+      70.days.time_ago.should == "2 months ago"
+    end
+  end
+end

data/spec/poolparty/helpers/binary_spec.rb

@@ -0,0 +1,26 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require File.dirname(__FILE__) + '/../../../lib/poolparty/helpers/binary'
+
+describe "Binary" do
+  before(:each) do
+    Dir.stub!(:[]).and_return %w(init console)
+  end
+  it "should have the binary location set on Binary" do
+    Binary.binary_directory.should =~ /lib\/poolparty\/helpers\/\.\.\/\.\.\/\.\.\/bin/
+  end
+  it "should be able to list the binaries in the bin directory" do
+    Binary.available_binaries_for("pool").should == %w(console init)
+  end
+  it "should be able to say the binary is in the binary_directory" do
+    Binary.available_binaries_for("pool").include?("console")
+  end
+  describe "get_existing_spec_location" do
+    before(:each) do
+      ::File.stub!(:file?).and_return false
+      ::File.stub!(:file?).with("#{Base.storage_directory}/pool.spec").and_return true
+    end
+    it "should be a String" do
+      Binary.get_existing_spec_location.class.should == String
+    end
+  end
+end

data/spec/poolparty/helpers/display_spec.rb

@@ -0,0 +1,13 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require File.dirname(__FILE__) + '/../../../lib/poolparty/helpers/display'
+
+include Display
+
+describe "Display" do
+  it "should respond to pool_describe" do
+    self.respond_to?(:pool_describe).should == true
+  end
+  it "should be able to get the list of remote bases" do
+    self.respond_to?(:available_bases).should == true
+  end
+end

data/spec/poolparty/helpers/optioner_spec.rb

@@ -0,0 +1,39 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require File.dirname(__FILE__) + '/../../../lib/poolparty/helpers/optioner'
+
+describe "Option Parser" do
+  describe "options" do
+    before(:each) do
+      @op = PoolParty::Optioner.new
+      @op.parse_options
+    end
+    it "should set the options as an Hash" do
+      @op.options.class.should == Hash
+    end
+    it "should have the verbose option set to false by default" do
+      @op.verbose.should == false
+    end
+    it "should call a method called on it that is not defined on the options if they exist" do
+      @op.options.should_receive(:[]).with(:verbose).once.and_return true
+      @op.verbose
+    end
+    it "should be able to display the version" do
+      @op.version.should == PoolParty::VERSION::STRING
+    end
+    it "should exit after displaying the help message" do
+      hide_output do
+        lambda {
+          @o = PoolParty::Optioner.new(["-h"], {:parse_options => false})
+          @o.should_receive(:process_options).once
+          @o.parse_options
+        }
+      end
+    end
+  end
+  it "should be able to take a block and set some options on the block" do
+    o = PoolParty::Optioner.new(["-w"], {}) do |opts, optioner|
+      opts.on('-w', '--wee')    { optioner.wee "wee" }
+    end
+    o.wee.should == "wee"
+  end
+end

data/spec/poolparty/helpers/provisioner_base_spec.rb

@@ -0,0 +1,121 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+require File.dirname(__FILE__) + '/../../../lib/poolparty/helpers/provisioner_base'
+
+include Provisioner
+
+describe "ProvisionerBase" do
+  before(:each) do    
+    @cloud = cloud :app do;end    
+    @remote_instance = PoolParty::Remote::RemoteInstance.new({:ip => "192.168.0.1", :status => "running", :name => "master"}, @cloud)
+    @cloud.stub!(:custom_install_tasks_for).with(@remote_instance).and_return []
+    stub_list_from_remote_for(@remote_instance)
+    stub_list_from_remote_for(@cloud)
+  end
+  describe "class methods" do
+    it "should have install" do
+      ProvisionerBase.respond_to?(:install).should == true
+    end
+    it "should have configure" do
+      ProvisionerBase.respond_to?(:configure).should == true
+    end
+  end
+  describe "tasks" do
+    before(:each) do
+      class Provisioner::TestProvisioner < Provisioner::ProvisionerBase
+        def tasks
+          [
+            "hello",
+            " ",
+            "world"
+          ]          
+        end
+      end
+      @tp = TestProvisioner.new(@remote_instance,@cloud)
+    end
+    it "should have tasks on the provisioner" do
+      ProvisionerBase.new(@remote_instance,@cloud).install_tasks.class.should == Array
+    end
+    it "should have empty tasks on the provisioner" do
+      ProvisionerBase.new(@remote_instance,@cloud).install_tasks.should be_empty
+    end
+    it "should allow a new class to write tasks that aren't empty upon instantiation" do
+      @tp.tasks.should_not be_empty
+    end
+    it "should have described 3 tasks in the task_list" do
+      @tp.tasks.size.should == 3
+    end
+  end
+  it "should return a hash when asking for the installers" do
+    ProvisionerBase.installers.class.should == Hash
+  end
+  it "should not return an empty string when asking for the Ubuntu installer" do
+    ProvisionerBase.installers[:ubuntu].should_not be_nil
+  end
+  it "should be able to fetch the ubuntu installer with the helper method installer" do
+    ProvisionerBase.new(@remote_instance,@cloud, "ubuntu").installer_for.should == "apt-get install -y "
+  end
+  it "should be able to fetch the fedora installer with the helper method installer" do
+    ProvisionerBase.new(@remote_instance,@cloud, "fedora").installer_for.should == "yum install "
+  end
+  describe "install_string" do
+    before(:each) do
+      class Provisioner::BTestProvisioner < Provisioner::ProvisionerBase
+        def install_tasks
+          [
+            "hello",
+            insert_space,
+            add_world
+          ]          
+        end
+        def insert_space
+          " "
+        end
+        def add_world
+          "cruel world"
+        end
+        def default_install_tasks
+          [] << install_tasks
+        end
+      end
+      @provisioner = BTestProvisioner.new(@remote_instance, @cloud)
+    end
+    it "should not run \"hello\"" do
+      @provisioner.should_not_receive(:hello)
+    end
+    it "should run insert_space" do
+      @provisioner.should_receive(:insert_space).once.and_return " "
+    end
+    it "should run add_world" do
+      @provisioner.should_receive(:insert_space).once.and_return "world"
+    end
+    it "should compound the string to be hello \n   \n cruel world" do
+      @provisioner.install_string.should == "hello \n   \n cruel world"
+    end
+    after do
+      @provisioner.install_string
+    end
+    describe "processing" do
+      before(:each) do
+        @provisioner = ProvisionerBase.new(@remote_instance, @cloud)
+        stub_list_from_remote_for(@cloud)
+        @cloud.stub!(:keypair).and_return "fake_keypair"
+        @cloud.stub!(:keypair_path).and_return "~/.ec2/fake_keypair"
+        Provisioner::Master.stub!(:new).and_return @provisioner
+        @cloud.stub!(:copy_file_to_storage_directory).and_return true
+      end
+      describe "provision_master" do
+        it "should call write_install_file" do
+          @provisioner.should_receive(:write_install_file).and_return(true)          
+        end
+        it "should call process_install!" do
+          @provisioner.should_receive(:process_install!).and_return(true)
+        end
+        after(:each) do
+          hide_output {
+            Provisioner.provision_master(@cloud, true)
+          }
+        end
+      end
+    end
+  end
+end

data/spec/poolparty/helpers/provisioners/master_spec.rb

@@ -0,0 +1,54 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+require File.dirname(__FILE__) + '/../../../../lib/poolparty/helpers/provisioner_base'
+
+include Provisioner
+
+describe "Master provisioner" do
+  before(:each) do
+    @cloud = cloud :app do; end
+    stub_list_from_remote_for(@cloud)
+    @remote_instance = PoolParty::Remote::RemoteInstance.new({:ip => "192.168.0.1", :status => "running", :name => "master"}, @cloud)
+    
+    @cloud.stub!(:master).and_return @ris.first
+    @master = Master.new(@cloud, :ubuntu)
+  end
+  describe "install_tasks" do
+    before(:each) do
+      @cloud.stub!(:master).and_return @ris.first
+      @master.stub!(:cloud).and_return @cloud
+    end
+    it "should call install_puppet_master" do
+      @master.should_receive(:install_puppet_master)
+    end
+    it "should call create_local_hosts_entry" do
+      @master.should_receive(:create_local_hosts_entry)
+    end
+    it "should call setup_fileserver" do
+      @master.should_receive(:setup_fileserver)
+    end
+    it "should call create_local_node" do
+      @master.should_receive(:create_local_node)      
+    end
+    it "should call the custom_install_tasks" do
+      @master.should_receive(:custom_install_tasks)
+    end
+    after do
+      @master.install
+    end
+  end
+  it "should return install_puppet_master as apt-get install puppet factor" do
+    @master.install_puppet_master.should =~ /apt-get install -y puppet puppetmaster/
+  end
+  it "should return setup basic structure" do
+    @master.setup_basic_structure.should =~ /puppetmasterd --mkusers/
+  end
+  it "should return setup_fileserver with the setup" do
+    @master.setup_fileserver.should =~ /\[files\]/
+  end
+  it "should be able to create_local_node" do
+    @master.create_local_node.should =~ /ode \"master\"/
+  end
+  it "should create a node1 node as well" do
+    @master.create_local_node.should =~ /ode \"node1\"/
+  end
+end

data/spec/poolparty/helpers/provisioners/slave_spec.rb

@@ -0,0 +1,28 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+require File.dirname(__FILE__) + '/../../../../lib/poolparty/helpers/provisioner_base'
+
+include Provisioner
+
+describe "Slave provisioner" do
+  before(:each) do
+    
+    @cloud = cloud :app do; end
+    @remote_instance = PoolParty::Remote::RemoteInstance.new({:ip => "192.168.0.1", :status => "running", :name => "master"}, @cloud)
+    stub_list_from_remote_for(@cloud)
+    
+    @cloud.stub!(:master).and_return @ris.first
+    
+    @slave = Slave.new(@remote_instance, @cloud, :ubuntu)
+  end
+  describe "install_tasks" do
+    it "should call install_puppet_slave" do
+      @slave.should_receive(:install_puppet_master)
+    end
+    after do
+      @slave.install
+    end
+  end
+  it "should return install_puppet as apt-get install puppet factor" do
+    @slave.setup_puppet.should =~ /puppetd/
+  end
+end

data/spec/poolparty/modules/cloud_resourcer_spec.rb

@@ -0,0 +1,135 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+class ResourcerTestClass
+  include CloudResourcer
+  include Configurable  
+  
+  def initialize(&block)
+    store_block(&block) if block
+  end
+  
+  # Stub keypair
+  def keypair
+    "rangerbob"
+  end
+end
+class TestParentClass
+  def options
+    {}
+  end
+  def services
+    @services ||= []
+  end
+  def add_service(s)
+    services << s
+  end
+end
+describe "CloudResourcer" do
+  before(:each) do
+    @tc = ResourcerTestClass.new
+  end
+  it "should have the method instances" do
+    @tc.respond_to?(:instances).should == true
+  end
+  it "should be able to accept a range and set the first to the minimum instances" do
+    @tc.instances 4..10
+    @tc.minimum_instances.should == 4
+  end
+  it "should set the max to the maximum instances to the last" do
+    @tc.instances 4..10
+    @tc.maximum_instances.should == 10
+  end
+  describe "keypair_path" do
+    before(:each) do
+    end
+    it "should look for the file in the known directories it should reside in" do
+      @tc.should_receive(:keypair_paths).once.and_return []
+      @tc.keypair_path
+    end
+    it "should see if the file exists" do
+      @t = "#{File.expand_path(Base.base_keypair_path)}"
+      ::File.should_receive(:exists?).with(@t+"/id_rsa-rangerbob").and_return false
+      ::File.stub!(:exists?).with(@t+"/rangerbob").and_return false
+      @tc.should_receive(:keypair_paths).once.and_return [@t]
+      @tc.keypair_path
+    end
+    it "should fallback to the second one if the first doesn't exist" do
+      @t = "#{File.expand_path(Base.base_keypair_path)}"
+      @q = "#{File.expand_path(Base.base_config_directory)}"
+      ::File.stub!(:exists?).with(@t+"/id_rsa-rangerbob").and_return false
+      ::File.stub!(:exists?).with(@t+"/rangerbob").and_return false
+      ::File.stub!(:exists?).with(@q+"/id_rsa-rangerbob").and_return false
+      ::File.should_receive(:exists?).with(@q+"/rangerbob").and_return true
+      @tc.should_receive(:keypair_paths).once.and_return [@t, @q]
+      @tc.keypair_path.should == "/etc/poolparty/rangerbob"
+    end
+    describe "exists" do
+      before(:each) do
+        @t = "#{File.expand_path(Base.base_keypair_path)}"
+        ::File.stub!(:exists?).with(@t+"/id_rsa-rangerbob").and_return false
+        ::File.stub!(:exists?).with(@t+"/rangerbob").and_return true
+      end
+      it "should have the keypair_path" do
+        @tc.respond_to?(:keypair_path).should == true
+      end
+      it "should set the keypair to the Base.keypair_path" do      
+        @tc.keypair_path.should =~ /#{File.expand_path(Base.base_keypair_path)}/
+      end
+      it "should set the keypair to have the keypair set" do
+        @tc.keypair.should =~ /rangerbob/
+      end
+      it "should set it to the Base keypair_path and the keypair" do
+        @tc.keypair_path.should == "#{File.expand_path(Base.base_keypair_path)}/#{@tc.keypair}"
+      end
+    end
+  end
+  it "should provide set_parent" do
+    @tc.respond_to?(:set_parent).should == true
+  end
+  describe "parents" do
+    before(:each) do
+      @testparent = TestParentClass.new
+    end
+    describe "setting" do
+      it "should add the child to its services" do
+        @testparent.should_receive(:add_service)
+      end
+      it "should call configure with options" do
+        @tc.should_receive(:configure).with(@testparent.options)      
+      end
+      after do
+        @tc.set_parent(@testparent)
+      end      
+    end
+    describe "parent's services" do
+      before(:each) do        
+        @tc.set_parent(@testparent)        
+      end
+      it "should set the parent" do
+        @tc.parent.should == @testparent
+      end
+      it "should have one service set" do
+        @testparent.services.size.should == 1
+      end
+      it "should have the child in the parent's services" do
+        @testparent.services.first.should == @tc
+      end
+    end
+    describe "storing block" do
+      before(:each) do
+        @new_tc = ResourcerTestClass.new do
+          "hi"
+        end
+      end
+      it "should store the block when creating a new one" do
+        @new_tc.store_block.should_not == nil
+      end
+      it "should have a reference to the stored block" do
+        @new_tc.store_block.class.should == Proc
+      end
+      it "should store the containing block" do
+        @new_tc.store_block.call.should == "hi"
+      end
+    end
+  end
+end