# frozen_string_literal: true
# encoding: utf-8

require 'spec_helper'

# these tests fail intermittently in evergreen
describe Mongo::Server::Connection do
  retry_test

  let(:address) do
    Mongo::Address.new(SpecConfig.instance.addresses.first)
  end

  let(:monitoring) do
    Mongo::Monitoring.new(monitoring: false)
  end

  let(:listeners) do
    Mongo::Event::Listeners.new
  end

  let(:app_metadata) do
    Mongo::Server::AppMetadata.new(SpecConfig.instance.test_options)
  end

  let(:cluster) do
    double('cluster').tap do |cl|
      allow(cl).to receive(:topology).and_return(topology)
      allow(cl).to receive(:app_metadata).and_return(app_metadata)
      allow(cl).to receive(:options).and_return({})
      allow(cl).to receive(:cluster_time).and_return(nil)
      allow(cl).to receive(:update_cluster_time)
      allow(cl).to receive(:run_sdam_flow)
      pool = double('pool')
      allow(pool).to receive(:disconnect!)
      allow(cl).to receive(:pool).and_return(pool)
    end
  end

  declare_topology_double

  let(:server) do
    register_server(
      Mongo::Server.new(address, cluster, monitoring, listeners,
        SpecConfig.instance.test_options.merge(monitoring_io: false))
    )
  end

  before(:all) do
    ClientRegistry.instance.close_all_clients
  end

  describe '#auth_mechanism' do
    require_no_external_user

    let(:connection) do
      described_class.new(server, server.options)
    end

    context 'when the hello response includes saslSupportedMechs' do
      min_server_fcv '4.0'

      let(:server_options) do
        SpecConfig.instance.test_options.merge(
          user: SpecConfig.instance.test_user.name,
          password: SpecConfig.instance.test_user.password,
          auth_source: 'admin',
        )
      end

      let(:app_metadata) do
        Mongo::Server::AppMetadata.new(server_options)
      end

      before do
        client = authorized_client.with(database: 'admin')
        info = client.database.users.info(SpecConfig.instance.test_user.name)
        expect(info.length).to eq(1)
        # this before block may have made 2 or 3 clients
        ClientRegistry.instance.close_all_clients
      end

      it 'uses scram256' do
        connection
        RSpec::Mocks.with_temporary_scope do
          pending_conn = nil
          Mongo::Server::PendingConnection.should receive(:new).and_wrap_original do |m, *args|
            pending_conn = m.call(*args)
          end
          connection.connect!
          expect(pending_conn.send(:default_mechanism)).to eq(:scram256)
        end
      end
    end

    context 'when the hello response indicates the auth mechanism is :scram' do
      require_no_external_user

      let(:features) do
        Mongo::Server::Description::Features.new(0..7)
      end

      it 'uses scram' do
        connection
        RSpec::Mocks.with_temporary_scope do
          expect(Mongo::Server::Description::Features).to receive(:new).and_return(features)

          pending_conn = nil
          Mongo::Server::PendingConnection.should receive(:new).and_wrap_original do |m, *args|
            pending_conn = m.call(*args)
          end
          connection.connect!
          expect(pending_conn.send(:default_mechanism)).to eq(:scram)
        end
      end
    end

    context 'when the hello response indicates the auth mechanism is :mongodb_cr' do
      let(:features) do
        Mongo::Server::Description::Features.new(0..2)
      end

      it 'uses mongodb_cr' do
        connection
        RSpec::Mocks.with_temporary_scope do
          expect(Mongo::Server::Description::Features).to receive(:new).and_return(features)

          pending_conn = nil
          Mongo::Server::PendingConnection.should receive(:new).and_wrap_original do |m, *args|
            pending_conn = m.call(*args)
          end
          connection.connect!
          expect(pending_conn.send(:default_mechanism)).to eq(:mongodb_cr)
        end
      end
    end
  end
end