# frozen_string_literal: true

require 'test_helper'

module Aws
  # Test services namespaces
  module Service1
    Client = Aws::SES::Client.dup
  end

  module Service2
    Client = Aws::SES::Client.dup
  end

  module Rails
    describe 'Railtie' do
      it 'adds action mailer delivery method' do
        expect(ActionMailer::Base.delivery_methods[:ses]).to eq Aws::Rails::SesMailer
        expect(ActionMailer::Base.delivery_methods[:sesv2]).to eq Aws::Rails::Sesv2Mailer
      end

      it 'sets the Aws logger' do
        expect(Aws.config[:logger]).to eq ::Rails.logger
      end

      describe '.use_rails_encrypted_credentials' do
        let(:rails_creds) { ::Rails.application.credentials.aws }
        it 'sets aws credentials' do
          expect(Aws.config[:access_key_id]).to eq rails_creds[:access_key_id]
          expect(Aws.config[:secret_access_key]).to eq rails_creds[:secret_access_key]
        end

        it 'does not load non credential keys into aws config' do
          expect(rails_creds[:non_credential_key]).not_to be_nil
          expect(Aws.config[:non_credential_key]).to be_nil
        end
      end

      describe '.instrument_sdk_operations' do
        it 'adds the Notifications plugin to sdk clients' do
          expect(Aws::Service1::Client).to receive(:add_plugin).with(Aws::Rails::Notifications)
          expect(Aws::Service2::Client).to receive(:add_plugin).with(Aws::Rails::Notifications)

          # Ensure other Clients don't get plugin added
          allow_any_instance_of(Class).to receive(:add_plugin)

          Aws::Rails.instrument_sdk_operations
        end
      end

      describe '.add_sqsd_middleware' do
        after(:each) do
          ENV.delete('AWS_PROCESS_BEANSTALK_WORKER_REQUESTS')
        end

        it 'adds middleware when AWS_PROCESS_BEANSTALK_WORKER_REQUESTS is set to true' do
          ENV['AWS_PROCESS_BEANSTALK_WORKER_REQUESTS'] = 'True'
          mock_rails_app = double
          mock_middleware_stack = []

          allow(mock_rails_app).to receive_message_chain(:config, :middleware, :use) do |middleware|
            mock_middleware_stack.push(middleware)
          end
          allow(mock_rails_app).to receive_message_chain(:config, :force_ssl).and_return(false)

          Aws::Rails.add_sqsd_middleware(mock_rails_app)

          expect(mock_middleware_stack.count).to eq(1)
          expect(mock_middleware_stack[0].inspect).to eq('Aws::Rails::EbsSqsActiveJobMiddleware')
        end
        it 'does not add middleware when AWS_PROCESS_BEANSTALK_WORKER_REQUESTS is not true' do
          ENV['AWS_PROCESS_BEANSTALK_WORKER_REQUESTS'] = 'False'
          mock_rails_app = double
          mock_middleware_stack = []

          allow(mock_rails_app).to receive_message_chain(:config, :middleware, :use) do |middleware|
            mock_middleware_stack.push(middleware)
          end
          allow(mock_rails_app).to receive_message_chain(:config, :force_ssl).and_return(false)

          Aws::Rails.add_sqsd_middleware(mock_rails_app)

          expect(mock_middleware_stack.count).to eq(0)
        end

        it 'does not add middleware when AWS_PROCESS_BEANSTALK_WORKER_REQUESTS is missing' do
          mock_rails_app = double
          mock_middleware_stack = []

          allow(mock_rails_app).to receive_message_chain(:config, :middleware, :use) do |middleware|
            mock_middleware_stack.push(middleware)
          end
          allow(mock_rails_app).to receive_message_chain(:config, :force_ssl).and_return(false)

          Aws::Rails.add_sqsd_middleware(mock_rails_app)

          expect(mock_middleware_stack.count).to eq(0)
        end
      end
    end
  end
end