#
# Author:: Adam Jacob (<adam@chef.io>)
# Copyright:: Copyright (c) 2008-2016 Chef Software, Inc.
# 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 "spec_helper"
require "wmi-lite/wmi"
require "socket"

describe Ohai::System, "hostname plugin" do
  before(:each) do
    @plugin = get_plugin("hostname")
    allow(@plugin).to receive(:collect_os).and_return(:default)
    allow(@plugin).to receive(:shell_out).with("hostname").and_return(mock_shell_out(0, "katie.local", ""))
  end

  context "default behavior"
  before(:each) do
    allow(@plugin).to receive(:resolve_fqdn).and_return("katie.bethell")
  end
  it_should_check_from("linux::hostname", "machinename", "hostname", "katie.local")

  it "should use #resolve_fqdn to find the fqdn" do
    @plugin.run
    expect(@plugin[:fqdn]).to eq("katie.bethell")
  end

  it "should set the domain to everything after the first dot of the fqdn" do
    @plugin.run
    expect(@plugin[:domain]).to eq("bethell")
  end

  it "should set the [short] hostname to everything before the first dot of the fqdn" do
    @plugin.run
    expect(@plugin[:hostname]).to eq("katie")
  end

  context "when a system has a bare hostname without a FQDN" do
    before(:each) do
      allow(@plugin).to receive(:collect_os).and_return(:default)
      allow(@plugin).to receive(:shell_out).with("hostname").and_return(mock_shell_out(0, "katie", ""))
    end

    it "should correctly set the [short] hostname" do
      @plugin.run
      expect(@plugin[:hostname]).to eq("katie")
    end
  end

  context "hostname --fqdn when it returns empty string" do
    before(:each) do
      allow(@plugin).to receive(:collect_os).and_return(:linux)
      allow(@plugin).to receive(:shell_out).with("hostname -s").and_return(
        mock_shell_out(0, "katie", ""))
      allow(@plugin).to receive(:shell_out).with("hostname --fqdn").and_return(
        mock_shell_out(0, "", ""), mock_shell_out(0, "katie.local", ""))
    end

    it "should be called twice" do
      @plugin.run
      expect(@plugin[:fqdn]).to eq("katie.local")
    end
  end

  context "hostname --fqdn when it works" do
    before(:each) do
      allow(@plugin).to receive(:collect_os).and_return(:linux)
      allow(@plugin).to receive(:shell_out).with("hostname -s").and_return(
        mock_shell_out(0, "katie", ""))
      allow(@plugin).to receive(:shell_out).with("hostname --fqdn").and_return(
        mock_shell_out(0, "katie.local", ""))
    end

    it "should be not be called twice" do
      @plugin.run
      expect(@plugin[:fqdn]).to eq("katie.local")
    end
  end
end

describe Ohai::System, "hostname plugin for windows", :windows_only do
  let(:success) { true }

  let(:host) do
    {
      "name" => "local",
      "dnshostname" => "local",
    }
  end

  let(:info) do
    [
      "local",
      [],
      23,
      "address1",
      "address2",
      "address3",
      "address4"
    ]
  end

  let(:local_hostent) do
    [
      "local",
      [],
      23,
      "address"
    ]
  end

  let(:fqdn_hostent) do
    [
      "local.dx.internal.cloudapp.net",
      [],
      23,
      "address"
    ]
  end

  before(:each) do
    @plugin = get_plugin("hostname")
    allow(WmiLite::Wmi).to receive(:new).and_return(success)
    allow(success).to receive(:first_of).with("Win32_ComputerSystem").and_return(host)
    allow(Socket).to receive(:gethostname).and_return("local")
    allow(Socket).to receive(:gethostbyname).with(anything()).and_return(info)
  end

  context "when hostname is not set for the machine" do
    it "should return short machine name" do
      allow(Socket).to receive(:gethostbyaddr).with(anything()).and_return(local_hostent)
      @plugin.run
      expect(@plugin[:fqdn]).to eq("local")
    end
  end

  context "when hostname is set for the machine" do
    it "should return the fqdn of the machine" do
      allow(Socket).to receive(:gethostbyaddr).with(anything()).and_return(fqdn_hostent)
      @plugin.run
      expect(@plugin[:fqdn]).to eq("local.dx.internal.cloudapp.net")
    end
  end
end