# frozen_string_literal: true

require 'spec_helper'

RSpec.describe RSpec::Hive::Connector do
  describe '#start_connection' do
    let(:connector) { described_class.new(configuration) }
    let(:tcli_connection) { class_double(RBHive::TCLIConnection) }
    let(:connection_delegator) { double(RSpec::Hive::ConnectionDelegator) }
    let(:host) { '127.0.0.1' }
    let(:port) { '10000' }
    let(:options_mock) { double('options') }
    let(:hive_options) do
      {'hive.exec.dynamic.partition' => 'true',
       'hive.exec.dynamic.partition.mode' => 'nonstrict',
       'hive.exec.max.dynamic.partitions.pernode' => '100000',
       'hive.exec.max.dynamic.partitions' => '100000',
       'mapred.child.java.opts' => '-Xmx2048m'}
    end
    let(:configuration) do
      instance_double(
        RSpec::Hive::Configuration,
        host: host,
        port: port,
        hive_options: hive_options,
        logger: instance_double(Logger)
      )
    end

    before do
      allow(connector).to receive(:connection_options) { options_mock }
      allow(configuration.logger).to receive(:info)
    end

    context 'when db_name is provided' do
      subject { connector.start_connection(db_name) }

      let(:db_name) { 'test' }

      before do
        expect(RBHive::TCLIConnection).to receive(:new).with(host, port, options_mock) { tcli_connection }
        expect(RSpec::Hive::ConnectionDelegator).to receive(:new).
          with(tcli_connection, configuration) { connection_delegator }
        expect(connection_delegator).to receive(:open).once
        expect(connection_delegator).to receive(:open_session).once
        expect(connection_delegator).to receive(:switch_database).with(db_name).once
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.dynamic.partition=true')
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.dynamic.partition.mode=nonstrict')
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.max.dynamic.partitions.pernode=100000')
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.max.dynamic.partitions=100000')
        expect(connection_delegator).to receive(:execute).with('SET mapred.child.java.opts=-Xmx2048m')
      end

      it { is_expected.to equal(connection_delegator) }
    end

    context 'when db_name is not provided' do
      subject { connector.start_connection }

      let(:db_random_name) { 'rand123' }

      before do
        expect(RSpec::Hive::DbName).to receive(:random_name) { db_random_name }
        expect(RBHive::TCLIConnection).to receive(:new).with(host, port, options_mock) { tcli_connection }
        expect(RSpec::Hive::ConnectionDelegator).to receive(:new).
          with(tcli_connection, configuration) { connection_delegator }
        expect(connection_delegator).to receive(:open).once
        expect(connection_delegator).to receive(:open_session).once
        expect(connection_delegator).to receive(:switch_database).with(db_random_name).once
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.dynamic.partition=true')
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.dynamic.partition.mode=nonstrict')
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.max.dynamic.partitions.pernode=100000')
        expect(connection_delegator).to receive(:execute).with('SET hive.exec.max.dynamic.partitions=100000')
        expect(connection_delegator).to receive(:execute).with('SET mapred.child.java.opts=-Xmx2048m')
      end

      it { is_expected.to equal(connection_delegator) }
    end
  end
end