solutious-rudy 0.4.0 → 0.5.0

data/test/50_commands/00_setup_test.rb

@@ -0,0 +1,11 @@
+require File.join(File.dirname(__FILE__), '..', 'helper')
+
+module Rudy::Test
+  class Case_50_Commands < Test::Unit::TestCase
+    include Rudy::AWS  
+    
+    @@logger = StringIO.new
+    
+
+  end
+end

data/test/50_commands/10_keypairs_test.rb

@@ -0,0 +1,79 @@
+
+module Rudy::Test
+  class Case_50_Commands
+    
+
+    context "#{name}_10 KeyPairs" do
+      setup do
+        @rkey = Rudy::KeyPairs.new(:logger => @@logger)
+        # So we can test with and without user specified keypairs
+        # (stage should have keypairs configured in ./.rudy/config or Rudyfile)
+        @rkey.global.environment = :test
+        stop_test !@rkey.is_a?(Rudy::KeyPairs), "We need Rudy::KeyPairs (#{@rkey})"
+      end
+      
+      teardown do
+        #if @@logger && @@logger.is_a?(StringIO)
+        #  @@logger.rewind
+        #  puts @@logger.read
+        #end
+      end
+      
+      should "(10) create a keypair" do
+        stop_test @rkey.any?, "Delete existing keypairs"
+        kp = @rkey.create
+        assert kp.is_a?(Rudy::AWS::EC2::KeyPair)
+        assert @rkey.exists?(kp.name), "KeyPair not registered with Amazon"
+        assert File.exists?(@rkey.path), "No private key: #{@rkey.path}"
+        assert File.exists?(@rkey.public_path), "No public key: #{@rkey.public_path}"
+      end
+      
+      should "(11) create a keypair with an arbitrary name" do
+        n = "test-%s" % Rudy::Utils.strand
+        kp = @rkey.create(n)
+        assert kp.is_a?(Rudy::AWS::EC2::KeyPair)
+        assert @rkey.exists?(kp.name), "KeyPair not registered with Amazon"
+        assert File.exists?(@rkey.path(n)), "No private key: #{@rkey.path(n)}"
+        assert File.exists?(@rkey.public_path(n)), "No public key: #{@rkey.public_path(n)}"
+      end
+      
+      should "(12) not create keypair if one exists" do
+        assert @rkey.exists?, "No #{@rkey.name} KeyPair"
+        begin; kp = @rkey.create; rescue; end # Quiet, you!
+        assert kp.nil?, "Keypair was still created"
+      end
+      
+      should "(13) not create keypair if a root keypair is defined in config" do
+        assert_equal (@rkey.global.environment = :keypairtest), :keypairtest # stage needs keypairs configured
+        assert @rkey.has_root_keypair?, "No root keypair path defined in config"
+        begin; kp = @rkey.create; rescue => ex; end
+        assert kp.nil?, "Keypair was still created"
+      end
+      
+      should "(20) list keypairs" do
+        # We assume we've created more than one keypair before this point
+        assert @rkey.any?, "No keypairs"
+        assert (kp_list = @rkey.list).is_a?(Array), "List not an Array"
+        assert kp_list.size > 1, "List not greater than 1 (#{kp_list.size})"
+        assert (kp_hash = @rkey.list_as_hash).is_a?(Hash), "List not a Hash"
+      end
+      
+      should "(30) find existing keypair for current machine group" do
+        assert !(test_kp = @rkey.user_keypairpath(:root)).nil?, "Test keypair is nil"
+        @rkey.global.environment = :keypairtest # user_keypairpath will now look at the stage
+        assert !(stage_kp = @rkey.user_keypairpath(:root)).nil?, "keypairtest keypair is nil"
+        assert test_kp != stage_kp, "Test and Stage keypairs are the same (#{test_kp}, #{stage_kp})"
+      end
+      
+      should "(99) destroy keypairs" do
+        assert @rkey.any?, "No keypairs registered"
+        @rkey.list.each { |kp| assert @rkey.destroy(kp.name), "Not destroyed #{kp.name}" }
+        assert !File.exists?(@rkey.path), "Still exists: #{@rkey.path}"
+        assert !File.exists?(@rkey.public_path), "Still exists: #{@rkey.public_path}"
+      end
+      
+      
+    end
+  
+  end
+end

data/test/50_commands/20_groups_test.rb

@@ -0,0 +1,77 @@
+
+module Rudy::Test
+  class Case_50_Commands
+    
+
+    context "#{name}_20 Groups" do
+      setup do
+        @rgroup = Rudy::Groups.new(:logger => @@logger)
+        stop_test !@rgroup.is_a?(Rudy::Groups), "We need Rudy::Groups (#{@rgroup})"
+      end
+      
+      teardown do
+        #if @@logger && @@logger.is_a?(StringIO)
+        #  @@logger.rewind
+        #  puts @@logger.read
+        #end
+      end
+      
+      should "(10) create a group" do
+        stop_test @rgroup.any?, "Delete existing Groups"
+        group = @rgroup.create
+        assert group.is_a?(Rudy::AWS::EC2::Group)
+        assert @rgroup.exists?(group.name), "Group not registered with Amazon"
+        assert_equal group.name, @rgroup.name
+        # We don't check the rules because we do that in the lower
+        # level tests for Rudy::AWS::EC2::Groups. 
+      end
+      
+      should "(11) create a group with an arbitrary name" do
+        n = "test-%s" % Rudy::Utils.strand
+        group = @rgroup.create(n)
+
+        assert group.is_a?(Rudy::AWS::EC2::Group)
+        assert @rgroup.exists?(group.name), "Group #{group.name} not registered with Amazon"
+        assert_equal group.name, n
+      end
+      
+      
+      should "(20) list groups" do
+        assert @rgroup.any?, "No groups"
+        assert @rgroup.exists?, "No #{@rgroup.name} group"
+        
+        kp_list = @rgroup.list
+        assert kp_list.is_a?(Array), "List not an Array"
+        
+        kp_hash = @rgroup.list_as_hash
+        assert kp_hash.is_a?(Hash), "List not a Hash"
+      end
+      
+      should "(30) not create group if one exists" do
+        assert @rgroup.exists?, "No #{@rgroup.name} Group"
+        begin
+          group = @rgroup.create
+        rescue # Quiet, you!
+        end
+        assert group.nil?, "Group was still created"
+      end
+      
+      xshould "(40) modify group rules" do
+        
+      end
+      
+      should "(99) destroy groups" do
+        assert @rgroup.exists?, "Group #{@rgroup.name} doesn't exist"
+        @rgroup.list.each do |group|
+          next if group.name == 'default'
+          assert @rgroup.destroy(group.name), "Did not destroy #{group.name}"
+        end
+        assert !@rgroup.exists?, "Group #{@rgroup.name} still exists"
+      end
+      
+      
+    end
+  
+  
+  end
+end

data/test/50_commands/40_volumes_test.rb

@@ -0,0 +1,55 @@
+
+
+module Rudy::Test
+  class Case_50_Commands
+    
+    context "#{name}_40 Volumes" do
+      
+      setup do
+        @rvol = Rudy::Volumes.new(:logger => @@logger)
+        stop_test !@rvol.is_a?(Rudy::Volumes), "We need Rudy::Volumes (#{@rvol})"
+      end
+      
+      should "(10) create a volume" do
+        volume_size = 2
+        #stop_test @rvol.any?(:available), "Destroy existing volumes"
+        volume = @rvol.create(volume_size)
+        assert volume.is_a?(Rudy::AWS::EC2::Volume), "Not a Volume"
+        # Should use the global zone by default
+        assert_equal @rvol.global.zone.to_s, volume.zone, "Zone incorrect: #{volume.zone}"
+        assert_equal volume_size.to_i, volume.size.to_i, "Size incorrect: #{volume.size}"
+        # Rudy::Volume should wait until the volume is available before returning
+        assert volume.available?, "Volume not available (#{volume.status})"
+      end
+      
+      should "(11) list a volume as available" do
+        
+      end
+      
+      should "(20) list volumes" do
+         volume_list = @rvol.list
+         assert volume_list.is_a?(Array), "Not an Array"
+         assert volume_list.size > 0, "No Volumes in Array"
+
+         volume_hash = @rvol.list_as_hash
+         assert volume_hash.is_a?(Hash), "Not a Hash"
+         assert volume_hash.keys.size > 0, "No Volumes in Hash"
+
+         assert_equal volume_list.size.to_i, volume_hash.keys.size.to_i, "Hash and Array not equal size"
+       end
+       
+      should "(90) destroy volumes" do
+        assert @rvol.any?, "No volumes"
+        volume_list = @rvol.list
+        volume_list.each do |vol|
+          next unless vol.available?
+          ret = @rvol.destroy(vol.awsid)
+          assert ret, "Not destroyed (#{ret} for #{vol.awsid})"
+          assert @rvol.deleting?(vol.awsid), "Volume not in deleting state (#{vol.awsid}: #{vol.state})"
+        end
+      end
+    end
+    
+  end
+end
+    

data/test/50_commands/50_instances_test.rb

@@ -0,0 +1,110 @@
+
+module Rudy::Test
+  class Case_50_Commands
+    
+    context "#{name}_50 Instances" do
+      
+      setup do
+        
+        @rmach = Rudy::Instances.new(:logger => @@logger)
+        stop_test !@rmach.is_a?(Rudy::Instances), "We needRudy::Instancess (#{@rmach})"
+        
+        @rgroup = Rudy::Groups.new(:logger => @@logger)
+        stop_test !@rgroup.is_a?(Rudy::Groups), "We need Rudy::Instances (#{@rgroup})"
+
+        @rkey = Rudy::KeyPairs.new(:logger => @@logger)
+        stop_test !@rkey.is_a?(Rudy::KeyPairs), "We need Rudy::KeyPairs (#{@rkey})"
+        
+        @rvol = Rudy::Volumes.new(:logger => @@logger)
+        stop_test !@rvol.is_a?(Rudy::Volumes), "We need Rudy::Volumes (#{@rvol})"
+        
+        @rkey.global.environment = :test
+        @rgroup.global.environment = :test
+        @rmach.global.environment = :test
+      end
+      
+      should "(01) have a test keypair" do
+        kp = @rkey.create
+        assert kp.is_a?(Rudy::AWS::EC2::KeyPair), "No keypair for #{@rkey.current_machine_group}"
+        assert File.exists?(@rkey.path), "No private key"
+      end
+      
+      should "(05) have a security group" do
+        stop_test @rgroup.exists?(@rmach.current_machine_group), "Destroy existing groups first"
+        @rgroup.any?
+        
+        assert @rmach.current_machine_group.is_a?(String), "No current machine group defined"
+        
+        group = @rgroup.create
+        
+        assert group.is_a?(Rudy::AWS::EC2::Group), "Not a Group"
+        assert @rgroup.exists?(@rmach.current_machine_group), "No matching security group"
+      end
+      
+      
+      should "(10) create an instance" do
+        stop_test @rmach.running?, "Shutdown the instances running in #{@rmach.current_machine_group}"
+        instances = @rmach.create
+        assert instances.is_a?(Array), "instances is not an Array"
+        assert instances.first.is_a?(Rudy::AWS::EC2::Instance), "instance is not a Rudy::AWS::EC2::Instance (#{instances.first.class})"
+        assert_equal 1, instances.size, "#{instances.size} instances were started"
+      end
+      
+      
+      should "(20) list instances in machine group" do
+        assert @rmach.running?, "No machines running"
+        instances = @rmach.list(:running)
+        assert instances.is_a?(Array), "instances is not an Array"
+        assert instances.first.is_a?(Rudy::AWS::EC2::Instance), "instance is not a Rudy::AWS::EC2::Instance"
+      end
+
+      should "(30) check console output" do
+        assert @rmach.console.is_a?(String), "No console output"
+      end
+      
+      should "(45) attach volume to instance and then detach it" do
+        volume = @rvol.create(1)
+        #volume = @rvol.get('vol-9934d4f0')
+        instances = @rmach.list(:running)
+        assert !instances.empty?, "No instances running"
+        instance = instances.first
+        assert !volume.attached?, "Volume is attached"
+        assert instance.running?, "Instance not running"
+        assert volume.available?, "Volume not available"
+        assert @rvol.attach(volume, instance), "Volume #{volume.awsid} not attached to #{instance.awsid}"
+        assert @rvol.detach(volume), "Volume not detached (#{volume.awsid})"
+        assert @rvol.destroy(volume), "Volume not destroyed (#{volume.awsid})"
+      end
+      
+      should "(90) destroy instances" do
+        assert @rmach.running?, "No instances running"
+        assert @rmach.destroy, "Instances not destroyed"
+      end
+      
+      should "(95) destroy security group" do
+        # We can't delete the machine group until all instances are terminated
+        Rudy.waiter(2, 60, @@logger) { !@@ec2.instances.any?(:running) }
+        @rgroup.list do |group|
+          next if group.name == 'default' # The default group is invisible
+          assert @rgroup.destroy(group.name), "Did not destroy #{group.name}"
+        end
+      end
+      
+      
+      should "(96) destroy test keypair" do
+        # We can't delete the keypair until all instances are terminated
+        Rudy.waiter(2, 60, @@logger) { !@@ec2.instances.any?(:running) }
+        assert @rkey.destroy, "Keypair (#{@rkey.name}) not destroyed"
+      end
+      
+    end
+    
+
+  end
+end
+
+
+# assert_equal 'John Doe', @user.full_name
+# assert_same_elements([:a, :b, :c], [:c, :a, :b])
+# assert_contains(['a', '1'], /\d/)
+# assert_contains(['a', '1'], 'a')

data/test/coverage.txt

@@ -0,0 +1,51 @@
+Finished in 134.699353 seconds.
+
+81 tests, 200 assertions, 0 failures, 0 errors
++----------------------------------------------------+-------+-------+--------+
+|                  File                              | Lines |  LOC  |  COV   |
++----------------------------------------------------+-------+-------+--------+
+|lib/annoy.rb                                        |   227 |   131 |  42.0% |
+|lib/aws_sdb.rb                                      |     3 |     2 | 100.0% |
+|lib/aws_sdb/error.rb                                |    42 |    33 |  66.7% |
+|lib/aws_sdb/service.rb                              |   215 |   192 |  96.4% |
+|lib/console.rb                                      |   401 |   263 |  35.7% |
+|lib/rudy.rb                                         |   275 |   186 |  57.5% |
+|lib/rudy/addresses.rb                               |    22 |    14 |  57.1% |
+|lib/rudy/aws.rb                                     |    72 |    47 |  91.5% |
+|lib/rudy/aws/ec2.rb                                 |    28 |    23 | 100.0% |
+|lib/rudy/aws/ec2/address.rb                         |   111 |    80 |  87.5% |
+|lib/rudy/aws/ec2/group.rb                           |   237 |   150 |  86.7% |
+|lib/rudy/aws/ec2/image.rb                           |    46 |    33 |  21.2% |
+|lib/rudy/aws/ec2/instance.rb                        |   283 |   180 |  71.1% |
+|lib/rudy/aws/ec2/keypair.rb                         |    90 |    71 |  93.0% |
+|lib/rudy/aws/ec2/snapshot.rb                        |    87 |    63 |  87.3% |
+|lib/rudy/aws/ec2/volume.rb                          |   184 |   127 |  60.6% |
+|lib/rudy/aws/s3.rb                                  |     3 |     2 | 100.0% |
+|lib/rudy/aws/simpledb.rb                            |    71 |    54 | 100.0% |
+|lib/rudy/backups.rb                                 |    10 |     5 | 100.0% |
+|lib/rudy/config.rb                                  |   113 |    72 |  91.7% |
+|lib/rudy/disks.rb                                   |     9 |     5 | 100.0% |
+|lib/rudy/groups.rb                                  |   113 |    80 |  82.5% |
+|lib/rudy/huxtable.rb                                |   212 |   152 |  54.6% |
+|lib/rudy/keypairs.rb                                |   100 |    78 | 100.0% |
+|lib/rudy/machines.rb                                |   193 |   142 |   9.2% |
+|lib/rudy/manager.rb                                 |    62 |    43 |  16.3% |
+|lib/rudy/metadata/backup.rb                         |   135 |   100 |  73.0% |
+|lib/rudy/metadata/disk.rb                           |   149 |   117 |  47.0% |
+|lib/rudy/metadata/machine.rb                        |    34 |    23 |  82.6% |
+|lib/rudy/routines.rb                                |    54 |    26 |  23.1% |
+|lib/rudy/routines/disk_handler.rb                   |   190 |   133 |   5.3% |
+|lib/rudy/routines/release.rb                        |    15 |     8 | 100.0% |
+|lib/rudy/routines/script_runner.rb                  |    65 |    47 |  10.6% |
+|lib/rudy/routines/shutdown.rb                       |    42 |    30 |  13.3% |
+|lib/rudy/routines/startup.rb                        |    48 |    33 |  12.1% |
+|lib/rudy/scm/svn.rb                                 |    68 |    50 |  26.0% |
+|lib/rudy/utils.rb                                   |   118 |    75 |  54.7% |
+|lib/rudy/volumes.rb                                 |   108 |    84 |  11.9% |
+|lib/storable.rb                                     |   284 |   185 |  43.8% |
+|lib/sysinfo.rb                                      |   255 |   154 |  35.7% |
+|lib/utils/crypto-key.rb                             |    93 |    36 |  55.6% |
++----------------------------------------------------+-------+-------+--------+
+|Total                                               |  4867 |  3329 |  55.6% |
++----------------------------------------------------+-------+-------+--------+
+55.6%   41 file(s)   4867 Lines   3329 LOC

data/test/helper.rb

@@ -0,0 +1,35 @@
+require 'rubygems' if defined? Gem
+ 
+libdir = File.join(File.dirname(__FILE__), '..', 'lib')
+$LOAD_PATH.unshift libdir unless $LOAD_PATH.include?(libdir)
+ 
+require 'test/unit'
+require 'shoulda'
+
+require 'rudy'
+
+
+puts Rudy.banner("THIS IS RUBY #{RUBY_VERSION}")
+
+# +should_stop+ should the test be stopped?
+# +str+ The message to print when +should_stop+ is true
+def stop_test(should_stop, str)
+  return unless should_stop
+  str ||= "Test stopped for unknown reason"
+  abort str.color(:red).bright
+end
+def xstop_test(*args, &ignore)
+  #puts %Q(Skipping stop_test: %s "%s") % [args.last.color(:blue), args.first.color(:blue).bright]
+end
+
+def skip(msg)
+  puts "%s (%s)" % ["SKIP".color(:blue).bright, msg]
+  :skip # this doesn't do anything, but I would like it to!
+end
+
+def xshould(*args, &ignore)
+  puts %Q(Skipping test: %s "%s") % [@name.color(:blue), args.first.color(:blue).bright]
+end
+def xcontext(*args, &ignore)
+  puts %q(Skipping context: "%s") % (@name || args.first).color(:blue).bright
+end

data/tryouts/disks.rb

@@ -0,0 +1,55 @@
+#--
+# TODO: Handle nested hashes and arrays. 
+# TODO: to_xml, see: http://codeforpeople.com/lib/ruby/xx/xx-2.0.0/README
+# TODO: Rename to Stuffany
+#++
+
+require 'yaml'
+require 'fileutils'
+
+module MetaData
+  extend self
+  @@paul = "front"
+  
+  def paul
+    @@paul
+  end
+end
+
+class Disk < Storable
+  include MetaData
+  extend MetaData
+  field :poop
+  field :help
+  
+
+end
+
+class Disks
+  def find *args
+  end
+  
+  def exists? *args
+  end
+  
+end
+
+d = Disk.new
+d.poop = "100a"
+d.help = true
+
+puts d.paul
+puts Disk.paul
+puts Disk.superclass::VERSION
+
+Disks.find disk_name
+Disks.find :env => 'prod', :role => 'app'
+Disks.exists? disk_name
+
+
+
+
+
+
+
+