boxgrinder-build 0.8.1 → 0.9.0

data/spec/plugins/delivery/ebs/ebs-plugin-spec.rb

@@ -0,0 +1,230 @@
+#
+# 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 'logger'
+require 'boxgrinder-build/plugins/delivery/ebs/ebs-plugin'
+require 'hashery/opencascade'
+
+module BoxGrinder
+
+  describe EBSPlugin do
+    before(:all) do
+      @arch = `uname -m`.chomp.strip
+    end
+
+    def prepare_plugin
+      @plugin = EBSPlugin.new
+
+      yield @plugin if block_given?
+
+      @config = mock('Config')
+      @config.stub!(:delivery_config).and_return({})
+      plugins = mock('Plugins')
+      plugins.stub!(:[]).with('ebs').and_return({})
+      @config.stub!(:[]).with(:plugins).and_return(plugins)
+
+      @appliance_config = mock('ApplianceConfig')
+
+      @appliance_config.stub!(:name).and_return('appliance_name')
+      @appliance_config.stub!(:version).and_return(1)
+      @appliance_config.stub!(:release).and_return(0)
+      @appliance_config.stub!(:os).and_return(OpenCascade.new({:name => 'fedora', :version => '14'}))
+      @appliance_config.stub!(:hardware).and_return(OpenCascade.new({:arch => 'x86_64', :base_arch => 'x86_64'}))
+      @appliance_config.stub!(:path).and_return(OpenCascade.new({:build => '/a/build/path'}))
+
+      @plugin = @plugin.init(
+          @config,
+          @appliance_config,
+          :log => Logger.new('/dev/null'),
+          :plugin_info => {:class => BoxGrinder::EBSPlugin, :type => :delivery, :name => :ebs, :full_name => "Elastic Block Storage"},
+          :config_file => "#{File.dirname(__FILE__)}/ebs.yaml"
+      )
+
+      @plugin_config = @plugin.instance_variable_get(:@plugin_config).merge(
+          {
+              'access_key' => 'access_key',
+              'secret_access_key' => 'secret_access_key',
+              'bucket' => 'bucket',
+              'account_number' => '0000-0000-0000',
+              'cert_file' => '/path/to/cert/file',
+              'key_file' => '/path/to/key/file'
+          }
+      )
+
+      @plugin.instance_variable_set(:@plugin_config, @plugin_config)
+    end
+
+    it "should register all operating systems with specific versions" do
+      prepare_plugin do |plugin|
+        avaibility_zone = mock('AZ')
+        avaibility_zone.should_receive(:string).and_return('avaibility-zone1')
+
+        plugin.should_receive(:open).with('http://169.254.169.254/latest/meta-data/placement/availability-zone').and_return(avaibility_zone)
+      end
+
+      supportes_oses = @plugin.instance_variable_get(:@supported_oses)
+
+      supportes_oses.size.should == 2
+      supportes_oses.keys.sort.should == ['fedora', 'rhel']
+      supportes_oses['rhel'].should == ['6']
+      supportes_oses['fedora'].should == ['13', '14', '15']
+    end
+
+    describe ".after_init" do
+      it "should set default avaibility zone to current one" do
+        prepare_plugin do |plugin|
+          avaibility_zone = mock('AZ')
+          avaibility_zone.should_receive(:string).and_return('avaibility-zone1')
+
+          plugin.should_receive(:open).with('http://169.254.169.254/latest/meta-data/placement/availability-zone').and_return(avaibility_zone)
+        end
+
+        @plugin.instance_variable_get(:@plugin_config)['availability_zone'].should == 'avaibility-zone1'
+      end
+
+      it "should not set default avaibility zone because we're not on EC2" do
+        prepare_plugin do |plugin|
+          plugin.should_receive(:open).with('http://169.254.169.254/latest/meta-data/placement/availability-zone').and_raise("Bleh")
+        end
+
+        @plugin.instance_variable_get(:@plugin_config)['availability_zone'].should == nil
+      end
+    end
+
+    describe '.already_registered?' do
+      it "should check if image is already registered and return false if there are no images registered for this account" do
+        prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+        plugin_config = mock('PluginConfiig')
+        plugin_config.should_receive(:[]).with('account_number').and_return('0000-0000-0000')
+
+        @plugin.instance_variable_set(:@plugin_config, plugin_config)
+
+        ec2 = mock('EC2')
+        ec2.should_receive(:describe_images).with(:owner_id => '000000000000')
+
+        @plugin.instance_variable_set(:@ec2, ec2)
+
+        @plugin.already_registered?('aname').should == false
+      end
+
+      it "should check if image is already registered and return false if there are no images with name aname_new" do
+        prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+        plugin_config = mock('PluginConfiig')
+        plugin_config.should_receive(:[]).with('account_number').and_return('0000-0000-0000')
+
+        @plugin.instance_variable_set(:@plugin_config, plugin_config)
+
+        ec2 = mock('EC2')
+        ec2.should_receive(:describe_images).with(:owner_id => '000000000000').and_return({'imagesSet' => {'item' => [{'name' => 'abc', 'imageId' => '1'}, {'name' => 'aname', 'imageId' => '2'}]}})
+
+        @plugin.instance_variable_set(:@ec2, ec2)
+
+        @plugin.already_registered?('aname_new').should == false
+      end
+
+      it "should check if image is already registered and return true image is registered" do
+        prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+        plugin_config = mock('PluginConfiig')
+        plugin_config.should_receive(:[]).with('account_number').and_return('0000-0000-0000')
+
+        @plugin.instance_variable_set(:@plugin_config, plugin_config)
+
+        ec2 = mock('EC2')
+        ec2.should_receive(:describe_images).with(:owner_id => '000000000000').and_return({'imagesSet' => {'item' => [{'name' => 'abc', 'imageId' => '1'}, {'name' => 'aname', 'imageId' => '2'}]}})
+
+        @plugin.instance_variable_set(:@ec2, ec2)
+
+        @plugin.already_registered?('aname').should == '2'
+      end
+    end
+
+    it "should adjust fstab" do
+      prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+      exec_helper = @plugin.instance_variable_get(:@exec_helper)
+
+      exec_helper.should_receive(:execute).with("cat a/dir/etc/fstab | grep -v '/mnt' | grep -v '/data' | grep -v 'swap' > a/dir/etc/fstab.new")
+      exec_helper.should_receive(:execute).with("mv a/dir/etc/fstab.new a/dir/etc/fstab")
+
+      @plugin.adjust_fstab('a/dir')
+    end
+
+    it "should get a new free device" do
+      prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+      File.should_receive(:exists?).with("/dev/sdf").and_return(false)
+      File.should_receive(:exists?).with("/dev/xvdf").and_return(false)
+
+      @plugin.free_device_suffix.should == "f"
+    end
+
+    it "should get a new free device next in order" do
+      prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+      File.should_receive(:exists?).with("/dev/sdf").and_return(false)
+      File.should_receive(:exists?).with("/dev/xvdf").and_return(true)
+      File.should_receive(:exists?).with("/dev/sdg").and_return(false)
+      File.should_receive(:exists?).with("/dev/xvdg").and_return(false)
+
+      @plugin.free_device_suffix.should == "g"
+    end
+
+    it "should should return true if on EC2" do
+      prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+      @plugin.should_receive(:open).with("http://169.254.169.254/1.0/meta-data/local-ipv4")
+
+      @plugin.valid_platform?.should == true
+    end
+
+    it "should should return true if NOT on EC2" do
+      prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+      @plugin.should_receive(:open).with("http://169.254.169.254/1.0/meta-data/local-ipv4").and_raise("Bleh")
+
+      @plugin.valid_platform?.should == false
+    end
+
+    describe ".ebs_appliance_name" do
+      it "should return basic appliance name" do
+        prepare_plugin { |plugin| plugin.stub!(:after_init) }
+        @plugin.ebs_appliance_name.should == "appliance_name/fedora/14/1.0/x86_64"
+      end
+
+      it "should return 2nd snapshot of appliance" do
+        prepare_plugin { |plugin| plugin.stub!(:after_init) }
+
+        @plugin_config.merge!('snapshot' => true)
+
+        ec2 = mock('EC2')
+        ec2.should_receive(:describe_images).twice.with(:owner_id => '000000000000').and_return({'imagesSet' => {'item' => [
+            {'imageId' => '1', 'name' => 'appliance_name/fedora/14/1.0/x86_64'},
+            {'imageId' => '2', 'name' => 'appliance_name/fedora/14/1.0-SNAPSHOT-1/x86_64'}
+        ]}})
+
+        @plugin.instance_variable_set(:@ec2, ec2)
+
+        @plugin.ebs_appliance_name.should == "appliance_name/fedora/14/1.0-SNAPSHOT-2/x86_64"
+      end
+    end
+  end
+end
+

data/spec/plugins/delivery/ebs/ebs.yaml

@@ -0,0 +1,3 @@
+access_key: access_key
+secret_access_key: secret_access_key
+account_number: 0000-0000-0000

data/spec/plugins/delivery/local/local-plugin-spec.rb

@@ -0,0 +1,133 @@
+#
+# 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 'boxgrinder-build/plugins/delivery/local/local-plugin'
+require 'hashery/opencascade'
+
+module BoxGrinder
+  describe LocalPlugin do
+
+    before(:each) do
+      @config = mock('Config')
+      @config.stub!(:delivery_config).and_return({})
+      plugins = mock('Plugins')
+      plugins.stub!(:[]).with('local').and_return({})
+      @config.stub!(:[]).with(:plugins).and_return(plugins)
+
+      @appliance_config = mock('ApplianceConfig')
+
+      @appliance_config.stub!(:path).and_return(OpenCascade.new({:build => 'build/path'}))
+      @appliance_config.stub!(:name).and_return('appliance')
+      @appliance_config.stub!(:version).and_return(1)
+      @appliance_config.stub!(:release).and_return(0)
+      @appliance_config.stub!(:os).and_return(OpenCascade.new({:name => :fedora, :version => '13'}))
+      @appliance_config.stub!(:hardware).and_return(OpenCascade.new({:arch => 'x86_64'}))
+
+      @plugin = LocalPlugin.new.init(@config, @appliance_config,
+                                     :log => Logger.new('/dev/null'),
+                                     :plugin_info => {:class => BoxGrinder::LocalPlugin, :type => :delivery, :name => :local, :full_name => "Local file system"},
+                                     :previous_deliverables => {:disk => "a_disk.raw", :metadata => 'metadata.xml'}
+      )
+
+      @config = @plugin.instance_variable_get(:@config)
+      @appliance_config = @plugin.instance_variable_get(:@appliance_config)
+      @exec_helper = @plugin.instance_variable_get(:@exec_helper)
+      @log = @plugin.instance_variable_get(:@log)
+      @dir = @plugin.instance_variable_get(:@dir)
+    end
+
+    describe ".execute" do
+
+      it "should package and deliver the appliance" do
+        @plugin.instance_variable_set(:@plugin_config, {
+            'overwrite' => false,
+            'path' => 'a/path',
+            'package' => true
+        })
+
+        FileUtils.should_receive(:mkdir_p).with('a/path')
+        package_helper = mock(PackageHelper)
+        package_helper.should_receive(:package).with('.', "build/path/local-plugin/tmp/appliance-1.0-fedora-13-x86_64-raw.tgz").and_return("deliverable")
+
+        PackageHelper.should_receive(:new).with(@config, @appliance_config, :log => @log, :exec_helper => @exec_helper).and_return(package_helper)
+
+        @exec_helper.should_receive(:execute).with("cp 'build/path/local-plugin/tmp/appliance-1.0-fedora-13-x86_64-raw.tgz' 'a/path'")
+        @plugin.should_receive(:deliverables_exists?).and_return(false)
+
+        @plugin.execute
+      end
+
+      it "should not package, but deliver the appliance" do
+        @plugin.instance_variable_set(:@plugin_config, {
+            'overwrite' => true,
+            'path' => 'a/path',
+            'package' => false
+        })
+
+        FileUtils.should_receive(:mkdir_p).with('a/path')
+        PackageHelper.should_not_receive(:new)
+
+        @exec_helper.should_receive(:execute).with("cp 'a_disk.raw' 'a/path'")
+        @exec_helper.should_receive(:execute).with("cp 'metadata.xml' 'a/path'")
+
+        @plugin.execute
+      end
+
+      it "should not deliver the package, because it is already delivered" do
+        @plugin.instance_variable_set(:@plugin_config, {
+            'overwrite' => false,
+            'path' => 'a/path',
+            'package' => false
+        })
+
+        PackageHelper.should_not_receive(:new)
+
+        @exec_helper.should_not_receive(:execute)
+        @plugin.should_receive(:deliverables_exists?).and_return(true)
+
+        @plugin.execute
+      end
+    end
+
+    describe ".deliverables_exists?" do
+      it "should return true for package" do
+        @plugin.instance_variable_set(:@plugin_config, {
+            'path' => 'a/path',
+            'package' => true
+        })
+
+        File.should_receive(:exists?).with('a/path/appliance-1.0-fedora-13-x86_64-raw.tgz').and_return(true)
+
+        @plugin.deliverables_exists?.should == true
+      end
+
+      it "should return true for non-packaged appliance" do
+        @plugin.instance_variable_set(:@plugin_config, {
+            'path' => 'a/path',
+            'package' => false
+        })
+
+        File.should_receive(:exists?).with('a/path/a_disk.raw').and_return(true)
+        File.should_receive(:exists?).with('a/path/metadata.xml').and_return(true)
+
+        @plugin.deliverables_exists?.should == true
+      end
+    end
+  end
+end
+