#
# 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
    Resolv.stub(:getname).and_return("1.2.3.4")
    @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 resolve hostname if reverse dns is set" do
      Resolv.stub(:getname).and_return("1234.resolved.com")
      @plugin.run
      @plugin[:rackspace][:public_hostname].should == "1234.resolved.com"
    end

    [Resolv::ResolvError, Resolv::ResolvTimeout].each do |exception|
      it "should return ip address when reverse dns returns exception: #{exception}" do
        Resolv.stub(:getname).and_raise(exception)
        @plugin.run
        @plugin[:rackspace][:public_hostname].should == "1.2.3.4"
      end
    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"
    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
      Resolv.stub(:getname).and_raise(Resolv::ResolvError)
      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('')
      File.stub(:exist?).with('/etc/resolv.conf').and_return(true)
      File.stub(:read).with('/etc/resolv.conf').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

  describe "does not have private networks" do
    before do
      stdout = 'BC764E20422B = "{"label": "public"}"\n'
      @plugin.stub(:shell_out).with("xenstore-ls vm-data/networking").and_return(mock_shell_out(0, stdout, "" ))
      stdout = '{"label": "public", "broadcast": "9.10.11.255", "ips": [{"ip": "9.10.11.12", "netmask": "255.255.255.0", "enabled": "1", "gateway": null}], "mac": "BC:76:4E:20:42:2B", "dns": ["69.20.0.164", "69.20.0.196"], "gateway": null}'
      @plugin.stub(:shell_out).with("xenstore-read vm-data/networking/BC764E20422B").and_return(mock_shell_out(0, stdout, "" ))

      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

    it "should not have private_networks object" do
      @plugin.run
      @plugin[:rackspace][:private_networks].should == []
    end
  end

  describe "has private networks" do
    before do
      @plugin[:network][:interfaces][:eth2] = {:addresses => {
        "fe80::be76:4eff:fe20:422b" => {
          "scope"=> "Link",
          "prefixlen"=> "64",
          "family"=> "inet6"
       },
        "9.10.11.12"=> {
          "broadcast"=> "9.10.11.255",
          "netmask"=> "255.255.255.0",
          "family"=> "inet"
        },
        "BC:76:4E:20:42:2B" => {
          "family"=> "lladdr"
        }
      }}
      stdout = 'BC764E20422B = "{"label": "private-network"}"\n'
      @plugin.stub(:shell_out).with("xenstore-ls vm-data/networking").and_return(mock_shell_out(0, stdout, "" ))
      stdout = '{"label": "private-network", "broadcast": "9.10.11.255", "ips": [{"ip": "9.10.11.12", "netmask": "255.255.255.0", "enabled": "1", "gateway": null}], "mac": "BC:76:4E:20:42:2B", "dns": ["69.20.0.164", "69.20.0.196"], "gateway": null}'
      @plugin.stub(:shell_out).with("xenstore-read vm-data/networking/BC764E20422B").and_return(mock_shell_out(0, stdout, "" ))

      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

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

    it "should have correct values for all attributes" do
      @plugin.run
      @plugin[:rackspace][:private_networks][0][:label].should == "private-network"
      @plugin[:rackspace][:private_networks][0][:broadcast].should == "9.10.11.255"
      @plugin[:rackspace][:private_networks][0][:mac].should == "BC:76:4E:20:42:2B"
    end

  end



end