poolparty 0.2.84 → 1.2.2

data/spec/poolparty/spec_helper.rb

@@ -1,6 +1,8 @@
 $:.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
@@ -12,30 +14,77 @@ 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 are_too_many_instances_running?  
-end
-def are_any_nodes_exceeding_minimum_runtime?  
+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
-def are_too_few_instances_running?
+
+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
+class TestClass < ::PoolParty::Cloud::Cloud
   include CloudResourcer
+  include PoolParty::Remote
   attr_accessor :parent
-  def initialize(&block)
-    super :test_cloud, nil, &block
+  def initialize(name=:name, &block)
+    super :test_cloud, &block
   end
-  def keypair
-    "fake_keypair"
+  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
@@ -86,14 +135,21 @@ def stub_list_from_local_for(o)
   EOS
   @file = "filename"
   @file.stub!(:read).and_return @list
-  o.stub!(:get_working_listing_file).and_return @file
   o.stub!(:open).and_return @file
 
   @ris = @list.split(/\n/).map {|line| PoolParty::Remote::RemoteInstance.new(line) }
 end
-def stub_remoter_for(o)
-  o.stub!(:ec2).and_return EC2::Base.new( :access_key_id => "not a key",  :secret_access_key => "even more not a key")
-  o.stub!(:list_of_running_instances).and_return sample_instances
+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!(:instances_by_status).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)
@@ -107,8 +163,6 @@ def stub_list_from_remote_for(o, launch_stub=true)
   stub_remoting_methods_for(o)
 end
 def stub_remoting_methods_for(o)
-  o.stub!(:keypair).and_return "fake_keypair"
-  o.stub!(:keypair_path).and_return "~/.ec2/fake_keypair"
   o.stub!(:other_clouds).and_return []
   o.stub!(:expand_when).and_return "cpu > 10"
   o.stub!(:copy_file_to_storage_directory).and_return true
@@ -119,22 +173,30 @@ def stub_remoting_methods_for(o)
   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!(:keypair).and_return "fake_keypair"
-  o.stub!(:describe_instances).and_return response_list_of_instances
+  o.stub!(:instances_by_status).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!(:instances_by_status).and_return(running_remote_instances)
 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")
+    @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
+  @response_list_of_instances+arr
 end
 
 def running_remote_instances
@@ -154,7 +216,7 @@ def add_stub_instance_to(o, num, status="running")
   stub_remoter_for(o)
 end
 def ris
-  @ris ||= response_list_of_instances.collect {|h| PoolParty::Remote::RemoteInstance.new(h) }
+  @ris ||= response_list_of_instances#.collect {|h| PoolParty::Remote::RemoteInstance.new(h) }
 end
 def remove_stub_instance_from(o, num)
   reset_response!
@@ -175,4 +237,26 @@ 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/development.rake

@@ -1,75 +1,91 @@
-namespace(:dev) do
-  task :initialize do
-    Kernel.system"mkdir ~/.ec2 >/dev/null 2>/dev/null" unless File.directory?("~/.ec2")      
-  end
-  # Setup a basic development environment for the user 
-  desc "Setup development environment specify the config_file"
-  task :setup => [:initialize, :setup_keypair] do    
-    certloc = "#{Base.ec2_dir}/#{Base.keypair}/cert-*.pem 2>/dev/null"
-    pkloc = "#{Base.ec2_dir}/#{Base.keypair}/pk-*.pem 2>/dev/null"
-    unless `ls #{certloc}`.length > 1 && `ls #{pkloc}`.length > 1
-      puts <<-EOM
-Make sure you run rake dev:setup_pemkeys before you run this command
-
-I cannot continue until your keys are setup. 
-exiting...
-      EOM
-      exit
+namespace :poolparty do
+  namespace(:ec2) do
+    
+    task :initialize do
+      Kernel.system"mkdir ~/.ec2 >/dev/null 2>/dev/null" unless File.directory?("~/.ec2")      
     end
-    keyfilename = ".#{Base.keypair}_pool_keys"
-    run <<-EOR
-      echo 'export AWS_ACCESS_KEY=\"#{Base.access_key}\"' > $HOME/#{keyfilename}
-      echo 'export AWS_SECRET_ACCESS=\"#{Base.secret_access_key}\"' >> $HOME/#{keyfilename}
-      echo 'export EC2_HOME=\"#{Base.ec2_dir}\"' >> $HOME/#{keyfilename}
-      echo 'export KEYPAIR_NAME=\"#{Base.keypair}\"' >> $HOME/#{keyfilename}
-      echo 'export EC2_PRIVATE_KEY=`ls ~/.ec2/#{Base.keypair}/pk-*.pem`;' >> $HOME/#{keyfilename}
-      echo 'export EC2_CERT=`ls ~/.ec2/#{Base.keypair}/cert-*.pem`;' >> $HOME/#{keyfilename}
-    EOR
-    puts <<-EOM
-To work on this cloud, source the file like: 
+    # Setup a basic development environment for the user 
+    
+    desc "Setup development environment specify the config_file"
+    task :setup => [:initialize, :setup_keypair] do    
+      certloc = "#{Default.ec2_dir}/#{Default.keypair}/cert-*.pem 2>/dev/null"
+      pkloc = "#{Default.ec2_dir}/#{Default.keypair}/pk-*.pem 2>/dev/null"
+      unless `ls #{certloc}`.length > 1 && `ls #{pkloc}`.length > 1
+        puts <<-EOM
+  Make sure you run rake dev:setup_pemkeys before you run this command
+
+  I cannot continue until your keys are setup. 
+  exiting...
+        EOM
+        exit
+      end
+      keyfilename = ".#{Default.keypair}_pool_keys"
+      run <<-EOR
+        echo 'export AWS_ACCESS_KEY=\"#{Default.access_key}\"' > $HOME/#{keyfilename}
+        echo 'export AWS_SECRET_ACCESS=\"#{Default.secret_access_key}\"' >> $HOME/#{keyfilename}
+        echo 'export EC2_HOME=\"#{Default.ec2_dir}\"' >> $HOME/#{keyfilename}
+        echo 'export KEYPAIR_NAME=\"#{Default.keypair}\"' >> $HOME/#{keyfilename}
+        echo 'export EC2_PRIVATE_KEY=`ls ~/.ec2/#{Default.keypair}/pk-*.pem`;' >> $HOME/#{keyfilename}
+        echo 'export EC2_CERT=`ls ~/.ec2/#{Default.keypair}/cert-*.pem`;' >> $HOME/#{keyfilename}
+      EOR
+      puts <<-EOM
+  To work on this cloud, source the file like: 
   
-  source $HOME/#{keyfilename}
+    source $HOME/#{keyfilename}
   
-    EOM
-  end
-  desc "Generate a new keypair"
-  task :setup_keypair => [:initialize] do
-    Base.keypair ||= "#{File.basename(Dir.pwd)}"    
-    run "ec2-delete-keypair #{Base.keypair}" if File.file?(Base.keypair_path)
-    puts "-- setting up keypair named #{Base.keypair} in #{Base.keypair_path}"
-    run <<-EOR        
-      chmod 600 #{Base.keypair_path} 2>/dev/null
-      mkdir ~/.ec2/#{Base.keypair} 2>/dev/null
-      ec2-add-keypair #{Base.keypair} > #{Base.keypair_path}
-    EOR
-  end
-  desc "Setup pem keys"
-  task :setup_pemkeys => [:initialize] do    
-    puts "Setting up stubbed pem keys in ~/.ec2/#{Base.keypair}"
-    run <<-EOR
-      mkdir -p ~/.ec2/#{Base.keypair} 2>/dev/null
-      echo 'UPDATE ME' > #{Base.ec2_dir}/#{Base.keypair}/cert-UPDATEME.pem
-      echo 'UPDATE ME' > #{Base.ec2_dir}/#{Base.keypair}/pk-UPDATEME.pem
-    EOR
-    puts "Don't forget to replace your ~/.ec2/#{Base.keypair}/*.pem keys with the real amazon keys"
-  end
-  desc "initialize setup"
-  task :init => [:setup_pemkeys]
-  desc "Authorize base ports for application"
-  task :authorize_ports => :initialize do
-    run <<-EOR
-      ec2-authorize -p 22 default
-      ec2-authorize -p 80 default
-    EOR
-  end
-  desc "Turn the gemspec into a yaml file"
-  task :gemspec_to_yaml => [:initialize, :gemspec] do
-    filepath = ::File.join(::File.dirname(__FILE__), "..", "poolparty.gemspec")
-    data = open(filepath).read
-    spec = eval("$SAFE = 3\n#{data}")
-    yml = YAML.dump spec
-    File.open(filepath, "w+") do |f|
-      f << yml
+      EOM
+    end
+    
+    desc "Generate a new keypair"
+    task :setup_keypair => [:initialize] do
+      Default.keypair ||= "#{File.basename(Dir.pwd)}"    
+      run "ec2-delete-keypair #{Default.keypair}" if File.file?(Default.keypair_path)
+      puts "-- setting up keypair named #{Default.keypair} in #{Default.keypair_path}"
+      run <<-EOR        
+        chmod 600 #{Default.keypair_path} 2>/dev/null
+        mkdir ~/.ec2/#{Default.keypair} 2>/dev/null
+        ec2-add-keypair #{Default.keypair} > #{Default.keypair_path}
+      EOR
+    end
+    
+    desc "Setup pem keys"
+    task :setup_pemkeys => [:initialize] do    
+      puts "Setting up stubbed pem keys in ~/.ec2/#{Default.keypair}"
+      run <<-EOR
+        mkdir -p ~/.ec2/#{Default.keypair} 2>/dev/null
+        echo 'UPDATE ME' > #{Default.ec2_dir}/#{Default.keypair}/cert-UPDATEME.pem
+        echo 'UPDATE ME' > #{Default.ec2_dir}/#{Default.keypair}/pk-UPDATEME.pem
+      EOR
+      puts "Don't forget to replace your ~/.ec2/#{Default.keypair}/*.pem keys with the real amazon keys"
     end
+    
+    desc "initialize setup"
+    task :init => [:setup_pemkeys]
+    desc "Authorize base ports for application"
+    task :authorize_ports => :initialize do
+      run <<-EOR
+        ec2-authorize -p 22 default
+        ec2-authorize -p 80 default
+      EOR
+    end
+    
+    desc "Turn the gemspec into a yaml file"
+    task :gemspec_to_yaml => [:initialize, :gemspec] do
+      filepath = ::File.join(::File.dirname(__FILE__), "..", "poolparty.gemspec")
+      data = open(filepath).read
+      spec = eval("$SAFE = 3\n#{data}")
+      yml = YAML.dump spec
+      File.open(filepath, "w+"){ |f|f << yml }
+    end
+    
+  end
+end
+
+namespace :ctags do
+  desc "create ctags"
+  task :create do
+    dir = File.dirname(__FILE__) + "/.."
+    Dir.chdir(dir)
+    sh "time ctags -R --language-force=Ruby -F ./TAGS ."
   end
 end

data/tasks/poolparty.rake

@@ -0,0 +1,54 @@
+desc "Run the specs"
+task :slow_spec do
+  stats = {:example=>0, :failures=>0, :pending=>0}
+  Dir["#{::File.dirname(__FILE__)}/../spec/poolparty/**/*_spec.rb"].each do |sp|
+    puts "---------------- #{::File.basename(sp)} ----------------"
+    results = `spec #{sp}`
+    results.match(/([1-9]+0?)\sfailures|errors/)
+    stats[:failures] += $i.to_i
+    puts results
+  end
+  puts "#{stats[:failures]} total errors"
+end
+namespace(:poolparty) do
+  namespace(:setup) do
+    desc "Generate a manifest for quicker loading times"
+    task :manifest do
+      $GENERATING_MANIFEST = true
+      out = capture_stdout do
+        $_poolparty_load_directories.each do |dir|
+          PoolParty.require_directory ::File.join(::File.dirname(__FILE__), '../lib/poolparty', dir)
+        end
+      end
+      ::File.open(::File.join(::File.dirname(__FILE__), '../config', "manifest.pp"), "w+") {|f| f << out.map {|f| "#{f}"} }
+      puts "Manifest created"
+    end
+  end
+  namespace :vendor do
+    desc "Initialize the submodules"
+    task :setup do
+      `git submodule init`
+    end
+    desc "Update the submodules"
+    task :update do
+      Dir["#{::File.dirname(__FILE__)}/../vendor/gems/*"].each do |dir|
+        puts "Fetching #{dir}..."
+        `cd #{dir} && git fetch && git rebase origin/master`
+      end
+      # `git submodule update`
+    end
+  end
+  namespace :deps do
+    task :clean_gem_cache do
+      gem_location = "#{::File.dirname(__FILE__)}/../vendor/dependencies"
+      cache_dir = "#{gem_location}/cache"
+      Dir["#{cache_dir}/*.gem"].each {|file| ::File.unlink file }
+    end
+    desc "Update dependencies gem"
+    task :update => [:clean_gem_cache] do
+      gem_location = "#{::File.dirname(__FILE__)}/../vendor/dependencies"
+      Suitcase::Zipper.gems open("#{gem_location}/gems_list").read.split("\n"), gem_location
+      Suitcase::Zipper.packages ['http://rubyforge.org/frs/download.php/45905/rubygems-1.3.1.tgz'], gem_location
+    end
+  end
+end

data/tasks/server.rake

@@ -1,42 +1,44 @@
 # Tasks to be run on the server
-namespace(:server) do                
-  task :init  do
-    PoolParty::Coordinator.init(false)
-  end
-  # bundle, upload and register your bundle on the server
-  desc "Bundle, upload and register your ami"
-  task :all => [:bundle, :upload, :register] do
-    puts "== your ami is ready"
-  end
-  # Cleanup the /mnt directory
-  desc "Clean the /mnt directory"
-  task :clean_mnt do
-    `rm -rf /mnt/image* img*`
-  end
-  # Before we can bundle, we have to make sure we have the cert and pk files
-  desc "Ensure the required bundle files are present in /mnt"
-  task :check_bundle_files do
-    raise Exception.new("You must have a private key in your /mnt directory") unless File.exists?("/mnt/pk-*.pem")
-    raise Exception.new("You must have your access key in your /mnt directory") unless File.exists?("/mnt/cert-*.pem")          
-  end
-  # Bundle the image
-  desc "Bundle this image into the /mnt directory"
-  task :bundle => [:clean_mnt, :check_bundle_files] do
-    puts `ec2-bundle-vol -k /mnt/pk-*.pem -u '#{Planner.user_id}' -d /mnt -c /mnt/cert-*.pem -r i386`
-  end
-  # Upload the bundle into the app_name bucket
-  desc "Upload the bundle to your bucket with a unique name: deletes old ami"
-  task :upload => [:init, :delete_bucket] do
-    puts `ec2-upload-bundle -b #{Planner.app_name} -m /mnt/image.manifest.xml -a #{Planner.access_key} -s #{Planner.secret_access_key}`
-  end
-  # Register the bucket with amazon and get back an ami
-  desc "Register the bundle with amazon"
-  task :register do
-    puts `ec2-register -K /mnt/pk-*.pem -C /mnt/cert-*.pem #{Planner.app_name}/image.manifest.xml`
-  end
-  # Delete the bucket
-  desc "Delete the bucket with the bundle under tha app name"
-  task :delete_bucket do
-    Planner.app_name.delete_bucket 
+namespace :poolparty do
+  namespace :server do                
+    task :init  do
+      PoolParty::Coordinator.init(false)
+    end
+    # bundle, upload and register your bundle on the server
+    desc "Bundle, upload and register your ami"
+    task :all => [:bundle, :upload, :register] do
+      puts "== your ami is ready"
+    end
+    # Cleanup the /mnt directory
+    desc "Clean the /mnt directory"
+    task :clean_mnt do
+      `rm -rf /mnt/image* img*`
+    end
+    # Before we can bundle, we have to make sure we have the cert and pk files
+    desc "Ensure the required bundle files are present in /mnt"
+    task :check_bundle_files do
+      raise Exception.new("You must have a private key in your /mnt directory") unless File.exists?("/mnt/pk-*.pem")
+      raise Exception.new("You must have your access key in your /mnt directory") unless File.exists?("/mnt/cert-*.pem")          
+    end
+    # Bundle the image
+    desc "Bundle this image into the /mnt directory"
+    task :bundle => [:clean_mnt, :check_bundle_files] do
+      puts `ec2-bundle-vol -k /mnt/pk-*.pem -u '#{Planner.user_id}' -d /mnt -c /mnt/cert-*.pem -r i386`
+    end
+    # Upload the bundle into the app_name bucket
+    desc "Upload the bundle to your bucket with a unique name: deletes old ami"
+    task :upload => [:init, :delete_bucket] do
+      puts `ec2-upload-bundle -b #{Planner.app_name} -m /mnt/image.manifest.xml -a #{Planner.access_key} -s #{Planner.secret_access_key}`
+    end
+    # Register the bucket with amazon and get back an ami
+    desc "Register the bundle with amazon"
+    task :register do
+      puts `ec2-register -K /mnt/pk-*.pem -C /mnt/cert-*.pem #{Planner.app_name}/image.manifest.xml`
+    end
+    # Delete the bucket
+    desc "Delete the bucket with the bundle under tha app name"
+    task :delete_bucket do
+      Planner.app_name.delete_bucket 
+    end
   end
 end

data/tasks/spec.rake

@@ -1,7 +1,11 @@
 require 'rake'
 require 'spec/rake/spectask'
+require 'rake/testtask'
+require 'rake/rdoctask'
 
-task :default  => :spec
+task :default  => [:spec, :test]
+desc "Update vendor directory and run tests"
+task :ci => ["poolparty:vendor:update", :spec, :test]
 
 Spec::Rake::SpecTask.new(:spec) do |t|
   t.warning = t.rcov = false
@@ -13,4 +17,34 @@ Spec::Rake::SpecTask.new(:spec_v) do |t|
   t.rcov_opts = ['--exclude', 'gems*,spec*,website*,test*']
   t.ruby_opts = []
   t.spec_files = Dir["spec/**/*_spec.rb"]
+end
+
+# task :test do
+#   sh "ruby -Ilib:test #{Dir["#{File.dirname(__FILE__)}/../test/poolparty/*/*.rb"].join(" ")}"
+# end
+
+Rake::TestTask.new(:test) do |t|
+  t.test_files = FileList['test/poolparty/**/*_test.rb']
+  t.warning = false
+  t.verbose = false
+end
+
+begin
+  require 'rcov/rcovtask'
+ 
+  Rcov::RcovTask.new(:rcov) do |t|
+    t.libs << FileList['lib/poolparty/**/*.rb']
+    dot_rakes = 
+    t.rcov_opts = [
+      '-xRakefile', '-xrakefile',
+      '-xlib/erlang',
+      '--text-report',
+      '--sort coverage'
+    ] + FileList['tasks/*.rake'].pathmap("-x%p")
+    t.test_files = FileList['test/poolparty/**/*_test.rb']
+    t.output_dir = 'coverage'
+    t.verbose = true
+  end
+rescue LoadError
+  puts "RCov is not available"
 end

data/test/fixtures/test_template.erb

@@ -0,0 +1 @@
+Hello <%= friends %>

data/test/poolparty/core/array_test.rb

@@ -0,0 +1,21 @@
+require "#{::File.dirname(__FILE__)}/../../test_helper"
+
+class TestWrappingArray < Test::Unit::TestCase
+  context "wrapping_next" do
+    setup do
+      @array = %w(a b c d)
+    end
+
+    should "return element next in the chain" do
+      @array.wrapping_next("a").should == "b"
+      @array.wrapping_next("b").should == "c"
+      @array.wrapping_next("c").should == "d"
+      @array.wrapping_next("d").should == "a"
+      @array.wrapping_next("a").should == "b"
+    end
+    should "raise if the element is not in the array" do
+      lambda {@array.wrapping_next("z")}.should raise_error
+    end
+  end
+  
+end

data/test/poolparty/core/hash_test.rb

@@ -0,0 +1,34 @@
+require "#{::File.dirname(__FILE__)}/../../test_helper"
+
+class TestHashClass < Test::Unit::TestCase
+  context "hash_get" do
+    setup do
+      @hsh = {
+        "10.0.0.3" => {"stuff" => "here"},
+        "10.0.0.1" => {"stuff" => "here"},
+        "10.0.0.2" => {"stuff" => "here"}
+      }
+    end
+    should "should return 0.1 if we are 0.2" do
+      assert @hsh.next_sorted_key("10.0.0.1"), "10.0.0.2"
+      assert @hsh.next_sorted_key("10.0.0.2"), "10.0.0.3"
+      assert @hsh.next_sorted_key("10.0.0.3"), "10.0.0.1"
+      assert @hsh.next_sorted_key("10.0.0.1"), "10.0.0.2"
+    end
+    should "should return self if there is only 1 element" do
+      k = {"10.0.0.2" => {"stuff" => "here"}}.next_sorted_key("10.0.0.2")
+      assert k, "10.0.0.2"
+    end
+  end
+  context "test method_missing" do
+    should "should be able to call a key on the hash as a method" do
+      {:first_name => "bob", :last_name => "frank"}.first_name.should == "bob"
+    end
+    should "should not return nil if there is no key set in the hash" do
+      hsh = {:first_name => "bob", :last_name => "frank"}
+      lambda {{:first_name => "bob", :last_name => "frank"}.neighbor}.should raise_error
+    end
+  end
+  
+  
+end

data/test/poolparty/core/string_test.rb

@@ -0,0 +1,29 @@
+require "#{::File.dirname(__FILE__)}/../../test_helper"
+
+class TestString < Test::Unit::TestCase
+  context "to_hash" do
+    setup do
+      @string =<<-EOS
+        a = "a"
+        b = "b"
+        c = "c"
+      EOS
+    end
+
+    should "turn it into a hash with the values from =" do
+      @string.to_hash[:a].should == "a"
+      @string.to_hash[:b].should == "b"
+      @string.to_hash[:c].should == "c"
+    end
+  end
+  context "macify" do
+    setup do
+      @mac = "00:0c:29:44:f1:0f"
+    end
+
+    should "should turn the mac into the response on the command-line" do
+      @mac.macify.should == "0:c:29:44:f1:f"
+    end
+  end
+  
+end