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