fairchild-poolparty 1.1.3

data/spec/poolparty/resources/package_spec.rb

@@ -0,0 +1,29 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "File" do
+  describe "instances" do
+    before(:each) do
+      @tc = TestBaseClass.new do
+        has_package(:name => "apache2")
+      end
+      @package = @tc.resource(:package).first
+    end
+    it "have the name in the options" do
+      @package.name.should == "apache2"
+    end
+    it "should ensure it's present" do
+      @package.ensures.should == "present"
+    end
+    describe "into PuppetResolver" do
+      before(:each) do
+        @compiled = PuppetResolver.new(@tc.to_properties_hash).compile
+      end
+      it "should set the filename to the name of the file" do
+        @compiled.should match(/package \{ "apache2"/)
+      end
+      it "have the mode set in the puppet output" do
+        @compiled.should match(/ensure => "present"/)
+      end
+    end
+  end
+end

data/spec/poolparty/resources/service_spec.rb

@@ -0,0 +1,29 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Service" do
+  describe "instances" do
+    before(:each) do
+      @tc = TestBaseClass.new do
+        has_service("apache2", {:hasrestart => true})
+      end
+      @service = @tc.resource(:service).first
+    end
+    it "have the name in the options" do
+      @service.name.should == "apache2"
+    end
+    it "should store the hasrestart option" do
+      @service.hasrestart.should == true
+    end
+    describe "into PuppetResolver" do
+      before(:each) do
+        @compiled = PuppetResolver.new(@tc.to_properties_hash).compile
+      end
+      it "should set the filename to the name of the file" do
+        @compiled.should match(/service \{ "apache2"/)
+      end
+      it "have the mode set in the puppet output" do
+        @compiled.should match(/hasrestart => true/)
+      end
+    end
+  end
+end

data/spec/poolparty/resources/sshkey_spec.rb

@@ -0,0 +1,40 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Sshkey" do
+  describe "instances" do
+    before(:each) do
+      @key = Key.new
+      Key.stub!(:new).and_return @key
+      @key.stub!(:content).and_return "DIGITSOFTHEKEY"
+      @tc = TestBaseClass.new do
+        has_sshkey({:name => "~/.ssh/id_rsa"}) do
+          enctype "rsa"
+        end
+      end
+      @dir = @tc.resource(:sshkey).first
+    end
+    it "have the name in the options" do
+      @dir.name.should == "~/.ssh/id_rsa"
+    end
+    it "should store the owner's name" do
+      @dir.enctype.should == "rsa"
+    end
+    describe "into PuppetResolver" do
+      before(:each) do
+        @compiled = PuppetResolver.new(@tc.to_properties_hash).compile
+      end
+      it "should set the filename to the name of the file" do
+        @compiled.should match(/sshkey \{ "~\/\.ssh\/id_rsa"/)
+      end
+      it "should say it's a sshkey in the ensure method" do
+        @compiled.should match(/ensure => "present"/)
+      end
+      it "have the mode set in the puppet output" do
+        @compiled.should match(/type => "rsa"/)
+      end
+      it "have the long string of digits key" do
+        @compiled.should match(/key => "DIGITSOFTHEKEY"/)
+      end
+    end
+  end
+end

data/spec/poolparty/resources/symlink_spec.rb

@@ -0,0 +1,29 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Symlink" do
+  describe "instances" do
+    before(:each) do
+      @tc = TestBaseClass.new do
+        has_symlink("/etc/apache2/sites-enabled/poolpartyrb.com", :source => "/etc/apache2/sites-available/poolpartyrb.com")
+      end
+      @dir = @tc.resource(:symlink).first
+    end
+    it "have the name in the options" do
+      @dir.name.should == "/etc/apache2/sites-enabled/poolpartyrb.com"
+    end
+    it "should store the source name" do
+      @dir.source.should == "/etc/apache2/sites-available/poolpartyrb.com"
+    end
+    describe "into PuppetResolver" do
+      before(:each) do
+        @compiled = PuppetResolver.new(@tc.to_properties_hash).compile
+      end
+      it "should set the filename to the name of the file" do
+        @compiled.should match(/file \{ "\/etc\/apache2\/sites-enabled\/poolpartyrb\.com"/)
+      end
+      it "set the owner as the owner" do
+        @compiled.should match(/ensure => "\/etc\/apache2\/sites-available\/poolpartyrb\.com"/)
+      end
+    end
+  end
+end

data/spec/poolparty/resources/user_spec.rb

@@ -0,0 +1,48 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "User" do
+  describe "instances" do
+    before(:each) do
+      @tc = TestBaseClass.new do
+        has_user("bob", {:job => "accountant", :comment => "Bob is outstanding"}) do
+          password "b0b"
+          home "/home/bob"
+        end
+      end
+      @user = @tc.resources[:user].first
+    end
+    it "have the name in the options" do      
+      @user.name.should == "bob"
+    end
+    it "should store the owner's name" do
+      @user.comment.should == "Bob is outstanding"
+    end
+    it "should store the password (from within the block)" do
+      @user.password.should == "b0b"
+    end
+    it "should store the home" do
+      @user.home.should == "/home/bob"
+    end
+    describe "into PuppetResolver" do
+      before(:each) do
+        @compiled = PuppetResolver.new(@tc.to_properties_hash).compile
+      end
+      it "should set the filename to the name of the file" do
+        @compiled.should match(/user \{ "bob"/)
+      end
+      it "set the owner as the owner" do
+        # puts "<pre>"+@compiled.to_yaml+"</pre>"
+        @compiled.should match(/comment => "Bob is outstanding"/)
+      end
+      it "should say it's a user in the ensure method" do
+        @compiled.should match(/ensure => "present"/)
+      end
+      it "have the mode set in the puppet output" do
+        @compiled.should match(/home => "\/home\/bob"/)
+      end
+      it "set the password" do
+        @compiled.should match(/password => "b0b"/)
+      end
+    end
+  end
+end

data/spec/poolparty/resources/variable_spec.rb

@@ -0,0 +1,26 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Variables" do
+  describe "instances" do
+    before(:each) do
+      @tc = TestBaseClass.new do
+        has_variable(:name => "master", :value => "herman")
+      end
+      @variable = @tc.resource(:variable).first
+    end
+    it "have the name in the options" do
+      @variable.name.should == "master"
+    end
+    it "should store the owner's name as well" do
+      @variable.value.should == "herman"
+    end
+    describe "into PuppetResolver" do
+      before(:each) do
+        @compiled = PuppetResolver.new(@tc.to_properties_hash).compile
+      end
+      it "have the variable set in the puppet output" do
+        @compiled.should match(/\$master = "herman"/)
+      end
+    end
+  end
+end

data/spec/poolparty/schema_spec.rb

@@ -0,0 +1,53 @@
+require "rubygems"
+require "spec"
+require "json"
+require File.dirname(__FILE__) + '/../../lib/poolparty/schema'
+
+describe "Schema" do
+  it "should not fail when called with a hash" do
+    lambda {PoolParty::Schema.new({:a => "a"})}.should_not raise_error
+  end
+  it "should be able to take a string of JSON" do
+    lambda{PoolParty::Schema.new({:a => "a"}.to_json)}.should_not raise_error
+  end
+  describe "methods" do
+    before(:each) do
+      @schema = PoolParty::Schema.new :a => "b", :b => {:a => "a in b", :b => {:a => "a in b.b"}}
+    end
+    it "should be able to call a method that's in the hash on the schema" do
+      @schema.a.should == "b"
+    end
+    it "should be able to call deeply into the hash" do
+      @schema.b.a.should == "a in b"
+    end
+    it "should be able to call really deep into the hash" do
+      @schema.b.b.a.should == "a in b.b"
+    end
+  end
+  describe "json" do
+    before(:each) do
+      @schema = PoolParty::Schema.new({:a => "b", :b => {:a => "a in b", :b => {:a => "a in b.b"}}}.to_json)
+    end
+    it "should be able to call a method that's in the hash on the schema" do
+      @schema.a.should == "b"
+    end
+    it "should be able to call deeply into the hash" do
+      @schema.b.a.should == "a in b"
+    end
+    it "should be able to call really deep into the hash" do
+      @schema.b.b.a.should == "a in b.b"
+    end
+  end
+  describe "with arrays" do
+    before(:each) do
+      @schema = PoolParty::Schema.new({:a => ["b"], :b => {:a => ["a in b", "a in b.a"], :b => {:a => "a in b.b"}}}.to_json)
+    end
+    it "should have an array in a" do
+      @schema.a.class.should == Array
+      @schema.a.should == ["b"]
+    end
+    it "should have an array in b (deep)" do
+      @schema.b.a.should == ["a in b", "a in b.a"]
+    end
+  end
+end

data/spec/poolparty/services/conditional_spec.rb

@@ -0,0 +1,52 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+# STUB!
+def add_service(arg)  
+end
+
+describe "Conditional" do
+  it "not fail when calling case_of" do
+    lambda {
+      case_of "b" do
+      end
+    }.should_not raise_error
+  end
+  it "should put the case object on the case stack as a Conditional object" do
+    c = case_of "b" do
+    end
+    c.attribute.should == "b"
+  end
+  it "should populate the when_statements when a when_is is called" do
+    c = case_of "b" do
+      when_is "b" do
+      end
+    end
+    c.when_statements.size.should == 1
+  end
+  it "freeze the case statement when calling end_of" do
+    c = case_of "b" do
+      when_is "b" do
+      end
+    end
+    c.options.frozen?.should == true
+  end
+  it "create a new service for the when_is block" do
+    c = case_of "b" do
+      when_is "b" do
+        "I'm TOTALLY B!"
+      end
+    end
+    c.when_statements[:b].is_a?(Service).should == true
+  end
+  it "create a new service for the when_is block" do
+    c = case_of "b" do
+      when_is "b" do
+        "I'm TOTALLY B!"
+      end
+      when_is "c" do
+        "You are not 'c', don't lie!"
+      end
+    end
+    c.when_statements[:b].is_a?(Service).should == true
+  end
+end

data/spec/poolparty/spec/core/string_spec.rb

@@ -0,0 +1,57 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+require File.dirname(__FILE__) + '/../../../../lib/poolparty/spec/core/string'
+
+describe "Spec strings" do
+  before(:each) do
+    @str=<<-EOE
+exec { 		"install_passenger_script":
+				unless => 'test -f /etc/apache2/conf.d/passenger.conf',
+				path => '/usr/bin:/bin:/usr/local/bin:$PATH',
+				command => '/usr/bin/ruby /var/poolparty/install_passenger.rb'
+		}
+		service { 		"apache2":
+    				require => Package['apache2'],
+    				name => 'apache2',
+    				ensure => 'running'
+    		}
+    
+exec { 		"force-reload-apache2":
+				path => '/usr/bin:/bin:/usr/local/bin:$PATH',				
+				command => '/etc/init.d/apache2 force-reload',
+				refreshonly => 'true'
+		}    
+    EOE
+  end
+  it "should have the method grab_entry_for" do
+    "hi".respond_to?(:grab_entry_for).should == true
+  end
+  describe "grab_entry_for" do
+    it "should be able to grab the entry for the apache2 service" do
+      @str.grab_entry_for(:service, "apache2").gsub(/[\n\t ]+/, '').should == "service{\"apache2\":require=>Package['apache2'],name=>'apache2',ensure=>'running'}"
+    end
+    it "should be able to grab the entry for the install_passenger_script exec" do
+      @str.grab_entry_for(:exec, "install_passenger_script").gsub(/[\n\t ]+/, '').should == "exec{\"install_passenger_script\":unless=>'test-f/etc/apache2/conf.d/passenger.conf',path=>'/usr/bin:/bin:/usr/local/bin:$PATH',command=>'/usr/bin/ruby/var/poolparty/install_passenger.rb'}"
+    end
+    it "should return an empty string if there is no entry" do
+      @str.grab_entry_for(:pocket, "rangers").should == ""
+    end
+  end
+  describe "grab_key_value_for" do
+    it "should be able to say the service apache2 is set to ensure running" do
+      @str.grab_key_value_for(:service, "apache2", :ensure).should == "'running'"
+    end
+    it "should be able to say the exec force-reload-apache2 has the command '/etc/init.d/apache2 force-reload'" do
+      @str.grab_key_value_for(:exec, "force-reload-apache2", :command).should == "'/etc/init.d/apache2 force-reload'"
+    end
+    it "should be able to grab them with strings with quotes" do
+      "exec { 		\"a2enmod ssl\":
+      				command => 'a2enmod ssl',
+      				require => [ Package['apache2-common'], Package['openssl'] ],
+      				path => '/usr/bin:/bin:/usr/local/bin:$PATH',
+      				unless => '/usr/bin/test -L /etc/apache2/mods-enabled/ssl.load',
+      				notify => Exec['restart-apache2']
+      		}
+      ".grab_key_value_for(:exec, "a2enmod ssl", :notify).should == "Exec['restart-apache2']"
+    end
+  end
+end

data/spec/poolparty/spec_helper.rb

@@ -0,0 +1,263 @@
+$:.unshift(File.dirname(__FILE__) + '/../../lib')
+require 'poolparty'
+
+# TODO: FIX THE STUBS
+# %w(context matchy).each do |library|
+%w(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}
+
+ENV["POOL_SPEC"] = nil
+ENV["AWS_ACCESS_KEY"] = 'fake_access_key'
+ENV["AWS_SECRET_ACCESS_KEY"] = 'fake_aws_secret_access_key'
+
+include PoolParty
+extend PoolParty
+
+def debugging(*args); false; end
+def are_too_many_instances_running?; end
+def are_any_nodes_exceeding_minimum_runtime?; end
+def are_too_few_instances_running?; end
+
+include Remote
+require File.dirname(__FILE__)+'/net/remoter_bases/ec2_mocks_and_stubs.rb'
+
+# Append this directory - which contains a mock key named id_rsa - to the list of searchable locations 
+class Key
+  class << self
+    alias :keypair_paths_without_spec_dir :keypair_paths unless method_defined?(:keypair_paths_without_spec_dir)
+    def keypair_paths
+      [keypair_paths_without_spec_dir, File.dirname(__FILE__)].flatten
+    end
+  end
+  
+end
+
+class TestRemoterClass < ::PoolParty::Remote::Ec2
+  include CloudResourcer
+  include CloudDsl
+  
+  def ami;"ami-abc123";end
+  def size; "small";end
+  def security_group; "default";end
+  def ebs_volume_id; "ebs_volume_id";end
+  def availabilty_zone; "us-east-1a";end
+  def verbose; false; end
+  def debugging; false; end
+  def ec2
+    @ec2 ||= EC2::Base.new( :access_key_id => "not_an_access_key", :secret_access_key => "not_a_secret_access_key")
+  end
+  def describe_instances(o={})
+    response_list_of_instances
+  end
+end
+
+class TestClass < ::PoolParty::Cloud::Cloud
+  include CloudResourcer
+  include PoolParty::Remote
+  attr_accessor :parent
+  def initialize(name=:name, &block)
+    super :test_cloud, &block
+  end
+  def verbose
+    false
+  end
+end
+class TestCloud < TestClass  
+end
+
+class TestBaseClass < PoolParty::PoolPartyBaseClass
+end
+
+def setup
+  PoolParty::Messenger.stub!(:messenger_send!).and_return false  
+end
+
+def new_test_cloud(force_new=false)
+  unless @test_cloud || force_new
+    @test_cloud = TestCloud.new("test_cloud_#{rand(10000)}")
+    stub_list_from_remote_for @test_cloud
+    @test_cloud.stub!(:describe_instances).and_return response_list_of_instances
+  end
+  @test_cloud
+end
+
+def setup_cl
+  require 'poolpartycl'
+end
+
+def stub_option_load
+    @str=<<-EOS
+:access_key:    
+  3.14159
+:secret_access_key:
+  "pi"
+    EOS
+    @sio = StringIO.new
+    StringIO.stub!(:new).and_return @sio
+    Base.stub!(:open).with("http://169.254.169.254/latest/user-data").and_return @sio
+    @sio.stub!(:read).and_return @str
+    Base.reset!
+end
+
+def wait_launch(time=5)
+  pid = fork {yield}
+  wait time
+  Process.kill("INT", pid)
+  Process.wait(pid, 0)
+end
+def reset_all!
+  $cloud = nil
+end
+def read_file(path)
+  require "open-uri"
+  open(path).read
+end
+def sample_instances_list
+  @sample_instances_lister ||= [
+    {:ip => "127.0.0.1", :name => "master", :launching_time => 2.days.ago}, 
+    {:ip => "127.0.0.2", :name => "node1", :launching_time => 2.days.ago}
+  ]
+end
+
+def sample_instances
+  sample_instances_list.map {|h| PoolParty::Remote::RemoteInstance.new(h) }
+end
+def stub_list_from_local_for(o)
+  @list =<<-EOS
+  master 192.168.0.1
+  node1 192.168.0.2
+  EOS
+  @file = "filename"
+  @file.stub!(:read).and_return @list
+  o.stub!(:open).and_return @file
+
+  @ris = @list.split(/\n/).map {|line| PoolParty::Remote::RemoteInstance.new(line) }
+end
+def stub_remoter_for(o)  
+  @ec2 = EC2::Base.new( :access_key_id => "not a key",  :secret_access_key => "even more not a key")
+  EC2::Base.stub!(:new).and_return @ec2
+  
+  o.class.stub!(:ec2).and_return @ec2 
+  o.stub!(:list_of_running_instances).and_return sample_instances
+  
+  o.stub!(:list_of_instances).and_return sample_instances
+  @ec2.stub!(:run_instances).and_return true
+  @ec2.stub!(:describe_instances).and_return sample_instances
+  @ec2.stub!(:describe_instance).and_return sample_instances
+end
+def stub_list_from_remote_for(o, launch_stub=true)
+  stub_remoter_for(o)
+  o.stub!(:access_key).and_return "NOT A KEY"
+  o.stub!(:secret_access_key).and_return "NOT A SECRET"
+  # o.stub!(:list_from_remote).and_return ris
+  # o.stub!(:remote_instances_list).once.and_return ris
+  # o.stub!(:master).and_return @ris[0]
+  o.stub!(:launch_new_instance!).and_return sample_instances.first if launch_stub  
+  stub_list_of_instances_for(o)
+  stub_remoting_methods_for(o)
+end
+def stub_remoting_methods_for(o)
+  o.stub!(:other_clouds).and_return []
+  o.stub!(:expand_when).and_return "cpu > 10"
+  o.stub!(:copy_file_to_storage_directory).and_return true
+  o.stub!(:rsync_storage_files_to).and_return true
+  o.stub!(:minimum_runnable_options).and_return []
+  o.stub!(:build_and_store_new_config_file).and_return true
+  o.stub!(:process_clean_reconfigure_for!).and_return true
+  o.stub!(:before_install).and_return true
+  o.stub!(:process_install).and_return true
+  o.stub!(:after_install).and_return true
+  o.stub!(:can_contract_cloud?).and_return false
+  o.stub!(:can_expand_cloud?).and_return false
+end
+def stub_list_of_instances_for(o)  
+  o.stub!(:list_of_running_instances).once.and_return running_remote_instances
+  # o.stub!(:describe_instances).and_return response_list_of_instances
+end
+
+def stub_running_remote_instances(o)
+  o.stub!(:list_of_running_instances).and_return(running_remote_instances.map {|h| PoolParty::Remote::RemoteInstance.new(h) })
+end
+
+def response_list_of_instances(arr=[])
+  unless @response_list_of_instances
+    @a1 = stub_instance(1); 
+    @a1[:name] = "master"
+    @a2 = stub_instance(1); 
+    @a3 = stub_instance(2, "terminated"); 
+    @a4 = stub_instance(3, "pending")
+    @b1 = stub_instance(4, "shutting down", "blist"); 
+    @c1 = stub_instance(5, "pending", "clist")
+    @response_list_of_instances = [@a1, @a2, @a3, @a4, @b1, @c1]
+  end
+  @response_list_of_instances+arr
+end
+
+def running_remote_instances
+  response_list_of_instances.select {|a| a[:status] =~ /running/ }
+end
+
+def reset_response!
+  setup
+  @ris = nil
+end
+
+def add_stub_instance_to(o, num, status="running")  
+  reset_response!  
+  response_list_of_instances << stub_instance(num, status)
+  sample_instances_list << stub_instance(num, status)
+  stub_list_of_instances_for o
+  stub_remoter_for(o)
+end
+def ris
+  @ris ||= response_list_of_instances.collect {|h| PoolParty::Remote::RemoteInstance.new(h) }
+end
+def remove_stub_instance_from(o, num)
+  reset_response!
+  response_list_of_instances.reject! {|r| r if r[:name] == "node#{num}" }  
+  # o.stub!(:remote_instances_list).once.and_return ris
+end
+def stub_instance(num=1, status="running", keypair="fake_keypair")
+  {:name => "node#{num}", :ip => "192.168.0.#{num}", :status => "#{status}", :launching_time => num.minutes.ago.to_s, :keypair => "#{keypair}"}
+end
+def drop_pending_instances_for(o)
+  puts "hi"
+  o.list_of_pending_instances.stub!(:size).and_return 0
+  1
+end
+
+# Stub for messenger_send!
+class Object
+  def messenger_send!(*args)
+    true
+  end
+end
+
+class Object
+  def to_html_list
+    str = ''
+    str << "<ul>"
+    str << self.collect {|k,v| 
+      "<li>#{k} => #{(v.instance_of?(Hash) || v.instance_of?(Array)) ? v.to_html_list : v.inspect}</li> "
+      }.join(" ")
+    str << "</ul>"
+  end
+end
+
+class Array 
+    def to_html_list
+         str =''
+        str<< "<ul class='array'>"
+        str<< self.collect {|v| 
+          "<li>#{(v.is_a?(Array) || v.is_a?(Hash)) ? v.to_html_list : v.inspect}</li>"
+          }.join(' ')
+        str<<"</ul>"
+    end
+end

data/spec/poolparty/test_spec_helper.rb

@@ -0,0 +1,13 @@
+$:.unshift(File.dirname(__FILE__) + '/../../lib')
+require 'poolparty'
+
+%w(context matchy).each do |library|
+  begin
+    require library
+  rescue
+    STDERR.puts "== Cannot run test without #{library}"
+  end
+end
+
+class TestBaseClass < PoolParty::PoolPartyBaseClass  
+end

data/tasks/cloud.rake

@@ -0,0 +1,3 @@
+# Cloud tasks
+namespace(:cloud) do
+end