newrelic-amazon-ec2 0.6.2

data/test/test_EC2_instances.rb

@@ -0,0 +1,429 @@
+#--
+# 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 instances " do
+
+  before do
+    @ec2 = AWS::EC2::Base.new( :access_key_id => "not a key", :secret_access_key => "not a secret" )
+
+    @run_instances_response_body = <<-RESPONSE
+    <RunInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2007-08-29">
+      <reservationId>r-47a5402e</reservationId>
+      <ownerId>495219933132</ownerId>
+      <groupSet>
+        <item>
+          <groupId>default</groupId>
+        </item>
+      </groupSet>
+      <instancesSet>
+        <item>
+          <instanceId>i-2ba64342</instanceId>
+          <imageId>ami-60a54009</imageId>
+          <instanceState>
+            <code>0</code>
+            <name>pending</name>
+          </instanceState>
+          <privateDnsName></privateDnsName>
+          <dnsName></dnsName>
+          <keyName>example-key-name</keyName>
+           <amiLaunchIndex>0</amiLaunchIndex>
+          <instanceType>m1.small</instanceType>
+          <launchTime>2007-08-07T11:51:50.000Z</launchTime>
+        </item>
+        <item>
+          <instanceId>i-2bc64242</instanceId>
+          <imageId>ami-60a54009</imageId>
+          <instanceState>
+            <code>0</code>
+            <name>pending</name>
+          </instanceState>
+          <privateDnsName></privateDnsName>
+          <dnsName></dnsName>
+          <keyName>example-key-name</keyName>
+          <amiLaunchIndex>1</amiLaunchIndex>
+          <instanceType>m1.small</instanceType>
+          <launchTime>2007-08-07T11:51:50.000Z</launchTime>
+        </item>
+        <item>
+          <instanceId>i-2be64332</instanceId>
+          <imageId>ami-60a54009</imageId>
+          <instanceState>
+            <code>0</code>
+     <name>pending</name>
+          </instanceState>
+          <privateDnsName></privateDnsName>
+          <dnsName></dnsName>
+          <keyName>example-key-name</keyName>
+          <amiLaunchIndex>2</amiLaunchIndex>
+          <instanceType>m1.small</instanceType>
+          <launchTime>2007-08-07T11:51:50.000Z</launchTime>
+        </item>
+      </instancesSet>
+    </RunInstancesResponse>
+    RESPONSE
+
+    @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
+
+    @reboot_instances_response_body = <<-RESPONSE
+    <RebootInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </RebootInstancesResponse>
+    RESPONSE
+
+    @terminate_instances_response_body = <<-RESPONSE
+    <TerminateInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <instancesSet>
+        <item>
+          <instanceId>i-28a64341</instanceId>
+          <shutdownState>
+            <code>32</code>
+            <name>shutting-down</name>
+          </shutdownState>
+          <previousState>
+            <code>0</code>
+            <name>pending</name>
+          </previousState>
+        </item>
+        <item>
+          <instanceId>i-21a64348</instanceId>
+          <shutdownState>
+            <code>32</code>
+            <name>shutting-down</name>
+          </shutdownState>
+          <previousState>
+            <code>0</code>
+            <name>pending</name>
+          </previousState>
+        </item>
+      </instancesSet>
+    </TerminateInstancesResponse>
+    RESPONSE
+
+    @monitor_instances_response_body = <<-RESPONSE
+    <MonitorInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2009-07-15/">
+        <requestId>fe62a64c-49fb-4a3c-8d9b-61aba146d390</requestId>
+        <instancesSet>
+            <item>
+                <instanceId>i-138fc47a</instanceId>
+                <monitoring>
+                    <state>pending</state>
+                </monitoring>
+            </item>
+            <item>
+                <instanceId>i-33457a5a</instanceId>
+                <monitoring>
+                    <state>pending</state>
+                </monitoring>
+            </item>
+        </instancesSet>
+    </MonitorInstancesResponse>
+    RESPONSE
+
+    @unmonitor_instances_response_body = <<-RESPONSE
+    <UnmonitorInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2009-07-15/">
+        <requestId>7dbc5095-f3ae-46d8-a5b1-19df118ceb05</requestId>
+        <instancesSet>
+            <item>
+                <instanceId>i-138fc47a</instanceId>
+                <monitoring>
+                    <state>disabling</state>
+                </monitoring>
+            </item>
+            <item>
+                <instanceId>i-33457a5a</instanceId>
+                <monitoring>
+                    <state>disabling</state>
+                </monitoring>
+            </item>
+        </instancesSet>
+    </UnmonitorInstancesResponse>
+
+    RESPONSE
+  end
+
+
+  specify "should be able to be run" do
+    @ec2.stubs(:make_request).with('RunInstances', "ImageId" => "ami-60a54009", "MinCount" => '1', "MaxCount" => '1', "AddressingType" => 'public', 'InstanceType' => 'm1.small').
+      returns stub(:body => @run_instances_response_body, :is_a? => true)
+
+    @ec2.run_instances( :image_id => "ami-60a54009" ).should.be.an.instance_of Hash
+
+    response = @ec2.run_instances( :image_id => "ami-60a54009" )
+
+    response.reservationId.should.equal "r-47a5402e"
+    response.ownerId.should.equal "495219933132"
+
+    response.groupSet.item[0].groupId.should.equal "default"
+
+    response.instancesSet.item.length.should.equal 3
+
+    response.instancesSet.item[0].instanceId.should.equal "i-2ba64342"
+    response.instancesSet.item[0].imageId.should.equal "ami-60a54009"
+    response.instancesSet.item[0].instanceState.code.should.equal "0"
+    response.instancesSet.item[0].instanceState.name.should.equal "pending"
+    response.instancesSet.item[0].privateDnsName
+    response.instancesSet.item[0].dnsName.should.be.nil
+    response.instancesSet.item[0].keyName.should.equal "example-key-name"
+    response.instancesSet.item[0].instanceType.should.equal "m1.small"
+    response.instancesSet.item[0].launchTime.should.equal "2007-08-07T11:51:50.000Z"
+
+    response.instancesSet.item[1].instanceId.should.equal "i-2bc64242"
+    response.instancesSet.item[1].imageId.should.equal "ami-60a54009"
+    response.instancesSet.item[1].instanceState.code.should.equal "0"
+    response.instancesSet.item[1].instanceState.name.should.equal "pending"
+    response.instancesSet.item[1].privateDnsName
+    response.instancesSet.item[1].dnsName.should.be.nil
+    response.instancesSet.item[1].keyName.should.equal "example-key-name"
+    response.instancesSet.item[1].instanceType.should.equal "m1.small"
+    response.instancesSet.item[1].launchTime.should.equal "2007-08-07T11:51:50.000Z"
+
+    response.instancesSet.item[2].instanceId.should.equal "i-2be64332"
+    response.instancesSet.item[2].imageId.should.equal "ami-60a54009"
+    response.instancesSet.item[2].instanceState.code.should.equal "0"
+    response.instancesSet.item[2].instanceState.name.should.equal "pending"
+    response.instancesSet.item[2].privateDnsName
+    response.instancesSet.item[2].dnsName.should.be.nil
+    response.instancesSet.item[2].keyName.should.equal "example-key-name"
+    response.instancesSet.item[2].instanceType.should.equal "m1.small"
+    response.instancesSet.item[2].launchTime.should.equal "2007-08-07T11:51:50.000Z"
+  end
+
+
+  specify "method 'run_instances' should reject invalid arguments" do
+    @ec2.stubs(:make_request).with('RunInstances', "ImageId" => "ami-60a54009", "MinCount" => '1', "MaxCount" => '1', "AddressingType" => 'public', 'InstanceType' => 'm1.small').
+      returns stub(:body => @run_instances_response_body, :is_a? => true)
+
+    lambda { @ec2.run_instances() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "" ) }.should.raise(AWS::ArgumentError)
+
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :min_count => 1 ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :min_count => 0 ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :min_count => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :min_count => "" ) }.should.raise(AWS::ArgumentError)
+
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :max_count => 1 ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :max_count => 0 ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :max_count => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :max_count => "" ) }.should.raise(AWS::ArgumentError)
+
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :addressing_type => "public" ) }.should.not.raise(AWS::ArgumentError)
+    #lambda { @ec2.run_instances( :image_id => "ami-60a54009", :addressing_type => "direct" ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :addressing_type => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :addressing_type => "" ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :addressing_type => "foo" ) }.should.raise(AWS::ArgumentError)
+
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :base64_encoded => true ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :base64_encoded => false ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :base64_encoded => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :base64_encoded => "" ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.run_instances( :image_id => "ami-60a54009", :base64_encoded => "foo" ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+  specify "should be able specify an availability_zone" do
+    @ec2.stubs(:make_request).with('RunInstances', "ImageId" => "ami-60a54009", "MinCount" => '1', "MaxCount" => '1', "Placement.AvailabilityZone" => "zone123", "UserData" => "Zm9v", "AddressingType" => 'public', 'InstanceType' => 'm1.small').
+      returns stub(:body => @run_instances_response_body, :is_a? => true)
+    @ec2.run_instances( :image_id => "ami-60a54009", :min_count => 1, :max_count => 1, :availability_zone => "zone123", :group_id => [], :user_data => "foo", :base64_encoded => true ).should.be.an.instance_of Hash
+  end
+
+  specify "should be able to call run_instances with :user_data and :base64_encoded => true (default is false)" do
+    @ec2.stubs(:make_request).with('RunInstances', "ImageId" => "ami-60a54009", "MinCount" => '1', "MaxCount" => '1', "UserData" => "Zm9v", "AddressingType" => 'public', 'InstanceType' => 'm1.small').
+      returns stub(:body => @run_instances_response_body, :is_a? => true)
+    @ec2.run_instances( :image_id => "ami-60a54009", :min_count => 1, :max_count => 1, :group_id => [], :user_data => "foo", :base64_encoded => true ).should.be.an.instance_of Hash
+  end
+
+  specify "should be able specify an kernel_id" do
+    @ec2.stubs(:make_request).with('RunInstances', "ImageId" => "ami-60a54009", "MinCount" => '1', "MaxCount" => '1', "Placement.AvailabilityZone" => "zone123", "UserData" => "Zm9v", "AddressingType" => 'public', 'InstanceType' => 'm1.small', 'KernelId' => 'kernfoo').
+      returns stub(:body => @run_instances_response_body, :is_a? => true)
+    @ec2.run_instances( :image_id => "ami-60a54009", :min_count => 1, :max_count => 1, :availability_zone => "zone123", :group_id => [], :user_data => "foo", :base64_encoded => true, :kernel_id => 'kernfoo' ).should.be.an.instance_of Hash
+  end
+
+  specify "should be able to call run_instances with :user_data and :base64_encoded => false" do
+    @ec2.stubs(:make_request).with('RunInstances', "ImageId" => "ami-60a54009", "MinCount" => '1', "MaxCount" => '1', "UserData" => "foo", "AddressingType" => 'public', 'InstanceType' => 'm1.small').
+      returns stub(:body => @run_instances_response_body, :is_a? => true)
+    @ec2.run_instances( :image_id => "ami-60a54009", :min_count => 1, :max_count => 1, :group_id => [], :user_data => "foo", :base64_encoded => false ).should.be.an.instance_of Hash
+  end
+
+  specify "should get no user data for when options has no user_data key" do
+    @ec2.extract_user_data({}).should == nil
+  end
+
+  specify "should get plain string user data when options has user_data and no base64 key" do
+    @ec2.extract_user_data({:user_data => "foo\nbar"}).should == "foo\nbar"
+  end
+
+  specify "should strip new lines and base64 encode when options has both user_data and base64" do
+    @ec2.extract_user_data({:user_data => "binary\ndata\nhere\n", :base64_encoded => true}).should == "YmluYXJ5CmRhdGEKaGVyZQo="
+  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 with no params and return an array of Items" do
+    @ec2.stubs(:make_request).with('DescribeInstances', {}).
+      returns stub(:body => @describe_instances_response_body, :is_a? => true)
+    @ec2.describe_instances.reservationSet.item.length.should.equal 1
+    response = @ec2.describe_instances
+    response.reservationSet.item[0].reservationId.should.equal "r-44a5402d"
+    response.reservationSet.item[0].ownerId.should.equal "UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM"
+    response.reservationSet.item[0].groupSet.item[0].groupId.should.equal "default"
+    response.reservationSet.item[0].instancesSet.item[0].instanceId.should.equal "i-28a64341"
+    response.reservationSet.item[0].instancesSet.item[0].imageId.should.equal "ami-6ea54007"
+    response.reservationSet.item[0].instancesSet.item[0].instanceState.code.should.equal "0"
+    response.reservationSet.item[0].instancesSet.item[0].instanceState.name.should.equal "running"
+    response.reservationSet.item[0].instancesSet.item[0].privateDnsName.should.equal "domU-12-31-35-00-1E-01.z-2.compute-1.internal"
+    response.reservationSet.item[0].instancesSet.item[0].dnsName.should.equal "ec2-72-44-33-4.z-2.compute-1.amazonaws.com"
+    response.reservationSet.item[0].instancesSet.item[0].keyName.should.equal "example-key-name"
+    response.reservationSet.item[0].instancesSet.item[0].productCodesSet.item[0].productCode.should.equal "774F4FF8"
+  end
+
+
+  specify "should be able to be described with params of Array of :instance_id's and return an array of Items" do
+    @ec2.stubs(:make_request).with('DescribeInstances', {"InstanceId.1" => "i-28a64341"}).
+      returns stub(:body => @describe_instances_response_body, :is_a? => true)
+    @ec2.describe_instances( :instance_id => "i-28a64341" ).reservationSet.item.length.should.equal 1
+    response = @ec2.describe_instances( :instance_id => "i-28a64341" )
+    response.reservationSet.item[0].reservationId.should.equal "r-44a5402d"
+    response.reservationSet.item[0].ownerId.should.equal "UYY3TLBUXIEON5NQVUUX6OMPWBZIQNFM"
+    response.reservationSet.item[0].groupSet.item[0].groupId.should.equal "default"
+    response.reservationSet.item[0].instancesSet.item[0].instanceId.should.equal "i-28a64341"
+    response.reservationSet.item[0].instancesSet.item[0].imageId.should.equal "ami-6ea54007"
+    response.reservationSet.item[0].instancesSet.item[0].instanceState.code.should.equal "0"
+    response.reservationSet.item[0].instancesSet.item[0].instanceState.name.should.equal "running"
+    response.reservationSet.item[0].instancesSet.item[0].privateDnsName.should.equal "domU-12-31-35-00-1E-01.z-2.compute-1.internal"
+    response.reservationSet.item[0].instancesSet.item[0].dnsName.should.equal "ec2-72-44-33-4.z-2.compute-1.amazonaws.com"
+    response.reservationSet.item[0].instancesSet.item[0].keyName.should.equal "example-key-name"
+    response.reservationSet.item[0].instancesSet.item[0].productCodesSet.item[0].productCode.should.equal "774F4FF8"
+  end
+
+
+  specify "method reboot_instances should raise an exception when called without nil/empty string arguments" do
+    lambda { @ec2.reboot_instances() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.reboot_instances( :instance_id => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.reboot_instances( :instance_id => "" ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+  specify "should be able to be rebooted when provided with an :instance_id" do
+    @ec2.expects(:make_request).with('RebootInstances', {"InstanceId.1"=>"i-2ea64347", "InstanceId.2"=>"i-21a64348"}).
+      returns stub(:body => @reboot_instances_response_body, :is_a? => true)
+    @ec2.reboot_instances( :instance_id => ["i-2ea64347", "i-21a64348"] ).class.should.equal Hash
+  end
+
+
+  specify "method terminate_instances should raise an exception when called without nil/empty string arguments" do
+    lambda { @ec2.terminate_instances() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.terminate_instances( :instance_id => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.terminate_instances( :instance_id => "" ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+  specify "should be able to be terminated when provided with an :instance_id" do
+    @ec2.stubs(:make_request).with('TerminateInstances', {"InstanceId.1"=>"i-28a64341", "InstanceId.2"=>"i-21a64348"}).
+      returns stub(:body => @terminate_instances_response_body, :is_a? => true)
+    @ec2.terminate_instances( :instance_id => ["i-28a64341", "i-21a64348"] ).class.should.equal Hash
+
+    @response = @ec2.terminate_instances( :instance_id => ["i-28a64341", "i-21a64348"] )
+
+    @response.instancesSet.item[0].instanceId.should.equal "i-28a64341"
+    @response.instancesSet.item[0].shutdownState.code.should.equal "32"
+    @response.instancesSet.item[0].shutdownState.name.should.equal "shutting-down"
+    @response.instancesSet.item[0].previousState.code.should.equal "0"
+    @response.instancesSet.item[0].previousState.name.should.equal "pending"
+
+    @response.instancesSet.item[1].instanceId.should.equal "i-21a64348"
+    @response.instancesSet.item[1].shutdownState.code.should.equal "32"
+    @response.instancesSet.item[1].shutdownState.name.should.equal "shutting-down"
+    @response.instancesSet.item[1].previousState.code.should.equal "0"
+    @response.instancesSet.item[1].previousState.name.should.equal "pending"
+  end
+
+  specify "method monitor_instances should raise an exception when called without nil/empty string arguments" do
+    lambda { @ec2.monitor_instances() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.monitor_instances( :instance_id => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.monitor_instances( :instance_id => "" ) }.should.raise(AWS::ArgumentError)
+  end
+
+  specify "should be able to be monitored when provided with an :instance_id" do
+    @ec2.stubs(:make_request).with('MonitorInstances', {"InstanceId.1"=>"i-138fc47a", "InstanceId.2"=>"i-33457a5a"}).
+      returns stub(:body => @monitor_instances_response_body, :is_a? => true)
+    @ec2.monitor_instances( :instance_id => ["i-138fc47a", "i-33457a5a"] ).class.should.equal Hash
+
+    @response = @ec2.monitor_instances( :instance_id => ["i-138fc47a", "i-33457a5a"] )
+
+    @response.instancesSet.item[0].instanceId.should.equal "i-138fc47a"
+    @response.instancesSet.item[0].monitoring.state.should.equal "pending"
+
+    @response.instancesSet.item[1].instanceId.should.equal "i-33457a5a"
+    @response.instancesSet.item[1].monitoring.state.should.equal "pending"
+  end
+
+  specify "method unmonitor_instances should raise an exception when called without nil/empty string arguments" do
+    lambda { @ec2.unmonitor_instances() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.unmonitor_instances( :instance_id => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.unmonitor_instances( :instance_id => "" ) }.should.raise(AWS::ArgumentError)
+  end
+
+  specify "should be able to be unmonitored when provided with an :instance_id" do
+    @ec2.stubs(:make_request).with('UnmonitorInstances', {"InstanceId.1"=>"i-138fc47a", "InstanceId.2"=>"i-33457a5a"}).
+      returns stub(:body => @unmonitor_instances_response_body, :is_a? => true)
+    @ec2.unmonitor_instances( :instance_id => ["i-138fc47a", "i-33457a5a"] ).class.should.equal Hash
+
+    @response = @ec2.unmonitor_instances( :instance_id => ["i-138fc47a", "i-33457a5a"] )
+
+    @response.instancesSet.item[0].instanceId.should.equal "i-138fc47a"
+    @response.instancesSet.item[0].monitoring.state.should.equal "disabling"
+
+    @response.instancesSet.item[1].instanceId.should.equal "i-33457a5a"
+    @response.instancesSet.item[1].monitoring.state.should.equal "disabling"
+  end
+
+end

data/test/test_EC2_keypairs.rb

@@ -0,0 +1,123 @@
+#--
+# 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 keypairs " do
+
+  before do
+    @ec2 = AWS::EC2::Base.new( :access_key_id => "not a key", :secret_access_key => "not a secret" )
+
+    @create_keypair_response_body = <<-RESPONSE
+      <CreateKeyPairResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <keyName>example-key-name</keyName>
+      <keyFingerprint>1f:51:ae:28:bf:89:e9:d8:1f:25:5d:37:2d:7d:b8:ca:9f:f5:f1:6f</keyFingerprint>
+      <keyMaterial>-----BEGIN RSA PRIVATE KEY-----
+      MIIEoQIBAAKCAQBuLFg5ujHrtm1jnutSuoO8Xe56LlT+HM8v/xkaa39EstM3/aFxTHgElQiJLChp
+      HungXQ29VTc8rc1bW0lkdi23OH5eqkMHGhvEwqa0HWASUMll4o3o/IX+0f2UcPoKCOVUR+jx71Sg
+      5AU52EQfanIn3ZQ8lFW7Edp5a3q4DhjGlUKToHVbicL5E+g45zfB95wIyywWZfeW/UUF3LpGZyq/
+      ebIUlq1qTbHkLbCC2r7RTn8vpQWp47BGVYGtGSBMpTRP5hnbzzuqj3itkiLHjU39S2sJCJ0TrJx5
+      i8BygR4s3mHKBj8l+ePQxG1kGbF6R4yg6sECmXn17MRQVXODNHZbAgMBAAECggEAY1tsiUsIwDl5
+      91CXirkYGuVfLyLflXenxfI50mDFms/mumTqloHO7tr0oriHDR5K7wMcY/YY5YkcXNo7mvUVD1pM
+      ZNUJs7rw9gZRTrf7LylaJ58kOcyajw8TsC4e4LPbFaHwS1d6K8rXh64o6WgW4SrsB6ICmr1kGQI7
+      3wcfgt5ecIu4TZf0OE9IHjn+2eRlsrjBdeORi7KiUNC/pAG23I6MdDOFEQRcCSigCj+4/mciFUSA
+      SWS4dMbrpb9FNSIcf9dcLxVM7/6KxgJNfZc9XWzUw77Jg8x92Zd0fVhHOux5IZC+UvSKWB4dyfcI
+      tE8C3p9bbU9VGyY5vLCAiIb4qQKBgQDLiO24GXrIkswF32YtBBMuVgLGCwU9h9HlO9mKAc2m8Cm1
+      jUE5IpzRjTedc9I2qiIMUTwtgnw42auSCzbUeYMURPtDqyQ7p6AjMujp9EPemcSVOK9vXYL0Ptco
+      xW9MC0dtV6iPkCN7gOqiZXPRKaFbWADp16p8UAIvS/a5XXk5jwKBgQCKkpHi2EISh1uRkhxljyWC
+      iDCiK6JBRsMvpLbc0v5dKwP5alo1fmdR5PJaV2qvZSj5CYNpMAy1/EDNTY5OSIJU+0KFmQbyhsbm
+      rdLNLDL4+TcnT7c62/aH01ohYaf/VCbRhtLlBfqGoQc7+sAc8vmKkesnF7CqCEKDyF/dhrxYdQKB
+      gC0iZzzNAapayz1+JcVTwwEid6j9JqNXbBc+Z2YwMi+T0Fv/P/hwkX/ypeOXnIUcw0Ih/YtGBVAC
+      DQbsz7LcY1HqXiHKYNWNvXgwwO+oiChjxvEkSdsTTIfnK4VSCvU9BxDbQHjdiNDJbL6oar92UN7V
+      rBYvChJZF7LvUH4YmVpHAoGAbZ2X7XvoeEO+uZ58/BGKOIGHByHBDiXtzMhdJr15HTYjxK7OgTZm
+      gK+8zp4L9IbvLGDMJO8vft32XPEWuvI8twCzFH+CsWLQADZMZKSsBasOZ/h1FwhdMgCMcY+Qlzd4
+      JZKjTSu3i7vhvx6RzdSedXEMNTZWN4qlIx3kR5aHcukCgYA9T+Zrvm1F0seQPbLknn7EqhXIjBaT
+      P8TTvW/6bdPi23ExzxZn7KOdrfclYRph1LHMpAONv/x2xALIf91UB+v5ohy1oDoasL0gij1houRe
+      2ERKKdwz0ZL9SWq6VTdhr/5G994CK72fy5WhyERbDjUIdHaK3M849JJuf8cSrvSb4g==
+      -----END RSA PRIVATE KEY-----</keyMaterial>
+      </CreateKeyPairResponse>
+    RESPONSE
+
+    @describe_keypairs_response_body = <<-RESPONSE
+    <DescribeKeyPairsResponse xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <keySet>
+        <item>
+          <keyName>example-key-name</keyName>
+          <keyFingerprint>1f:51:ae:28:bf:89:e9:d8:1f:25:5d:37:2d:7d:b8:ca:9f:f5:f1:6f</keyFingerprint>
+        </item>
+      </keySet>
+    </DescribeKeyPairsResponse>
+    RESPONSE
+
+    @delete_keypair_body = <<-RESPONSE
+    <DeleteKeyPair xmlns="http://ec2.amazonaws.com/doc/2007-03-01">
+      <return>true</return>
+    </DeleteKeyPair>
+    RESPONSE
+
+  end
+
+
+  specify "should be able to be created" do
+    @ec2.stubs(:make_request).with('CreateKeyPair', {"KeyName"=>"example-key-name"}).
+      returns stub(:body => @create_keypair_response_body, :is_a? => true)
+
+    @ec2.create_keypair( :key_name => "example-key-name" ).should.be.an.instance_of Hash
+
+    response = @ec2.create_keypair( :key_name => "example-key-name" )
+    response.keyName.should.equal "example-key-name"
+    response.keyFingerprint.should.equal "1f:51:ae:28:bf:89:e9:d8:1f:25:5d:37:2d:7d:b8:ca:9f:f5:f1:6f"
+    response.keyMaterial.should.not.equal ""
+    response.keyMaterial.should.not.be.nil
+  end
+
+
+  specify "method create_keypair should reject bad arguments" do
+    @ec2.stubs(:make_request).with('CreateKeyPair', {"KeyName"=>"example-key-name"}).
+      returns stub(:body => @create_keypair_response_body, :is_a? => true)
+
+    lambda { @ec2.create_keypair( :key_name => "example-key-name" ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.create_keypair() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.create_keypair( :key_name => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.create_keypair( :key_name => "" ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+  specify "should be able to be described with describe_keypairs" do
+    @ec2.stubs(:make_request).with('DescribeKeyPairs', {"KeyName.1"=>"example-key-name"}).
+      returns stub(:body => @describe_keypairs_response_body, :is_a? => true)
+    @ec2.describe_keypairs( :key_name => "example-key-name" ).should.be.an.instance_of Hash
+    response = @ec2.describe_keypairs( :key_name => "example-key-name" )
+    response.keySet.item[0].keyName.should.equal "example-key-name"
+    response.keySet.item[0].keyFingerprint.should.equal "1f:51:ae:28:bf:89:e9:d8:1f:25:5d:37:2d:7d:b8:ca:9f:f5:f1:6f"
+  end
+
+
+  specify "should be able to be deleted with delete_keypairs" do
+    @ec2.stubs(:make_request).with('DeleteKeyPair', {"KeyName"=>"example-key-name"}).
+      returns stub(:body => @delete_keypair_body, :is_a? => true)
+    @ec2.delete_keypair( :key_name => "example-key-name" ).should.be.an.instance_of Hash
+    response = @ec2.delete_keypair( :key_name => "example-key-name" )
+    response.return.should.equal "true"
+  end
+
+
+  specify "method delete_keypair should reject bad argument" do
+    @ec2.stubs(:make_request).with('DeleteKeyPair', {"KeyName"=>"example-key-name"}).
+      returns stub(:body => @delete_keypair_body, :is_a? => true)
+
+    lambda { @ec2.delete_keypair( :key_name => "example-key-name" ) }.should.not.raise(AWS::ArgumentError)
+    lambda { @ec2.delete_keypair() }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.delete_keypair( :key_name => nil ) }.should.raise(AWS::ArgumentError)
+    lambda { @ec2.delete_keypair( :key_name => "" ) }.should.raise(AWS::ArgumentError)
+  end
+
+
+end