#
# 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')

describe Ohai::System, "plugin rackspace" do
  before(:each) do
    @plugin = get_plugin("rackspace")
    @plugin[:hostname] = "katie"
    @plugin[:network] = {:interfaces => {:eth0 => {"addresses"=> {
      "1.2.3.4"=> {
        "broadcast"=> "67.23.20.255",
        "netmask"=> "255.255.255.0",
        "family"=> "inet"
      },
      "2a00:1a48:7805:111:e875:efaf:ff08:75"=> {
        "family"=> "inet6",
        "prefixlen"=> "64",
        "scope"=> "Global"
      },
      "fe80::4240:95ff:fe47:6eed"=> {
        "scope"=> "Link",
        "prefixlen"=> "64",
        "family"=> "inet6"
      },
      "40:40:95:47:6E:ED"=> {
        "family"=> "lladdr"
      }
      }}
    }
  }

  @plugin[:network][:interfaces][:eth1] = {:addresses => {
    "fe80::4240:f5ff:feab:2836" => {
      "scope"=> "Link",
      "prefixlen"=> "64",
      "family"=> "inet6"
    },
    "5.6.7.8"=> {
      "broadcast"=> "10.176.191.255",
      "netmask"=> "255.255.224.0",
      "family"=> "inet"
    },
    "40:40:F5:AB:28:36" => {
      "family"=> "lladdr"
    }
    }}

    # In olden days we could detect rackspace by a -rscloud suffix on the kernel
    # This is here to make #has_rackspace_kernel? fail until we remove that check
    @plugin[:kernel] = { :release => "1.2.13-not-rackspace" }

    # We need a generic stub here for the later stubs with arguments to work
    # Because, magic.
    @plugin.stub(:shell_out).and_return(mock_shell_out(1, "", ""))
  end

  shared_examples_for "!rackspace"  do
    it "should NOT create rackspace" do
      @plugin.run
      @plugin[:rackspace].should be_nil
    end
  end

  shared_examples_for "rackspace" do

    it "should create rackspace" do
      @plugin.run
      @plugin[:rackspace].should_not be_nil
    end

    it "should have all required attributes" do
      @plugin.run
      @plugin[:rackspace][:public_ip].should_not be_nil
      @plugin[:rackspace][:private_ip].should_not be_nil
      @plugin[:rackspace][:public_ipv4].should_not be_nil
      @plugin[:rackspace][:local_ipv4].should_not be_nil
      @plugin[:rackspace][:public_ipv6].should_not be_nil
      @plugin[:rackspace][:local_ipv6].should be_nil
      @plugin[:rackspace][:local_hostname].should_not be_nil
      @plugin[:rackspace][:public_hostname].should_not be_nil
    end

    it "should have correct values for all attributes" do
      @plugin.run
      @plugin[:rackspace][:public_ip].should == "1.2.3.4"
      @plugin[:rackspace][:private_ip].should == "5.6.7.8"
      @plugin[:rackspace][:public_ipv4].should == "1.2.3.4"
      @plugin[:rackspace][:local_ipv4].should == "5.6.7.8"
      @plugin[:rackspace][:public_ipv6].should == "2a00:1a48:7805:111:e875:efaf:ff08:75"
      @plugin[:rackspace][:local_hostname].should == 'katie'
      @plugin[:rackspace][:public_hostname].should == "1-2-3-4.static.cloud-ips.com"
    end

    it "should capture region information" do
      provider_data = <<-OUT
provider = "Rackspace"
service_type = "cloudServers"
server_id = "21301000"
created_at = "2012-12-06T22:08:16Z"
region = "dfw"
OUT
      @plugin.stub(:shell_out).with("xenstore-ls vm-data/provider_data").and_return(mock_shell_out(0, provider_data, ""))
      @plugin.run
      @plugin[:rackspace][:region].should == "dfw"
    end
  end

  describe "with rackspace cloud file" do
    it_should_behave_like "rackspace"

    before(:each) do
      File.stub(:exist?).with('/etc/chef/ohai/hints/rackspace.json').and_return(true)
      File.stub(:read).with('/etc/chef/ohai/hints/rackspace.json').and_return('')
      File.stub(:exist?).with('C:\chef\ohai\hints/rackspace.json').and_return(true)
      File.stub(:read).with('C:\chef\ohai\hints/rackspace.json').and_return('')
    end

    describe 'with no public interfaces (empty eth0)' do
      before do
        # unset public (eth0) addresses
        @plugin[:network][:interfaces][:eth0]['addresses'] = {} 
      end

      it "should have all required attributes" do
        @plugin.run
        # expliticly nil
        @plugin[:rackspace][:public_ip].should be_nil
        @plugin[:rackspace][:public_ipv4].should be_nil
        @plugin[:rackspace][:public_ipv6].should be_nil
        @plugin[:rackspace][:public_hostname].should be_nil
        # per normal
        @plugin[:rackspace][:private_ip].should_not be_nil
        @plugin[:rackspace][:local_ipv4].should_not be_nil
        @plugin[:rackspace][:local_ipv6].should be_nil
        @plugin[:rackspace][:local_hostname].should_not be_nil
      end
  
      it "should have correct values for all attributes" do
        @plugin.run
        @plugin[:rackspace][:private_ip].should == "5.6.7.8"
        @plugin[:rackspace][:local_ipv4].should == "5.6.7.8"
        @plugin[:rackspace][:local_hostname].should == 'katie'
      end
    end
  end

  describe "without cloud file" do
    it_should_behave_like "!rackspace"
  
    before(:each) do
      File.stub(:exist?).with('/etc/chef/ohai/hints/rackspace.json').and_return(false)
      File.stub(:exist?).with('C:\chef\ohai\hints/rackspace.json').and_return(false)
    end
  end
  
  describe "with ec2 cloud file" do
    it_should_behave_like "!rackspace"
  
    before(:each) do
      File.stub(:exist?).with('/etc/chef/ohai/hints/ec2.json').and_return(true)
      File.stub(:read).with('/etc/chef/ohai/hints/ec2.json').and_return('')
      File.stub(:exist?).with('C:\chef\ohai\hints/ec2.json').and_return(true)
      File.stub(:read).with('C:\chef\ohai\hints/ec2.json').and_return('')

      File.stub(:exist?).with('/etc/chef/ohai/hints/rackspace.json').and_return(false)
      File.stub(:exist?).with('C:\chef\ohai\hints/rackspace.json').and_return(false)
    end
  end

  describe "xenstore provider returns rackspace" do
    it_should_behave_like "rackspace"

    before(:each) do
      stdout = "Rackspace\n"
      @plugin.stub(:shell_out).with("xenstore-read vm-data/provider_data/provider").and_return(mock_shell_out(0, stdout, "" ))
    end
  end

  describe "xenstore provider does not return rackspace" do
    it_should_behave_like "!rackspace"

    before(:each) do
      stdout = "cumulonimbus\n"
      @plugin.stub(:shell_out).with("xenstore-read vm-data/provider_data/provider").and_return(mock_shell_out(0, stdout, "" ))
    end
  end
end