rack-attack spec

rack_attack_spec.rb

require 'spec_helper'

class MockRequest
  attr_reader :ip, :env

  def initialize hash
    @ip  = hash[:ip]  || '127.0.0.1'
    @env = hash[:env] || {}
  end
end

RSpec.describe 'Block China and Russia' do
  let(:blacklist) { Rack::Attack.blacklists['block China and Russia'] }

  it 'has a list of IP addresses to block' do
    expect(BlockedIP.count).to be > 0
  end

  context 'with a visit from China' do
    let(:request) { MockRequest.new(ip: '89.22.1.1') }

    it 'should block the request' do
      expect(blacklist[request]).to be_truthy
    end

    it 'should be fast' do
      expect(Benchmark.realtime { blacklist[request] }).to be < 0.001
    end
  end

  context 'with a visit from the US of A' do
    let(:request) { MockRequest.new(ip: '70.183.119.185') }

    it 'should not block the request' do
      expect(blacklist[request]).to be_falsey
    end
  end
end