104 lines
2.5 KiB
Ruby
104 lines
2.5 KiB
Ruby
|
require 'spec_helper'
|
||
|
|
require 'rate_limiter'
|
||
|
|
|
||
|
|
describe RateLimiter do
|
||
|
|
|
||
|
|
let(:user) { Fabricate(:user) }
|
||
|
|
let(:rate_limiter) { RateLimiter.new(user, "peppermint-butler", 2, 60) }
|
||
|
|
|
||
|
|
context 'disabled' do
|
||
|
|
before do
|
||
|
|
RateLimiter.stubs(:disabled?).returns(true)
|
||
|
|
rate_limiter.performed!
|
||
|
|
rate_limiter.performed!
|
||
|
|
end
|
||
|
|
|
||
|
|
it "returns true for can_perform?" do
|
||
|
|
rate_limiter.can_perform?.should be_true
|
||
|
|
end
|
||
|
|
|
||
|
|
it "doesn't raise an error on performed!" do
|
||
|
|
lambda { rate_limiter.performed! }.should_not raise_error
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'enabled' do
|
||
|
|
before do
|
||
|
|
RateLimiter.stubs(:disabled?).returns(false)
|
||
|
|
rate_limiter.clear!
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'never done' do
|
||
|
|
it "should perform right away" do
|
||
|
|
rate_limiter.can_perform?.should be_true
|
||
|
|
end
|
||
|
|
|
||
|
|
it "performs without an error" do
|
||
|
|
lambda { rate_limiter.performed! }.should_not raise_error
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context "multiple calls" do
|
||
|
|
before do
|
||
|
|
rate_limiter.performed!
|
||
|
|
rate_limiter.performed!
|
||
|
|
end
|
||
|
|
|
||
|
|
it "returns false for can_perform when the limit has been hit" do
|
||
|
|
rate_limiter.can_perform?.should be_false
|
||
|
|
end
|
||
|
|
|
||
|
|
it "raises an error the third time called" do
|
||
|
|
lambda { rate_limiter.performed! }.should raise_error
|
||
|
|
end
|
||
|
|
|
||
|
|
context "as an admin/moderator" do
|
||
|
|
|
||
|
|
it "returns true for can_perform if the user is an admin" do
|
||
|
|
user.admin = true
|
||
|
|
rate_limiter.can_perform?.should be_true
|
||
|
|
end
|
||
|
|
|
||
|
|
it "doesn't raise an error when an admin performs the task" do
|
||
|
|
user.admin = true
|
||
|
|
lambda { rate_limiter.performed! }.should_not raise_error
|
||
|
|
end
|
||
|
|
|
||
|
|
it "returns true for can_perform if the user is a mod" do
|
||
|
|
user.trust_level = TrustLevel.Levels[:moderator]
|
||
|
|
rate_limiter.can_perform?.should be_true
|
||
|
|
end
|
||
|
|
|
||
|
|
it "doesn't raise an error when a moderator performs the task" do
|
||
|
|
user.trust_level = TrustLevel.Levels[:moderator]
|
||
|
|
lambda { rate_limiter.performed! }.should_not raise_error
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
context "rollback!" do
|
||
|
|
before do
|
||
|
|
rate_limiter.rollback!
|
||
|
|
end
|
||
|
|
|
||
|
|
it "returns true for can_perform since there is now room" do
|
||
|
|
rate_limiter.can_perform?.should be_true
|
||
|
|
end
|
||
|
|
|
||
|
|
it "raises no error now that there is room" do
|
||
|
|
lambda { rate_limiter.performed! }.should_not raise_error
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
end
|