newrelic-amazon-ec2 0.6.2

data/test/test_EC2_products.rb

@@ -0,0 +1,48 @@
+#--
+# Amazon Web Services EC2 Query API Ruby library
+#
+# Ruby Gem Name::  amazon-ec2
+# Author::    Glenn Rempe  (mailto:glenn@rempe.us)
+# Copyright:: Copyright (c) 2007-2008 Glenn Rempe
+# License::   Distributes under the same terms as Ruby
+# Home::      http://github.com/grempe/amazon-ec2/tree/master
+#++
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+context "An EC2 instance " do
+
+  before do
+    @ec2 = AWS::EC2::Base.new( :access_key_id => "not a key", :secret_access_key => "not a secret" )
+
+    @confirm_product_instance_response_body = <<-RESPONSE
+    <ConfirmProductInstanceResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <result>true</result>
+      <ownerId>254933287430</ownerId>
+    </ConfirmProductInstanceResponse>
+    RESPONSE
+
+  end
+
+
+  specify "should indicate whether a product code is attached to an instance" do
+    @ec2.stubs(:make_request).with('ConfirmProductInstance', {"ProductCode"=>"774F4FF8", "InstanceId"=>"i-10a64379"}).
+       returns stub(:body => @confirm_product_instance_response_body, :is_a? => true)
+
+    @ec2.confirm_product_instance( :product_code => "774F4FF8", :instance_id => "i-10a64379" ).should.be.an.instance_of Hash
+    response = @ec2.confirm_product_instance( :product_code => "774F4FF8", :instance_id => "i-10a64379" )
+    response.ownerId.should.equal "254933287430"
+    response.result.should.equal "true"
+  end
+
+
+  specify "method get_console_output should raise an exception when called without nil/empty string arguments" do
+    lambda { @ec2.confirm_product_instance() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.confirm_product_instance(:product_code => "774F4FF8", :instance_id => nil) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.confirm_product_instance(:product_code => "774F4FF8", :instance_id => "") }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.confirm_product_instance(:product_code => nil, :instance_id => "i-10a64379") }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.confirm_product_instance(:product_code => "", :instance_id => "i-10a64379") }.should.raise(AWS::ArgumentError)
+  end
+
+
+end

data/test/test_EC2_responses.rb

@@ -0,0 +1,53 @@
+#--
+# Amazon Web Services EC2 Query API Ruby library
+#
+# Ruby Gem Name::  amazon-ec2
+# Author::    Glenn Rempe  (mailto:glenn@rempe.us)
+# Copyright:: Copyright (c) 2007-2008 Glenn Rempe
+# License::   Distributes under the same terms as Ruby
+# Home::      http://github.com/grempe/amazon-ec2/tree/master
+#++
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+context "The Response classes " do
+
+
+  before do
+    @http_xml = <<-RESPONSE
+    <RebootInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </RebootInstancesResponse>
+    RESPONSE
+
+    @response = AWS::Response.parse(:xml => @http_xml)
+  end
+
+
+  specify "should show the response as a formatted string when calling #inspect" do
+    # sorting the response hash first since ruby 1.8.6 and ruby 1.9.1 sort the hash differently before the inspect
+    @response.sort.inspect.should.equal %{[[\"return\", \"true\"], [\"xmlns\", \"http://ec2.amazonaws.com/doc/2007-03-01\"]]}
+  end
+
+
+  specify "should be a Hash" do
+    @response.kind_of?(Hash).should.equal true
+  end
+
+
+  specify "should return its members" do
+    @response.keys.length.should.equal 2
+    test_array = ["return", "xmlns"].sort
+    @response.keys.sort.should.equal test_array
+  end
+
+
+  # Note: since we are now returning a hash of the xml, there should be no need for anyone to re-parse the xml.
+  # Therefore storing the xml on the object is a waste of memory, and is not done.
+  #
+  # specify "should return the original amazon XML response in the 'xml' attribute of the response object." do
+  #  @response.xml.should.equal @http_xml
+  # end
+
+
+end

data/test/test_EC2_s3_xmlsimple.rb

@@ -0,0 +1,80 @@
+#--
+# Amazon Web Services EC2 Query API Ruby library
+#
+# Ruby Gem Name::  amazon-ec2
+# Author::    Glenn Rempe  (mailto:glenn@rempe.us)
+# Copyright:: Copyright (c) 2007-2008 Glenn Rempe
+# License::   Distributes under the same terms as Ruby
+# Home::      http://github.com/grempe/amazon-ec2/tree/master
+#++
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+# NOTE : These tests exercise amazon-ec2 when used with the aws/s3 gem
+# which was demonstrating some breaking behavior.  The fix was to
+# add the XmlSimple option "'keeproot' => false" in responses.rb
+
+context "EC2 aws-s3 compat test" do
+
+  before do
+    @ec2 = AWS::EC2::Base.new( :access_key_id => "not a key", :secret_access_key => "not a secret" )
+
+    @describe_instances_response_body = <<-RESPONSE
+    <DescribeInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2007-08-29">
+      <reservationSet>
+        <item>
+          <reservationId>r-44a5402d</reservationId>
+          <ownerId>UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM</ownerId>
+          <groupSet>
+            <item>
+              <groupId>default</groupId>
+            </item>
+          </groupSet>
+          <instancesSet>
+            <item>
+              <instanceId>i-28a64341</instanceId>
+              <imageId>ami-6ea54007</imageId>
+              <instanceState>
+                <code>0</code>
+                <name>running</name>
+              </instanceState>
+              <privateDnsName>domU-12-31-35-00-1E-01.z-2.compute-1.internal</privateDnsName>
+              <dnsName>ec2-72-44-33-4.z-2.compute-1.amazonaws.com</dnsName>
+              <keyName>example-key-name</keyName>
+              <productCodesSet>
+                <item><productCode>774F4FF8</productCode></item>
+              </productCodesSet>
+              <instanceType>m1.small</instanceType>
+              <launchTime>2007-08-07T11:54:42.000Z</launchTime>
+            </item>
+          </instancesSet>
+        </item>
+      </reservationSet>
+    </DescribeInstancesResponse>
+    RESPONSE
+
+ end
+
+ specify "should be able to be described and return the correct Ruby response class" do
+   @ec2.stubs(:make_request).with('DescribeInstances', {}).
+     returns stub(:body => @describe_instances_response_body, :is_a? => true)
+   @ec2.describe_instances.should.be.an.instance_of Hash
+   response = @ec2.describe_instances
+   response.reservationSet.item[0].reservationId.should.equal "r-44a5402d"
+ end
+
+ specify "should be able to be described and return the correct Ruby response class when the aws/s3 lib is required" do
+   begin
+     require 'aws/s3s'
+     @ec2.stubs(:make_request).with('DescribeInstances', {}).
+       returns stub(:body => @describe_instances_response_body, :is_a? => true)
+     @ec2.describe_instances.should.be.an.instance_of Hash
+     response = @ec2.describe_instances
+     response.reservationSet.item[0].reservationId.should.equal "r-44a5402d"
+   rescue LoadError
+     # do nothing.  no aws/s3 gem installed to test against
+   end
+ end
+
+end
+

data/test/test_EC2_security_groups.rb

@@ -0,0 +1,205 @@
+#--
+# Amazon Web Services EC2 Query API Ruby library
+#
+# Ruby Gem Name::  amazon-ec2
+# Author::    Glenn Rempe  (mailto:glenn@rempe.us)
+# Copyright:: Copyright (c) 2007-2008 Glenn Rempe
+# License::   Distributes under the same terms as Ruby
+# Home::      http://github.com/grempe/amazon-ec2/tree/master
+#++
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+context "EC2 security groups " do
+
+  before do
+    @ec2 = AWS::EC2::Base.new( :access_key_id => "not a key", :secret_access_key => "not a secret" )
+
+    @create_security_group_response_body = <<-RESPONSE
+    <CreateSecurityGroupResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </CreateSecurityGroupResponse>
+    RESPONSE
+
+    @delete_security_group_response_body = <<-RESPONSE
+    <DeleteSecurityGroupResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </DeleteSecurityGroupResponse>
+    RESPONSE
+
+    @describe_security_groups_response_body = <<-RESPONSE
+    <DescribeSecurityGroupsResponse xm-lns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <securityGroupInfo>
+        <item>
+          <ownerId>UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM</ownerId>
+          <groupName>WebServers</groupName>
+          <groupDescription>Web</groupDescription>
+          <ipPermissions>
+            <item>
+              <ipProtocol>tcp</ipProtocol>
+              <fromPort>80</fromPort>
+              <toPort>80</toPort>
+              <groups/>
+              <ipRanges>
+                <item>
+                  <cidrIp>0.0.0.0/0</cidrIp>
+                </item>
+              </ipRanges>
+            </item>
+          </ipPermissions>
+        </item>
+        <item>
+          <ownerId>UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM</ownerId>
+          <groupName>RangedPortsBySource</groupName>
+          <groupDescription>A</groupDescription>
+          <ipPermissions>
+            <item>
+              <ipProtocol>tcp</ipProtocol>
+              <fromPort>6000</fromPort>
+              <toPort>7000</toPort>
+              <groups/>
+              <ipRanges/>
+            </item>
+          </ipPermissions>
+        </item>
+      </securityGroupInfo>
+    </DescribeSecurityGroupsResponse>
+    RESPONSE
+
+    @authorize_security_group_ingress_response_body = <<-RESPONSE
+    <AuthorizeSecurityGroupIngressResponse xm-lns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </AuthorizeSecurityGroupIngressResponse>
+    RESPONSE
+
+    @revoke_security_group_ingress_response_body = <<-RESPONSE
+    <RevokeSecurityGroupIngressResponse xm-lns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </RevokeSecurityGroupIngressResponse>
+    RESPONSE
+
+  end
+
+
+  specify "should be able to be created" do
+    @ec2.stubs(:make_request).with('CreateSecurityGroup', {"GroupName"=>"WebServers", "GroupDescription"=>"Web"}).
+      returns stub(:body => @create_security_group_response_body, :is_a? => true)
+    @ec2.create_security_group( :group_name => "WebServers", :group_description => "Web" ).should.be.an.instance_of Hash
+  end
+
+
+  specify "method create_security_group should reject bad arguments" do
+    @ec2.stubs(:make_request).with('CreateSecurityGroup', {"GroupName"=>"WebServers", "GroupDescription"=>"Web"}).
+      returns stub(:body => @create_security_group_response_body, :is_a? => true)
+
+    lambda { @ec2.create_security_group( :group_name => "WebServers", :group_description => "Web" ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.create_security_group() }.should.raise(AWS::ArgumentError)
+
+    # :group_name can't be nil or empty
+    lambda { @ec2.create_security_group( :group_name => "", :group_description => "Web" ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.create_security_group( :group_name => nil, :group_description => "Web" ) }.should.raise(AWS::ArgumentError)
+
+    # :group_description can't be nil or empty
+    lambda { @ec2.create_security_group( :group_name => "WebServers", :group_description => "" ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.create_security_group( :group_name => "WebServers", :group_description => nil ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+  specify "should be able to be deleted" do
+    @ec2.stubs(:make_request).with('DeleteSecurityGroup', {"GroupName"=>"WebServers"}).
+      returns stub(:body => @delete_security_group_response_body, :is_a? => true)
+    @ec2.delete_security_group( :group_name => "WebServers" ).should.be.an.instance_of Hash
+  end
+
+
+  specify "method delete_security_group should reject bad arguments" do
+    @ec2.stubs(:make_request).with('DeleteSecurityGroup', {"GroupName"=>"WebServers"}).
+      returns stub(:body => @delete_security_group_response_body, :is_a? => true)
+
+    lambda { @ec2.delete_security_group( :group_name => "WebServers" ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.delete_security_group() }.should.raise(AWS::ArgumentError)
+
+    # :group_name can't be nil or empty
+    lambda { @ec2.delete_security_group( :group_name => "" ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.delete_security_group( :group_name => nil ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+  specify "should be able to be described with describe_security_groups" do
+    @ec2.stubs(:make_request).with('DescribeSecurityGroups', { "GroupName.1" => "WebServers", "GroupName.2" => "RangedPortsBySource" }).
+      returns stub(:body => @describe_security_groups_response_body, :is_a? => true)
+    @ec2.describe_security_groups( :group_name => ["WebServers", "RangedPortsBySource"] ).should.be.an.instance_of Hash
+
+    response = @ec2.describe_security_groups( :group_name => ["WebServers", "RangedPortsBySource"] )
+
+    response.securityGroupInfo.item[0].ownerId.should.equal "UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM"
+    response.securityGroupInfo.item[0].groupName.should.equal "WebServers"
+    response.securityGroupInfo.item[0].groupDescription.should.equal "Web"
+    response.securityGroupInfo.item[0].ipPermissions.item[0].ipProtocol.should.equal "tcp"
+    response.securityGroupInfo.item[0].ipPermissions.item[0].fromPort.should.equal "80"
+    response.securityGroupInfo.item[0].ipPermissions.item[0].toPort.should.equal "80"
+    response.securityGroupInfo.item[0].ipPermissions.item[0].groups.should.be.nil
+    response.securityGroupInfo.item[0].ipPermissions.item[0].ipRanges.item[0].cidrIp.should.equal "0.0.0.0/0"
+
+    response.securityGroupInfo.item[1].ownerId.should.equal "UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM"
+    response.securityGroupInfo.item[1].groupName.should.equal "RangedPortsBySource"
+    response.securityGroupInfo.item[1].groupDescription.should.equal "A"
+    response.securityGroupInfo.item[1].ipPermissions.item[0].ipProtocol.should.equal "tcp"
+    response.securityGroupInfo.item[1].ipPermissions.item[0].fromPort.should.equal "6000"
+    response.securityGroupInfo.item[1].ipPermissions.item[0].toPort.should.equal "7000"
+    response.securityGroupInfo.item[1].ipPermissions.item[0].groups.should.be.nil
+    response.securityGroupInfo.item[1].ipPermissions.item[0].ipRanges.should.be.nil
+  end
+
+
+  specify "method describe_security_groups should reject bad arguments" do
+    @ec2.stubs(:make_request).with('DescribeSecurityGroups', {"GroupName.1"=>"WebServers"}).
+      returns stub(:body => @describe_security_groups_response_body, :is_a? => true)
+
+      lambda { @ec2.describe_security_groups( :group_name => "WebServers" ) }.should.not.raise(AWS::ArgumentError)
+
+  end
+
+
+  specify "permissions should be able to be added to a security group with authorize_security_group_ingress." do
+    @ec2.stubs(:make_request).with('AuthorizeSecurityGroupIngress', { "GroupName"=>"WebServers",
+                                                                      "IpProtocol"=>"tcp",
+                                                                      "FromPort"=>"8000",
+                                                                      "ToPort"=>"80",
+                                                                      "CidrIp"=>"0.0.0.0/24",
+                                                                      "SourceSecurityGroupName"=>"Source SG Name",
+                                                                      "SourceSecurityGroupOwnerId"=>"123"}).
+      returns stub(:body => @authorize_security_group_ingress_response_body, :is_a? => true)
+
+    @ec2.authorize_security_group_ingress( :group_name => "WebServers",
+                                           :ip_protocol => "tcp",
+                                           :from_port => "8000",
+                                           :to_port => "80",
+                                           :cidr_ip => "0.0.0.0/24",
+                                           :source_security_group_name => "Source SG Name",
+                                           :source_security_group_owner_id => "123"
+                                           ).should.be.an.instance_of Hash
+  end
+
+
+  specify "permissions should be able to be revoked from a security group with revoke_security_group_ingress." do
+    @ec2.stubs(:make_request).with('RevokeSecurityGroupIngress', { "GroupName"=>"WebServers",
+                                                                   "IpProtocol"=>"tcp",
+                                                                   "FromPort"=>"8000",
+                                                                   "ToPort"=>"80",
+                                                                   "CidrIp"=>"0.0.0.0/24",
+                                                                   "SourceSecurityGroupName"=>"Source SG Name",
+                                                                   "SourceSecurityGroupOwnerId"=>"123"}).
+      returns stub(:body => @revoke_security_group_ingress_response_body, :is_a? => true)
+
+    @ec2.revoke_security_group_ingress( :group_name => "WebServers",
+                                        :ip_protocol => "tcp",
+                                        :from_port => "8000",
+                                        :to_port => "80",
+                                        :cidr_ip => "0.0.0.0/24",
+                                        :source_security_group_name => "Source SG Name",
+                                        :source_security_group_owner_id => "123"
+                                        ).should.be.an.instance_of Hash
+  end
+
+end

data/test/test_EC2_snapshots.rb

@@ -0,0 +1,83 @@
+#--
+# Amazon Web Services EC2 Query API Ruby library, EBS snapshots support
+#
+# Ruby Gem Name::  amazon-ec2
+# Author::    Yann Klis  (mailto:yann.klis@novelys.com)
+# Copyright:: Copyright (c) 2008 Yann Klis
+# License::   Distributes under the same terms as Ruby
+# Home::      http://github.com/grempe/amazon-ec2/tree/master
+#++
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+context "EC2 snaphots " do
+
+  before do
+    @ec2 = AWS::EC2::Base.new( :access_key_id => "not a key", :secret_access_key => "not a secret" )
+
+    @describe_snapshots_response_body = <<-RESPONSE
+    <DescribeSnapshotsResponse xmlns="http://ec2.amazonaws.com/doc/2008-05-05">
+      <snapshotId>snap-78a54011</snapshotId>
+      <volumeId>vol-4d826724</volumeId>
+      <status>pending</status>
+      <startTime>2008-05-07T12:51:50.000Z</startTime>
+      <progress>80%</progress>
+    </DescribeSnapshotsResponse>
+    RESPONSE
+
+    @create_snapshot_response_body = <<-RESPONSE
+    <CreateSnapshotResponse xmlns="http://ec2.amazonaws.com/doc/2008-05-05">
+      <snapshotId>snap-78a54011</snapshotId>
+      <volumeId>vol-4d826724</volumeId>
+      <status>pending</status>
+      <startTime>2008-05-07T12:51:50.000Z</startTime>
+      <progress></progress>
+    </CreateSnapshotResponse>
+    RESPONSE
+
+    @delete_snapshot_response_body = <<-RESPONSE
+    <DeleteSnapshotResponse xmlns="http://ec2.amazonaws.com/doc/2008-05-05">
+      <return>true</return>
+    </DeleteSnapshotResponse>
+    RESPONSE
+
+  end
+
+
+  specify "should be able to be described with describe_snapshots" do
+    @ec2.stubs(:make_request).with('DescribeSnapshots', {"SnapshotId.1"=>"snap-78a54011"}).
+      returns stub(:body => @describe_snapshots_response_body, :is_a? => true)
+
+    @ec2.describe_snapshots( :snapshot_id => "snap-78a54011" ).should.be.an.instance_of Hash
+
+    response = @ec2.describe_snapshots( :snapshot_id => "snap-78a54011" )
+    response.snapshotId.should.equal "snap-78a54011"
+    response.volumeId.should.equal "vol-4d826724"
+    response.status.should.equal "pending"
+    response.progress.should.equal "80%"
+  end
+
+  specify "should be able to be created with a volume_id" do
+    @ec2.stubs(:make_request).with('CreateSnapshot', {"VolumeId" => "vol-4d826724"}).
+      returns stub(:body => @create_snapshot_response_body, :is_a? => true)
+
+    @ec2.create_snapshot( :volume_id => "vol-4d826724" ).should.be.an.instance_of Hash
+
+    response = @ec2.create_snapshot( :volume_id => "vol-4d826724" )
+    response.snapshotId.should.equal "snap-78a54011"
+    response.volumeId.should.equal "vol-4d826724"
+    response.status.should.equal "pending"
+    response.progress.should.equal nil
+  end
+
+  specify "should be able to be deleted with a snapsot_id" do
+    @ec2.stubs(:make_request).with('DeleteSnapshot', {"SnapshotId" => "snap-78a54011"}).
+      returns stub(:body => @delete_snapshot_response_body, :is_a? => true)
+
+    @ec2.delete_snapshot( :snapshot_id => "snap-78a54011" ).should.be.an.instance_of Hash
+
+    response = @ec2.delete_snapshot( :snapshot_id => "snap-78a54011" )
+    response.return.should.equal "true"
+  end
+
+end