require 'test_plugin_helper'

class RemoteExecutionProviderTest < ActiveSupport::TestCase
  describe '.providers' do
    let(:providers) { RemoteExecutionProvider.providers }
    it { providers.must_be_kind_of HashWithIndifferentAccess }
    it 'makes providers accessible using symbol' do
      providers[:SSH].must_equal SSHExecutionProvider
    end
    it 'makes providers accessible using string' do
      providers['SSH'].must_equal SSHExecutionProvider
    end
  end

  describe '.register_provider' do
    before { RemoteExecutionProvider.providers.delete(:new) }
    let(:new_provider) { RemoteExecutionProvider.providers[:new] }
    it { new_provider.must_be_nil }

    context 'registers a provider under key :new' do
      before { RemoteExecutionProvider.register(:new, String) }
      it { new_provider.must_equal String }
    end
  end

  describe '.provider_for' do
    it 'accepts symbols' do
      RemoteExecutionProvider.provider_for(:SSH).must_equal SSHExecutionProvider
    end

    it 'accepts strings' do
      RemoteExecutionProvider.provider_for('SSH').must_equal SSHExecutionProvider
    end
  end

  describe '.provider_names' do
    let(:provider_names) { RemoteExecutionProvider.provider_names }

    it 'returns only strings' do
      provider_names.each do |name|
        name.must_be_kind_of String
      end
    end

    context 'provider is registetered under :custom symbol' do
      before { RemoteExecutionProvider.register(:Custom, String) }

      it { provider_names.must_include 'SSH' }
      it { provider_names.must_include 'Custom' }
    end
  end

  describe SSHExecutionProvider do
    before { User.current = FactoryGirl.build(:user, :admin) }
    after { User.current = nil }

    before do
      Setting::RemoteExecution.load_defaults
    end

    let(:job_invocation) { FactoryGirl.create(:job_invocation, :with_template) }
    let(:template_invocation) { job_invocation.pattern_template_invocations.first }
    let(:host) { FactoryGirl.create(:host) }
    let(:proxy_options) { SSHExecutionProvider.proxy_command_options(template_invocation, host) }

    describe 'effective user' do
      it 'takes the effective user from value from the template invocation' do
        template_invocation.effective_user = 'my user'
        proxy_options[:effective_user].must_equal 'my user'
      end
    end

    describe 'ssh user' do
      it 'uses the remote_execution_ssh_user on the host param' do
        host.params['remote_execution_ssh_user'] = 'my user'
        host.host_parameters << FactoryGirl.create(:host_parameter, :host => host, :name => 'remote_execution_ssh_user', :value => 'my user')
        proxy_options[:ssh_user].must_equal 'my user'
      end
    end

    describe 'sudo' do
      it 'uses the remote_execution_ssh_user on the host param' do
        host.params['remote_execution_effective_user_method'] = 'sudo'
        method_param = FactoryGirl.create(:host_parameter, :host => host, :name => 'remote_execution_effective_user_method', :value => 'sudo')
        host.host_parameters << method_param
        proxy_options[:effective_user_method].must_equal 'sudo'
        method_param.update_attributes!(:value => 'su')
        host.clear_host_parameters_cache!
        proxy_options = SSHExecutionProvider.proxy_command_options(template_invocation, host)
        proxy_options[:effective_user_method].must_equal 'su'
      end
    end

    describe 'ssh port from settings' do
      before do
        Setting[:remote_execution_ssh_port] = '66'
      end

      it 'has ssh port changed in settings and check return type' do
        proxy_options[:ssh_port].must_be_kind_of Integer
        proxy_options[:ssh_port].must_equal 66
      end
    end

    describe 'ssh port from params' do
      it 'takes ssh port number from params and check return type' do
        host.params['remote_execution_ssh_port'] = '30'
        host.host_parameters << FactoryGirl.build(:host_parameter, :name => 'remote_execution_ssh_port', :value => '30')
        host.clear_host_parameters_cache!
        proxy_options[:ssh_port].must_be_kind_of Integer
        proxy_options[:ssh_port].must_equal 30
      end
    end

    describe '#find_ip_or_hostname' do
      let(:host) do
        FactoryGirl.create(:host) do |host|
          host.interfaces = [FactoryGirl.build(:nic_managed, flags.merge(:ip => nil, :name => 'somehost.somedomain.org', :primary => true)),
                             FactoryGirl.build(:nic_managed, flags.merge(:ip => '127.0.0.1'))]
        end
      end

      let(:flags) do
        { :primary => false, :provision => false, :execution => false }
      end

      it 'gets fqdn from flagged interfaces if not preferring ips' do
        # falling to primary interface
        SSHExecutionProvider.find_ip_or_hostname(host).must_equal 'somehost.somedomain.org'

        # execution wins if present
        execution_interface = FactoryGirl.build(:nic_managed,
                                                flags.merge(:execution => true, :name => 'special.somedomain.org'))
        host.interfaces << execution_interface
        host.primary_interface.update_attributes(:execution => false)
        host.interfaces.each(&:save)
        host.reload
        SSHExecutionProvider.find_ip_or_hostname(host).must_equal execution_interface.fqdn
      end

      it 'gets ip from flagged interfaces' do
        host.params['remote_execution_connect_by_ip'] = true
        # no ip address set on relevant interface - fallback to fqdn
        SSHExecutionProvider.find_ip_or_hostname(host).must_equal 'somehost.somedomain.org'

        # provision interface with ip while primary without
        provision_interface = FactoryGirl.build(:nic_managed,
                                                flags.merge(:provision => true, :ip => '10.0.0.1'))
        host.interfaces << provision_interface
        host.primary_interface.update_attributes(:provision => false)
        host.interfaces.each(&:save)
        host.reload
        SSHExecutionProvider.find_ip_or_hostname(host).must_equal provision_interface.ip

        # both primary and provision interface have IPs: the primary wins
        host.primary_interface.update_attributes(:ip => '10.0.0.2', :execution => false)
        host.reload
        SSHExecutionProvider.find_ip_or_hostname(host).must_equal host.primary_interface.ip

        # there is an execution interface with IP: it wins
        execution_interface = FactoryGirl.build(:nic_managed,
                                                flags.merge(:execution => true, :ip => '10.0.0.3'))
        host.interfaces << execution_interface
        host.primary_interface.update_attributes(:execution => false)
        host.interfaces.each(&:save)
        host.reload
        SSHExecutionProvider.find_ip_or_hostname(host).must_equal execution_interface.ip
      end
    end
  end
end