/spec/unit/plugins/cloud_v2_spec.rb

#
# Author:: Cary Penniman (<cary@rightscale.com>)
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper.rb')
require 'ipaddr'

describe "CloudAttrs object" do
  before(:each) do
    @plugin = get_plugin("cloud_v2")
  end

  let(:cloud_node) do
    {"public_ipv4_addrs"=>["1.2.3.1"],
     "local_ipv4_addrs"=>["1.2.4.1"],
     "public_ipv6_addrs"=>["3ffe:505:2::1"],
     "local_ipv6_addrs"=>["3ffe:506:2::1"],
     "public_ipv4"=>"1.2.3.1",
     "local_ipv4"=>"1.2.4.1",
     "public_ipv6"=>"3ffe:505:2::1",
     "local_ipv6"=>"3ffe:506:2::1",
     "public_hostname"=>"myhost.somewhere.com",
     "local_hostname"=>"my-localhost",
     "provider"=>"my_awesome_cloud"
     }
  end

  it "populates cloud mash" do
    @cloud_attr_obj = ::CloudAttrs.new()
    @cloud_attr_obj.add_ipv4_addr("1.2.3.1", :public)
    @cloud_attr_obj.add_ipv4_addr("1.2.4.1", :private)
    @cloud_attr_obj.add_ipv6_addr("3ffe:505:2::1", :public)
    @cloud_attr_obj.add_ipv6_addr("3ffe:506:2::1", :private)
    @cloud_attr_obj.public_hostname = "myhost.somewhere.com"
    @cloud_attr_obj.local_hostname = "my-localhost"
    @cloud_attr_obj.provider = "my_awesome_cloud"
    @cloud_attr_obj.cloud_mash.should == cloud_node
  end

  it "throws exception with a bad ipv4 address" do
    @cloud_attr_obj = ::CloudAttrs.new()
    lambda {  @cloud_attr_obj.add_ipv6_addr("somebogusstring", :public) }.should raise_error
  end

  it "throws exception with a bad ipv6 address" do
    @cloud_attr_obj = ::CloudAttrs.new()
    lambda {  @cloud_attr_obj.add_ipv6_addr("FEED:B0B:DEAD:BEEF", :public)  }.should raise_error
  end

  it "throws exception with ipv6 address passed to ipv4" do
    @cloud_attr_obj = ::CloudAttrs.new()
    lambda {  @cloud_attr_obj.add_ipv4_addr("3ffe:506:2::1", :public) }.should raise_error
  end

  it "throws exception with ipv4 address passed to ipv6" do
    @cloud_attr_obj = ::CloudAttrs.new()
    lambda {  @cloud_attr_obj.add_ipv6_addr("1.2.3.4", :public) }.should raise_error
  end


end

describe Ohai::System, "plugin cloud" do
  before(:each) do
    @plugin = get_plugin("cloud_v2")
  end

  describe "with no cloud mashes" do
    it "doesn't populate the cloud data" do
      @plugin[:ec2] = nil
      @plugin[:rackspace] = nil
      @plugin[:eucalyptus] = nil
      @plugin[:linode] = nil
      @plugin[:azure] = nil
      @plugin[:gce] = nil
      @plugin.run
      @plugin[:cloud_v2].should be_nil
    end
  end

  describe "with EC2 mash" do
    before do
      @plugin[:ec2] = Mash.new()
    end

    it "populates cloud public ip" do
      @plugin[:ec2]['public_ipv4'] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs][0].should == @plugin[:ec2]['public_ipv4']
    end

    it "populates cloud private ip" do
      @plugin[:ec2]['local_ipv4'] = "10.252.42.149"
      @plugin.run
      @plugin[:cloud_v2][:local_ipv4_addrs][0].should == @plugin[:ec2]['local_ipv4']
    end

    it "populates cloud provider" do
      @plugin.run
      @plugin[:cloud_v2][:provider].should == "ec2"
    end
  end

  describe "with GCE mash" do
    before do
      @plugin[:gce] = Mash.new()
      @plugin[:gce]['instance'] = Mash.new()
      @plugin[:gce]['instance']['networkInterfaces'] = [
        {
          "accessConfigs" => [ {"externalIp" => "8.35.198.173", "type"=>"ONE_TO_ONE_NAT"} ],
          "ip" => "10.240.0.102",
          "network"=> "projects/foo/networks/default"
        }
      ]
    end

    it "populates cloud public ip" do
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs][0].should == "8.35.198.173"
    end

    it "populates cloud private ip" do
      @plugin.run
      @plugin[:cloud_v2][:local_ipv4_addrs][0].should == "10.240.0.102"
    end

    it "populates cloud provider" do
      @plugin.run
      @plugin[:cloud_v2][:provider].should == "gce"
    end
  end

  describe "with rackspace" do
    before do
      @plugin[:rackspace] = Mash.new()
    end

    it "populates cloud public ip" do
      @plugin[:rackspace][:public_ipv4] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4].should == @plugin[:rackspace][:public_ipv4]
    end

    it "populates cloud public ipv6" do
      @plugin[:rackspace][:public_ipv6] = "2a00:1a48:7805:111:e875:efaf:ff08:75"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv6].should == @plugin[:rackspace][:public_ipv6]
    end

    it "populates cloud private ip" do
      @plugin[:rackspace][:local_ipv4] = "10.252.42.149"
      @plugin.run
      @plugin[:cloud_v2][:local_ipv4].should == @plugin[:rackspace][:local_ipv4]
    end

    it "populates cloud private ipv6" do
      @plugin[:rackspace][:local_ipv6] = "2a00:1a48:7805:111:e875:efaf:ff08:75"
      @plugin.run
      @plugin[:cloud_v2][:local_ipv6].should == @plugin[:rackspace][:local_ipv6]
    end

    it "populates first cloud public ip" do
      @plugin[:rackspace][:public_ipv4] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs].first.should == @plugin[:rackspace][:public_ipv4]
    end

    it "populates first cloud public ip" do
      @plugin[:rackspace][:local_ipv4] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:local_ipv4_addrs].first.should == @plugin[:rackspace][:local_ipv4]
    end

    it "populates cloud provider" do
      @plugin.run
      @plugin[:cloud_v2][:provider].should == "rackspace"
    end
  end

  describe "with linode mash" do
    before do
      @plugin[:linode] = Mash.new()
    end

    it "populates cloud public ip" do
      @plugin[:linode]['public_ip'] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs][0].should == @plugin[:linode][:public_ip]
    end

    it "populates cloud private ip" do
      @plugin[:linode]['private_ip'] = "10.252.42.149"
      @plugin.run
      @plugin[:cloud_v2][:local_ipv4_addrs][0].should == @plugin[:linode][:private_ip]
    end

    it "populates first cloud public ip" do
      @plugin[:linode]['public_ip'] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs].first.should == @plugin[:linode][:public_ip]
    end

    it "populates cloud provider" do
      @plugin.run
      @plugin[:cloud_v2][:provider].should == "linode"
    end
  end

  describe "with eucalyptus mash" do
    before do
      @plugin[:eucalyptus] = Mash.new()
    end

    it "populates cloud public ip" do
      @plugin[:eucalyptus]['public_ipv4'] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs][0].should == @plugin[:eucalyptus]['public_ipv4']
    end

    it "populates cloud private ip" do
      @plugin[:eucalyptus]['local_ipv4'] = "10.252.42.149"
      @plugin.run
      @plugin[:cloud_v2][:local_ipv4_addrs][0].should == @plugin[:eucalyptus]['local_ipv4']
    end

    it "populates cloud provider" do
      @plugin.run
      @plugin[:cloud_v2][:provider].should == "eucalyptus"
    end
  end

  describe "with Azure mash" do
    before do
      @plugin[:azure] = Mash.new()
    end

    it "populates cloud public ip" do
      @plugin[:azure]['public_ip'] = "174.129.150.8"
      @plugin.run
      @plugin[:cloud_v2][:public_ipv4_addrs][0].should == @plugin[:azure]['public_ip']
    end

    it "doesn't populates cloud vm_name" do
      @plugin[:azure]['vm_name'] = "linux-vm"
      @plugin.run
      @plugin[:cloud_v2][:vm_name].should_not == @plugin[:azure]['vm_name']
    end

    it "populates cloud public_hostname" do
      @plugin[:azure]['public_fqdn'] = "linux-vm-svc.cloudapp.net"
      @plugin.run
      @plugin[:cloud_v2][:public_hostname].should == @plugin[:azure]['public_fqdn']
    end

    it "doesn't populate cloud public_ssh_port" do
      @plugin[:azure]['public_ssh_port'] = "22"
      @plugin.run
      @plugin[:cloud_v2][:public_ssh_port].should be_nil
    end

    it "should not populate cloud public_ssh_port when winrm is used" do
      @plugin[:azure]['public_winrm_port'] = "5985"
      @plugin.run
      @plugin[:cloud_v2][:public_ssh_port].should be_nil
    end

    it "populates cloud public_winrm_port" do
      @plugin[:azure]['public_winrm_port'] = "5985"
      @plugin.run
      @plugin[:cloud_v2][:public_winrm_port].should be_nil
    end

    it "populates cloud provider" do
      @plugin.run
      @plugin[:cloud_v2][:provider].should == "azure"
    end
  end

end