boxgrinder-build 0.9.3 → 0.9.4

data/spec/appliance-spec.rb

@@ -17,6 +17,7 @@
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
 require 'rubygems'
+require 'rspec'
 require 'boxgrinder-build/appliance'
 require 'ostruct'
 require 'logger'

data/spec/helpers/augeas-helper-spec.rb

@@ -17,6 +17,7 @@
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
 require 'rubygems'
+require 'rspec'
 require 'boxgrinder-build/helpers/augeas-helper'
 
 module BoxGrinder

data/spec/helpers/ec2-helper-spec.rb

@@ -0,0 +1,260 @@
+#
+# Copyright 2010 Red Hat, Inc.
+#
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 3 of
+# the License, or (at your option) any later version.
+#
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+require 'rspec'
+require 'boxgrinder-build/helpers/ec2-helper'
+require 'aws-sdk'
+
+module BoxGrinder
+  describe EC2Helper do
+
+    FAST_POLL = 0.1
+    FAST_TO   = 1
+
+    before(:each) do
+      AWS.stub!
+      AWS.config({:access_key_id => '', :secret_access_key => ''})
+      @ec2 = AWS::EC2.new()
+      @ec2helper = EC2Helper.new(@ec2)
+      @ami = mock(AWS::EC2::Image)
+      @instance = mock(AWS::EC2::Instance)
+      @snapshot = mock(AWS::EC2::Snapshot)
+      @volume = mock(AWS::EC2::Volume)
+    end
+
+    describe ".wait_for_image_state" do
+      before(:each) do
+        @ami.stub!(:state).and_return(:ham, :biscuit, :available)
+      end
+
+      it "should wait until the image state is :available before returning" do
+        @ami.stub(:exists?).and_return(true)
+
+        @ami.should_receive(:exists?).once.and_return(true)
+        @ami.should_receive(:state).exactly(3).times
+        @ec2helper.wait_for_image_state(:available, @ami, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+
+      it "should wait until the image exists before sampling for state" do
+        @ami.stub(:exists?).and_return(false, false, true)
+
+        @ami.should_receive(:exists?).once.exactly(3).times
+        @ami.should_receive(:state).exactly(3).times
+        @ec2helper.wait_for_image_state(:available, @ami, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+    end
+
+    describe ".wait_for_image_death" do
+      it "should wait until image no longer exists" do
+        @ami.stub!(:exists?).and_return(true, true, false)
+
+        @ami.should_receive(:exists?).exactly(3).times
+        @ec2helper.wait_for_image_death(@ami, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+
+      it "should return normally if InvalidImageID::NotFound error occurs" do
+        @ami.stub!(:exists?).and_raise(aws_sdk_exception_hack(AWS::EC2::Errors::InvalidImageID::NotFound))
+
+        @ami.should_receive(:exists?).once
+        @ec2helper.wait_for_image_death(@ami, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+    end
+
+    describe ".wait_for_instance_status" do
+      it "should wait until the instance status is :available before returning" do
+        @instance.stub!(:status).and_return(:bacon, :buttie, :available)
+
+        @instance.should_receive(:status).exactly(3).times
+        @ec2helper.wait_for_instance_status(:available, @instance, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+    end
+
+    describe ".wait_for_instance_death" do
+      it "should wait until instance status is :terminated before returning" do
+        @instance.stub!(:exists?).and_return(true)
+        @instance.stub!(:status).and_return(:stottie, :bread, :terminated)
+
+        @instance.should_receive(:status).exactly(3).times
+        @ec2helper.wait_for_instance_death(@instance, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+
+      it "should return normally if InvalidInstance::NotFound error occurs" do
+        @instance.stub!(:exists?).and_return(true)
+        @instance.stub!(:status).and_raise(aws_sdk_exception_hack(AWS::EC2::Errors::InvalidInstanceID::NotFound))
+
+        @instance.should_receive(:exists?).once
+        @instance.should_receive(:status).once
+        @ec2helper.wait_for_instance_death(@instance, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+
+      it "should return immediately if the instance does not appear to exist at all" do
+        @instance.stub!(:exists?).and_return(false)
+
+        @instance.should_receive(:exists?).once
+        @ec2helper.wait_for_instance_death(@instance, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+    end
+
+    describe ".wait_for_snapshot_status" do
+      it "should wait until the instance status is :completed before returning" do
+        @snapshot.stub!(:status).and_return(:crumpet, :tea, :completed, :cricket)
+        @snapshot.stub!(:progress).and_return(99)
+
+        @snapshot.should_receive(:status).exactly(3).times
+        @ec2helper.wait_for_snapshot_status(:completed, @snapshot, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+    end
+
+    describe ".wait_for_volume_status" do
+      it "should wait until the instance status is :available before returning" do
+        @volume.stub!(:status).and_return(:edictum, :imperatoris, :available)
+
+        @volume.should_receive(:status).exactly(3).times
+        @ec2helper.wait_for_volume_status(:available, @volume, :frequency => FAST_POLL, :timeout => FAST_TO)
+      end
+    end
+
+    describe ".current_availability_zone" do
+      it "should return the current availability zone" do
+        EC2Helper.stub!(:get_meta_data).and_return('protuberant-potato-1a')
+
+        EC2Helper.should_receive(:get_meta_data).
+            with('/latest/meta-data/placement/availability-zone/').
+            and_return('protuberant-potato-1a')
+        EC2Helper.current_availability_zone.should == 'protuberant-potato-1a'
+      end
+    end
+
+    describe ".current_instance_id" do
+      it "should return the current instance id" do
+        EC2Helper.stub!(:get_meta_data).and_return('voluminous-verbiage')
+
+        EC2Helper.should_receive(:get_meta_data).
+            with('/latest/meta-data/instance-id').
+            and_return('voluminous-verbiage')
+        EC2Helper.current_instance_id.should == 'voluminous-verbiage'
+      end
+    end
+
+    describe ".availability_zone_to_region" do
+      it "should convert an availability zone to a region" do
+        EC2Helper.availability_zone_to_region('protuberant-potato-1a').
+            should == 'protuberant-potato-1'
+      end
+    end
+
+    describe ".ami_by_name" do
+      before(:each) do
+        @ec2.stub!(:images)
+      end
+
+      it "should query for an image filtered by name" do
+        @ec2.should_receive(:images)
+        @ec2.images.should_receive(:with_owner).with('987654321')
+        @ec2.images.should_receive(:filter).with('name','inauspicious-interlocution').and_return([@ami])
+
+        @ec2helper.ami_by_name('inauspicious-interlocution', '987654321').should == @ami
+      end
+
+      it "should return nil if no results are returned" do
+        @ec2.should_receive(:images)
+        @ec2.images.should_receive(:with_owner).with('987654321')
+        @ec2.images.should_receive(:filter).with('name','inauspicious-interlocution').and_return([])
+
+        @ec2helper.ami_by_name('inauspicious-interlocution', '987654321').should == nil
+      end
+    end
+
+    describe ".snapshot_by_id" do
+      before(:each) do
+        snap_mock = mock('snapshots')
+        @ec2.stub!(:snapshots).and_return(snap_mock)
+      end
+
+      it "should query for a snapshot by snapshot-id" do
+
+        @ec2.snapshots.
+            should_receive(:filter).
+            with('snapshot-id','count-bezukhov').
+            and_return([@snapshot])
+
+        @ec2helper.snapshot_by_id('count-bezukhov').should == @snapshot
+
+      end
+
+      it "should return nil if no results are returned" do
+        @ec2.snapshots.
+            should_receive(:filter).
+            with('snapshot-id','count-bezukhov').
+            and_return([])
+
+        @ec2helper.snapshot_by_id('count-bezukhov').should == nil
+      end
+
+    end
+
+    describe ".live_instances" do
+      before(:each) do
+        instances_mock = mock('instances')
+        @ec2.stub!(:instances).and_return(instances_mock)
+
+        @instance_1 = mock(AWS::EC2::Instance)
+        @instance_2 = mock(AWS::EC2::Instance)
+        @instance_3 = mock(AWS::EC2::Instance)
+
+        @instance_1.stub!(:status).and_return(:available)
+        @instance_2.stub!(:status).and_return(:available)
+        @ami.stub!(:id).and_return('war-and-peace')
+      end
+
+      it "should query for live instances by ami" do
+        @ec2.instances.
+            should_receive(:filter).
+            with('image-id','war-and-peace').
+            and_return([@instance_1, @instance_2])
+
+        @ec2helper.live_instances(@ami).should == [@instance_1, @instance_2]
+      end
+
+      it "should ignore any :terminated instances" do
+        @instance_3.stub!(:status).and_return(:terminated)
+
+        @ec2.should_receive(:instances)
+        @ec2.instances.should_receive(:filter).
+            with('image-id','war-and-peace').
+            and_return([@instance_3, @instance_2, @instance_1])
+
+        @ec2helper.live_instances(@ami).should == [@instance_2, @instance_1]
+      end
+
+    end
+
+    describe ".endpoints" do
+
+    end
+
+    # Complex auto-generated exceptions means we can't use the normal method of raising
+    def aws_sdk_exception_hack(klass)
+      mock_param = mock('fake_param')
+      mock_param.stub!(:status).and_return(1)
+      mock_param.stub!(:body).and_return('')
+      klass.new(mock_param, mock_param)
+    end
+
+  end
+end

data/spec/helpers/guestfs-helper-spec.rb

@@ -16,18 +16,22 @@
 # Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
-require 'rubygems'
 require 'boxgrinder-build/helpers/guestfs-helper'
+require 'rubygems'
+require 'rspec'
+require 'hashery/opencascade'
 
 module BoxGrinder
   describe GuestFSHelper do
     before(:each) do
+      ENV.delete("LIBGUESTFS_MEMSIZE")
+
       @log = Logger.new('/dev/null')
       @appliance_config = mock('ApplianceConfig')
       @appliance_config.stub!(:hardware).and_return(:partitions => {})
 
       @config = mock('Config')
-      @config.stub!(:dir).and_return(:tmp => '/tmp')
+      @config.stub!(:dir).and_return(OpenCascade.new(:tmp => '/tmp'))
 
       @helper = GuestFSHelper.new('a/raw/disk', @appliance_config, @config, :log => @log)
     end
@@ -41,6 +45,7 @@ module BoxGrinder
         guestfs.should_receive(:set_verbose)
         guestfs.should_receive(:set_trace)
         guestfs.should_receive(:set_selinux).with(1)
+        guestfs.should_receive(:set_memsize).with(300)
 
         @helper.should_receive(:hw_virtualization_available?).and_return(true)
 
@@ -59,6 +64,7 @@ module BoxGrinder
         guestfs.should_receive(:set_verbose)
         guestfs.should_receive(:set_trace)
         guestfs.should_receive(:set_selinux).with(1)
+        guestfs.should_receive(:set_memsize).with(300)
 
         @helper.should_receive(:hw_virtualization_available?).and_return(true)
 
@@ -76,6 +82,7 @@ module BoxGrinder
         guestfs.should_receive(:set_verbose)
         guestfs.should_receive(:set_trace)
         guestfs.should_receive(:set_selinux).with(1)
+        guestfs.should_receive(:set_memsize).with(300)
 
         @helper.should_receive(:hw_virtualization_available?).and_return(false)
 
@@ -85,6 +92,26 @@ module BoxGrinder
 
         @helper.prepare_guestfs {}
       end
+      
+      it "should prepare guestfs with custom memory settings" do
+        ENV['LIBGUESTFS_MEMSIZE'] = "500"
+
+        guestfs = mock('Guestfs')
+        @helper.instance_variable_set(:@guestfs, guestfs)
+
+        guestfs.should_receive(:set_append).with('noapic')
+        guestfs.should_receive(:set_verbose)
+        guestfs.should_receive(:set_trace)
+        guestfs.should_receive(:set_selinux).with(1)
+        guestfs.should_receive(:set_memsize).with(500)
+
+        @helper.should_receive(:hw_virtualization_available?).and_return(true)
+
+        guestfs.should_receive(:add_drive).with('a/raw/disk')
+        guestfs.should_receive(:set_network).with(1)
+
+        @helper.prepare_guestfs {}
+      end
     end
 
     describe ".initialize_guestfs" do
@@ -129,7 +156,7 @@ module BoxGrinder
       end
 
       it "should run guestfs with two partitions" do
-        @appliance_config.stub!(:hardware).and_return(:partitions => {'/' => nil, '/home' => nil})
+        @appliance_config.stub!(:hardware).and_return(OpenCascade.new(:partitions => {'/' => nil, '/home' => nil}))
 
         guestfs = mock('Guestfs')
         @helper.instance_variable_set(:@guestfs, guestfs)
@@ -145,7 +172,7 @@ module BoxGrinder
       end
 
       it "should run guestfs with no partitions and don't load selinux" do
-        @appliance_config.stub!(:hardware).and_return(:partitions => {'/' => nil, '/home' => nil})
+        @appliance_config.stub!(:hardware).and_return(OpenCascade.new(:partitions => {'/' => nil, '/home' => nil}))
 
         guestfs = mock('Guestfs')
         @helper.instance_variable_set(:@guestfs, guestfs)
@@ -199,7 +226,7 @@ module BoxGrinder
       it "should mount two partitions" do
         guestfs = mock('Guestfs')
 
-        @appliance_config.stub!(:hardware).and_return(:partitions => {'/' => nil, '/home' => nil})
+        @appliance_config.stub!(:hardware).and_return(OpenCascade.new(:partitions => {'/' => nil, '/home' => nil}))
         guestfs.should_receive(:list_partitions).and_return(['/dev/vda1', '/dev/vda2'])
 
         @helper.should_receive(:mount_partition).with('/dev/vda1', '/', '')
@@ -212,7 +239,7 @@ module BoxGrinder
       it "should mount partitions with extended partitions" do
         guestfs = mock('Guestfs')
 
-        @appliance_config.stub!(:hardware).and_return(:partitions => {'/' => nil, '/home' => nil, '/var/www' => nil, '/var/mock' => nil})
+        @appliance_config.stub!(:hardware).and_return(OpenCascade.new(:partitions => {'/' => nil, '/home' => nil, '/var/www' => nil, '/var/mock' => nil}))
         guestfs.should_receive(:list_partitions).and_return(['/dev/vda1', '/dev/vda2', '/dev/vda3', '/dev/vda4', '/dev/vda5'])
 
         @helper.should_receive(:mount_partition).with('/dev/vda1', '/', '')
@@ -370,4 +397,4 @@ module BoxGrinder
       end
     end
   end
-end
+end

data/spec/helpers/image-helper-spec.rb

@@ -17,6 +17,7 @@
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
 require 'rubygems'
+require 'rspec'
 require 'boxgrinder-build/helpers/image-helper'
 
 module BoxGrinder

data/spec/helpers/linux-helper-spec.rb

@@ -17,6 +17,7 @@
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
 require 'rubygems'
+require 'rspec'
 require 'boxgrinder-build/helpers/linux-helper'
 
 module BoxGrinder

data/spec/helpers/package-helper-spec.rb

@@ -17,6 +17,7 @@
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
 require 'rubygems'
+require 'rspec'
 require 'logger'
 require 'boxgrinder-build/helpers/package-helper'
 

data/spec/helpers/plugin-helper-spec.rb

@@ -17,6 +17,7 @@
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
 require 'rubygems'
+require 'rspec'
 require 'boxgrinder-core/helpers/log-helper'
 require 'boxgrinder-build/helpers/plugin-helper'
 require 'ostruct'

data/spec/helpers/s3-helper-spec.rb

@@ -0,0 +1,168 @@
+#
+# Copyright 2010 Red Hat, Inc.
+#
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 3 of
+# the License, or (at your option) any later version.
+#
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+require 'rubygems'
+require 'rspec'
+require 'boxgrinder-build/helpers/s3-helper'
+require 'aws-sdk'
+
+module BoxGrinder
+  describe S3Helper do
+
+    before(:each) do
+      AWS.stub!
+      AWS.config({:access_key_id => '', :secret_access_key => ''})
+      @ec2 = AWS::EC2.new
+      @s3 = AWS::S3.new
+      @ec2helper = EC2Helper.new(@ec2)
+      @s3helper = S3Helper.new(@ec2, @s3)
+      @s3obj = mock(AWS::S3::S3Object)
+      @bucket = mock(AWS::S3::Bucket)
+    end
+
+    describe ".bucket" do
+
+      it "should return the existing bucket if it already exists" do
+          @bucket.stub!(:exists).and_return(true)
+          @s3.stub_chain(:buckets, :[]).and_return(@bucket)
+
+          @s3.buckets.should_receive(:[]).with('tolstoy').and_return(@bucket)
+          @bucket.should_receive(:exists?).and_return(true)
+
+          @s3helper.bucket(:bucket => 'tolstoy').should == @bucket
+      end
+
+      context "when the bucket does not yet exist"  do
+
+        before(:each) do
+          @bucket.stub!(:exists).and_return(false)
+          @s3.stub_chain(:buckets, :[]).and_return(@bucket)
+          @s3.stub_chain(:buckets, :create).and_return(@bucket_real)
+
+          @bucket_real = mock(AWS::S3::Bucket)
+          @bucket_real.stub!(:exists).and_return(true)
+        end
+
+        context ":create_if_missing" do
+
+          it "should return nil if :create_if_missing is not set" do
+            @s3.buckets.should_receive(:[]).with('tolstoy').and_return(@bucket)
+            @bucket.should_receive(:exists?).once.and_return(false)
+            @s3helper.bucket(:bucket => 'tolstoy').should == nil
+          end
+
+          context "When :create_if_missing is set" do
+
+            it "should return a bucket, with other settings defaulted" do
+              @s3.buckets.should_receive(:[]).with('tolstoy').and_return(@bucket)
+              @bucket.should_receive(:exists?).once.and_return(false)
+              @s3.buckets.
+                  should_receive(:create).
+                  once.
+                  with('tolstoy', :acl => :private, :location_constraint => 'us-east-1').
+                  and_return(@bucket_real)
+
+              @s3helper.bucket(:bucket => 'tolstoy', :create_if_missing => true).should == @bucket_real
+            end
+
+            it "should return a bucket assigned with the attributes provided by the user" do
+                @s3.buckets.should_receive(:[]).with('tolstoy').and_return(@bucket)
+                @bucket.should_receive(:exists?).once.and_return(false)
+                @s3.buckets.
+                    should_receive(:create).
+                    once.
+                    with('tolstoy', :acl => :ham, :location_constraint => 'biscuit').
+                    and_return(@bucket_real)
+
+                @s3helper.bucket(:bucket => 'tolstoy',  :create_if_missing => true,
+                                 :acl => :ham, :location_constraint => 'biscuit').should == @bucket_real
+            end
+
+          end
+        end
+      end
+    end
+
+    describe ".object_exists?"  do
+
+      it "should return true if the object exists" do
+        @s3obj.stub!(:exists?).and_return(true)
+        @s3obj.should_receive(:exists?).and_return(true)
+        @s3helper.object_exists?(@s3obj).should == true
+      end
+
+      it "should return false if the object does not exist" do
+        @s3obj.stub!(:exists?).and_return(false)
+        @s3obj.should_receive(:exists?).and_return(false)
+        @s3helper.object_exists?(@s3obj).should == false
+      end
+
+    end
+
+    describe ".delete_folder" do
+
+      it "should delete a folder from a bucket" do
+        object = mock(AWS::S3::S3Object)
+        objects = [object]
+
+        @bucket.stub!(:objects)
+        @bucket.objects.
+            should_receive(:with_prefix).
+            with('cacoethes/carpendi').
+            and_return(objects)
+
+        object.should_receive(:delete).once
+
+        @s3helper.delete_folder(@bucket, '/cacoethes/carpendi/')
+      end
+
+    end
+
+    describe ".stub_s3obj" do
+
+      it "should return a stub AWS::S3::S3Object associated with the specified bucket" do
+        object = mock(AWS::S3::S3Object)
+        objects = mock(AWS::S3::ObjectCollection)
+        objects.stub!(:[]).and_return(object)
+        objects.stub!(:to_sym).and_return(:anything)
+
+        @bucket.stub!(objects).and_return(objects)
+        @bucket.should_receive(:objects).and_return(objects)
+        objects.should_receive(:[]).and_return(object)
+
+        @s3helper.stub_s3obj(@bucket, 'vini/vidi/vici') == @s3obj
+      end
+
+    end
+
+    describe ".parse_path" do
+
+      it "should return root if is empty" do
+        @s3helper.parse_path('').should == '/'
+      end
+
+      it "should remove extranous slashes" do
+        @s3helper.parse_path('//cave/canem//').
+            should == 'cave/canem/'
+      end
+
+    end
+
+
+  end
+end