require 'test_plugin_helper'

module ForemanSalt
  class HostsControllerExtensionsTest < ActionController::TestCase
    tests ::HostsController

    describe "setting salt master proxy on multiple hosts" do
      before do
        setup_user "edit"
        as_admin do
          @hosts = FactoryBot.create_list(:host, 2)
          @proxy = FactoryBot.create(:smart_proxy, :with_salt_feature)
        end
      end

      test 'user without edit permission should not be able to change salt master' do
        @request.env['HTTP_REFERER'] = hosts_path

        params = { :host_ids => @hosts.map(&:id),
                   :proxy => { :proxy_id => '' } }

        post :update_multiple_salt_master, params: params,
          session: set_session_user.merge(:user => users(:one).id)
        assert_response :forbidden
      end

      test "should change the salt master proxy" do
        @request.env['HTTP_REFERER'] = hosts_path

        params = { :host_ids => @hosts.map(&:id),
                   :proxy => { :proxy_id => @proxy.id } }

        post :update_multiple_salt_master, params: params,
        session: set_session_user.merge(:user => users(:admin).id)

        assert_empty flash[:error]

        @hosts.each do |host|
          as_admin do
            assert_equal @proxy, host.reload.salt_proxy
          end
        end
      end

      test "should clear the salt master proxy of multiple hosts" do
        @request.env['HTTP_REFERER'] = hosts_path

        params = { :host_ids => @hosts.map(&:id),
                   :proxy => { :proxy_id => '' } }

        post :update_multiple_salt_master, params: params,
          session: set_session_user.merge(:user => users(:admin).id)

        assert_empty flash[:error]

        @hosts.each do |host|
          as_admin do
            assert_nil host.reload.salt_proxy
          end
        end
      end
    end

    describe "setting salt environment on multiple hosts" do
      before do
        setup_user "edit"
        as_admin do
          @hosts = FactoryBot.create_list(:host, 2)
          @proxy = FactoryBot.create(:smart_proxy, :with_salt_feature)
          @salt_environment = FactoryBot.create :salt_environment
        end
      end

      test 'user without edit permission should not be able to change salt environment' do
        @request.env['HTTP_REFERER'] = hosts_path

        params = { :host_ids => @hosts.map(&:id),
                   :salt_environment => { :id => @salt_environment.id } }

        post :update_multiple_salt_environment, params: params,
        session: set_session_user.merge(:user => users(:one).id)
        assert_response :forbidden
      end

      test "should change the salt environment" do
        @request.env['HTTP_REFERER'] = hosts_path

        params = { :host_ids => @hosts.map(&:id),
                   :salt_environment => { :id => @salt_environment.id } }

        post :update_multiple_salt_environment, params: params,
        session: set_session_user.merge(:user => users(:admin).id)

        assert_empty flash[:error]

        @hosts.each do |host|
          as_admin do
            assert_equal @salt_environment, host.reload.salt_environment
          end
        end
      end

      test "should clear the salt environment of multiple hosts" do
        @request.env['HTTP_REFERER'] = hosts_path

        params = { :host_ids => @hosts.map(&:id),
                   :salt_environment => { :id => '' } }

        post :update_multiple_salt_environment, params: params,
          session: set_session_user.merge(:user => users(:admin).id)

        assert_empty flash[:error]

        @hosts.each do |host|
          as_admin do
            assert_nil host.reload.salt_environment
          end
        end
      end
    end
  end
end