Sha256: 587287cffd820a5c0b192c8d8bfba0f6a7f8c225395956bffad1861484aa8ee5
Contents?: true
Size: 1.93 KB
Versions: 22
Compression:
Stored size: 1.93 KB
Contents
require 'helper'
require 'sidekiq'
describe Bearcat::RateLimiting do
let!(:subject) { Bearcat::RateLimiting::RedisLimiter.new }
let!(:client) { Bearcat::Client.new(rate_limiter: subject, token: SecureRandom.hex) }
let!(:token) { client.send(:rate_limit_key) }
def fillup(count, rate: 100)
count.times do
subject.apply(token) do
rate
end
end
end
it "fills the bucket with each request" do
allow(Time).to receive(:now).and_return(Time.at(5000))
expect(subject).to_not receive(:sleep)
fillup(5)
expect(subject.increment(token, 0, 0)).to eql({ count: 500.0, timestamp: 5000.0 })
end
it "drains the bucket over time" do
allow(Time).to receive(:now).and_return(Time.at(5000))
expect(subject).to_not receive(:sleep)
fillup(5)
allow(Time).to receive(:now).and_return(Time.at(5009))
expect(subject.increment(token, 0, 0)).to eql({ count: 410.0, timestamp: 5009.0 })
end
it "sleeps requests when full" do
allow(Time).to receive(:now).and_return(Time.at(5000))
expect(subject).to receive(:sleep).at_least(:once)
fillup(10)
end
it "rate limits requests" do
allow(Time).to receive(:now).and_return(Time.at(5000))
allow(subject).to receive(:sleep)
fillup(10)
stub_request(:get, /.*/).to_return(body: "{}")
expect(subject).to receive(:sleep).once
client.course(1)
end
it "retries after a Canvas 403 - Rate Limited" do
stub_request(:get, /.*/).to_return(status: 403, body: "(Rate Limit Exceeded)")
expect(subject).to receive(:apply).exactly(3).times.and_call_original
expect(subject).to receive(:sleep).twice
expect { client.course(1) }.to raise_error(Footrest::HttpError::Forbidden)
end
it "pushes x-rate-limit-remaining to the bucket" do
stub_request(:get, /.*/).to_return(headers: { 'X-Rate-Limit-Remaining' => 120 }, body: "{}")
expect(subject).to receive(:checkin_known).with(token, 20)
client.course(1)
end
end
Version data entries
22 entries across 22 versions & 1 rubygems