add api limiter test
This commit is contained in:
43
test/unit/api_limiter_test.rb
Normal file
43
test/unit/api_limiter_test.rb
Normal file
@@ -0,0 +1,43 @@
|
||||
require 'test_helper'
|
||||
|
||||
class ApiLimiterTest < ActiveSupport::TestCase
|
||||
context "for reads" do
|
||||
context "for an anonymous user" do
|
||||
setup do
|
||||
@count = 5
|
||||
@user = AnonymousUser.new
|
||||
CurrentUser.user = @user
|
||||
end
|
||||
|
||||
should "respect api limits" do
|
||||
@user.expects(:api_hourly_limit).with(true).times(@count + 1).returns(@count)
|
||||
|
||||
@count.times do
|
||||
assert_equal(false, ApiLimiter.throttled?(CurrentUser.id || "127.0.0.1", "GET"))
|
||||
end
|
||||
|
||||
assert_equal(true, ApiLimiter.throttled?(CurrentUser.id || "127.0.0.1", "GET"))
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
context "for writes" do
|
||||
context "for an anonymous user" do
|
||||
setup do
|
||||
@count = 5
|
||||
@user = AnonymousUser.new
|
||||
CurrentUser.user = @user
|
||||
end
|
||||
|
||||
should "respect api limits" do
|
||||
@user.expects(:api_hourly_limit).with(false).times(@count + 1).returns(@count)
|
||||
|
||||
@count.times do
|
||||
assert_equal(false, ApiLimiter.throttled?(CurrentUser.id || "127.0.0.1", "POST"))
|
||||
end
|
||||
|
||||
assert_equal(true, ApiLimiter.throttled?(CurrentUser.id || "127.0.0.1", "POST"))
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user