spec/lib/dnsxl_check_spec.rb


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
62
63
64
65
66
67
68

require 'spec_helper'
require 'ostruct'

describe 'DNSXLCheck', lib: true, no_db: true do
  let(:spam_ip)    { '127.0.0.2' }
  let(:no_spam_ip) { '127.0.0.3' }
  let(:invalid_ip) { 'a.b.c.d' }
  let!(:dnsxl_check) { DNSXLCheck.create_from_list([OpenStruct.new({ domain: 'test', weight: 1 })]) }

  before(:context) do
    class DNSXLCheck::Resolver
      class << self
        alias_method :old_search, :search
        def search(query)
          return false if query.match(/always\.failing\.domain\z/)
          return true  if query.match(/\A2\.0\.0\.127\./)
          return false if query.match(/\A3\.0\.0\.127\./)
        end
      end
    end
  end

  describe '#test' do
    before do
      dnsxl_check.threshold = 0.75
      dnsxl_check.add_list('always.failing.domain', 1)
    end

    context 'when threshold is used' do
      before { dnsxl_check.use_threshold= true }

      it { expect(dnsxl_check.test(spam_ip)).to be_falsey }
    end

    context 'when threshold is not used' do
      before { dnsxl_check.use_threshold= false }

      it { expect(dnsxl_check.test(spam_ip)).to be_truthy }
    end
  end

  describe '#test_with_threshold' do
    it { expect{ dnsxl_check.test_with_threshold(invalid_ip) }.to raise_error(ArgumentError) }

    it { expect(dnsxl_check.test_with_threshold(spam_ip)).to    be_truthy }
    it { expect(dnsxl_check.test_with_threshold(no_spam_ip)).to be_falsey }
  end

  describe '#test_strict' do
    before do
      dnsxl_check.threshold = 1
      dnsxl_check.add_list('always.failing.domain', 1)
    end

    it { expect{ dnsxl_check.test_strict(invalid_ip) }.to raise_error(ArgumentError) }

    it { expect(dnsxl_check.test_with_threshold(spam_ip)).to    be_falsey }
    it { expect(dnsxl_check.test_with_threshold(no_spam_ip)).to be_falsey }
    it { expect(dnsxl_check.test_strict(spam_ip)).to    be_truthy }
    it { expect(dnsxl_check.test_strict(no_spam_ip)).to be_falsey }
  end

  describe '#threshold=' do
    it { expect{ dnsxl_check.threshold = 0   }.to     raise_error(ArgumentError) }
    it { expect{ dnsxl_check.threshold = 1.1 }.to     raise_error(ArgumentError) }
    it { expect{ dnsxl_check.threshold = 0.5 }.not_to raise_error }
  end
end