blob: 6ac0608d4851de8bc46072513861c4c2939fb589 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::Database::LoadBalancing::SrvResolver do
let(:resolver) { Net::DNS::Resolver.new(nameservers: '127.0.0.1', port: 8600, use_tcp: true) }
let(:additional) { dns_response_packet_from_fixture('srv_with_a_rr_in_additional_section').additional }
describe '#address_for' do
let(:host) { 'patroni-02-db-gstg.node.east-us-2.consul.' }
subject { described_class.new(resolver, additional).address_for(host) }
context 'when additional section contains an A record' do
it 'returns an IP4 address' do
expect(subject).to eq(IPAddr.new('10.224.29.102'))
end
end
context 'when additional section contains an AAAA record' do
let(:host) { 'a.gtld-servers.net.' }
let(:additional) { dns_response_packet_from_fixture('a_with_aaaa_rr_in_additional_section').additional }
it 'returns an IP6 address' do
expect(subject).to eq(IPAddr.new('2001:503:a83e::2:30'))
end
end
context 'when additional section does not contain A nor AAAA records' do
let(:additional) { [] }
context 'when host resolves to an A record' do
before do
allow(resolver).to receive(:search).with(host, Net::DNS::ANY).and_return(dns_response_packet_from_fixture('a_rr'))
end
it 'returns an IP4 address' do
expect(subject).to eq(IPAddr.new('10.224.29.102'))
end
end
context 'when host does resolves to an AAAA record' do
before do
allow(resolver).to receive(:search).with(host, Net::DNS::ANY).and_return(dns_response_packet_from_fixture('aaaa_rr'))
end
it 'returns an IP6 address' do
expect(subject).to eq(IPAddr.new('2a00:1450:400e:80a::200e'))
end
end
end
end
def dns_response_packet_from_fixture(fixture_name)
fixture = File.read(Rails.root + "spec/fixtures/dns/#{fixture_name}.json")
encoded_payload = Gitlab::Json.parse(fixture)['payload']
payload = Base64.decode64(encoded_payload)
Net::DNS::Packet.parse(payload)
end
end
|