require 'test_plugin_helper'

class DiscoveredHostsControllerTest < ActionController::TestCase
  setup :initialize_host

  setup do
    @request.env['HTTP_REFERER'] = '/discovery_rules'
    @facts = {
      "interfaces"             => "lo,eth0",
      "ipaddress"              => "192.168.100.42",
      "ipaddress_eth0"         => "192.168.100.42",
      "macaddress_eth0"        => "AA:BB:CC:DD:EE:FF",
      "discovery_bootif"       => "AA:BB:CC:DD:EE:FF",
      "physicalprocessorcount" => "42",
    }
    FactoryGirl.create(:setting,
                       :name => 'discovery_reboot',
                       :value => true,
                       :category => 'Setting::Discovered')
  end

  def test_index
    get :index, {}, set_session_user
    assert_response :success
  end

  def test_index_with_custom_column
    FactoryGirl.create(:setting,
                       :name => 'discovery_fact_column',
                       :value => "bios_vendor",
                       :category => 'Setting::Discovered')
    facts = @facts.merge({"bios_vendor" => "QEMU"})
    Host::Discovered.import_host_and_facts(facts).first
    get :index, {}, set_session_user
    assert_select "td", /QEMU/
    assert_response :success
  end

  def test_edit_form_elements
    host = Host::Discovered.import_host_and_facts(@facts).first
    get :edit, {:id => host.id}, set_session_user
    assert_select "select" do |elements|
      elements.each do |element|
        assert_match(/^host_/, element.attributes['id'])
        assert_match(/^host\[/, element.attributes['name'])
      end
    end
  end

  def test_edit_form_attributes
    host = Host::Discovered.import_host_and_facts(@facts).first
    Host.transaction do
      get :edit, {:id => host.id}, set_session_user
    end
    assert_not_nil host.cpu_count
  end

  def test_add_entry_to_nav_menu
    get :index, {}, set_session_user
    assert_tag :tag        => 'a',
               :attributes => { :href => '/discovered_hosts' }
  end

  def test_reboot_success
    @request.env["HTTP_REFERER"] = discovered_hosts_url
    host = FactoryGirl.create(:host,
                              :ip   => '1.2.3.4',
                              :type => "Host::Discovered")
    ::ProxyAPI::BMC.any_instance.stubs(:power).returns(true)
    post "reboot", { :id => host.id }, set_session_user
    assert_redirected_to discovered_hosts_url
    assert_equal "Rebooting host #{host.name}", flash[:notice]
  end

  def test_reboot_failure
    @request.env["HTTP_REFERER"] = discovered_hosts_url
    host = FactoryGirl.create(:host,
                              :ip   => '1.2.3.4',
                              :type => "Host::Discovered")
    ::ProxyAPI::BMC.any_instance.stubs(:power).returns(false)
    post "reboot", { :id => host.id }, set_session_user
    assert_redirected_to discovered_hosts_url
    assert_equal "Failed to reboot host #{host.name}", flash[:error]
  end

  def test_reboot_error
    @request.env["HTTP_REFERER"] = discovered_hosts_url
    host = FactoryGirl.create(:host,
                              :ip   => '1.2.3.4',
                              :type => "Host::Discovered")
    ::ProxyAPI::BMC.any_instance.expects(:power).raises("request failed")
    post "reboot", { :id => host.id }, set_session_user
    assert_redirected_to discovered_hosts_url
    assert_match(/ERF50-6734/, flash[:error])
  end

  def test_auto_provision_success
    disable_orchestration
    facts = @facts.merge({"somefact" => "abc"})
    host = Host::Discovered.import_host_and_facts(facts).first
    FactoryGirl.create(:discovery_rule, :priority => 1, :search => "facts.somefact = abc", :hostgroup => hostgroups(:common))
    post :auto_provision, { :id => host.id }, set_session_user
    assert_response :redirect
    assert_nil flash[:error]
  end

  def test_auto_provision_no_rule_success
    disable_orchestration
    facts = @facts.merge({"somefact" => "abc"})
    host = Host::Discovered.import_host_and_facts(facts).first
    post :auto_provision, { :id => host.id }, set_session_user
    assert_response :redirect
    assert_nil flash[:error]
  end

  def test_auto_provision_all_success
    disable_orchestration
    facts = @facts.merge({"somefact" => "abc"})
    Host::Discovered.import_host_and_facts(facts).first
    FactoryGirl.create(:discovery_rule, :priority => 1, :search => "facts.somefact = abc", :hostgroup => hostgroups(:common))
    post :auto_provision_all, {}, set_session_user
    assert_response :redirect
    assert_nil flash[:error]
  end

  def test_auto_provision_all_no_rule_success
    disable_orchestration
    facts = @facts.merge({"somefact" => "abc"})
    Host::Discovered.import_host_and_facts(facts).first
    FactoryGirl.create(:discovery_rule, :priority => 1, :search => "facts.somefact = abc", :hostgroup => hostgroups(:common))
    post :auto_provision_all, {}, set_session_user
    assert_response :redirect
    assert_nil flash[:error]
  end

  def test_reboot_all_success
    @request.env["HTTP_REFERER"] = discovered_hosts_url
    host = FactoryGirl.create(:host,
                              :ip   => '1.2.3.4',
                              :type => "Host::Discovered")
    ::ProxyAPI::BMC.any_instance.stubs(:power).returns(true)
    post "reboot", { :id => host.id }, set_session_user
    assert_redirected_to discovered_hosts_url
    assert_equal "Rebooting host #{host.name}", flash[:notice]
  end

  def test_reboot_all_failure
    @request.env["HTTP_REFERER"] = discovered_hosts_url
    host = FactoryGirl.create(:host,
                              :ip   => '1.2.3.4',
                              :type => "Host::Discovered")
    ::ProxyAPI::BMC.any_instance.stubs(:power).returns(false)
    post "reboot_all", { }, set_session_user
    assert_redirected_to discovered_hosts_url
    assert_equal "Errors during reboot: #{host.name}: failed to reboot", flash[:error]
  end

  def test_reboot_all_error
    @request.env["HTTP_REFERER"] = discovered_hosts_url
    FactoryGirl.create(:host,
                       :ip   => '1.2.3.4',
                       :type => "Host::Discovered")
    ::ProxyAPI::BMC.any_instance.expects(:power).raises("request must fail")
    post "reboot_all", { }, set_session_user
    assert_redirected_to discovered_hosts_url
    assert_match(/ERF50-6734/, flash[:error])
  end

  private

  def initialize_host
    User.current = users(:admin)
  end
end