require production_code

describe Snmpjr::Getter do
  let(:target) { double :target }
  let(:session) { double Snmpjr::Session }
  let(:pdu) { double Snmpjr::Pdu }
  let(:created_pdu_single) { double :created_pdu_single }

  subject { described_class.new(:target => target, :max_oids_per_request => 1) }

  before do
    allow(Snmpjr::Pdu).to receive(:new).and_return pdu
    allow(pdu).to receive(:create).with(['1.2.3.4.5.6']).and_return created_pdu_single
    allow(Snmpjr::Session).to receive(:new).and_return session
    allow(session).to receive(:start)
    allow(session).to receive(:send).with(created_pdu_single, target).and_return ['Foo']
    allow(session).to receive(:close)
  end

  describe "#get_multiple" do
    let(:created_pdu_multiple_1) { double :created_pdu_multiple_1 }
    let(:created_pdu_multiple_2) { double :created_pdu_multiple_2 }

    before do
      allow(pdu).to receive(:create).with(['1.2.3.4.5.6']).and_return created_pdu_multiple_1
      allow(pdu).to receive(:create).with(['6.5.4.3.2.1']).and_return created_pdu_multiple_2
      allow(session).to receive(:send).with(created_pdu_multiple_1, target).and_return ['Foo']
      allow(session).to receive(:send).with(created_pdu_multiple_2, target).and_return ['Bar']
    end

    it 'starts an snmp session' do
      subject.get_multiple ['1.2.3.4.5.6', '6.5.4.3.2.1']
      expect(session).to have_received(:start)
    end

    it 'performs multiple gets for each oid' do
      expect(subject.get_multiple ['1.2.3.4.5.6', '6.5.4.3.2.1']).to eq(['Foo', 'Bar'])
      expect(session).to have_received(:send).with(created_pdu_multiple_1, target)
      expect(session).to have_received(:send).with(created_pdu_multiple_2, target)
    end

    it 'closes the snmp session' do
      subject.get_multiple ['1.2.3.4.5.6', '6.5.4.3.2.1']
      expect(session).to have_received(:close)
    end
  end

  describe "#get" do
    it 'starts an snmp session' do
      subject.get '1.2.3.4.5.6'
      expect(session).to have_received(:start)
    end

    it 'performs a synchronous get' do
      expect(subject.get '1.2.3.4.5.6').to eq 'Foo'
      expect(session).to have_received(:send).with(created_pdu_single, target)
    end

    it 'closes the snmp session' do
      subject.get '1.2.3.4.5.6'
      expect(session).to have_received(:close)
    end
  end
end