dnsxl_check_spec.rb 2.16 KB
Newer Older
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