54 lines
1.4 KiB
Ruby
54 lines
1.4 KiB
Ruby
require 'minitest/autorun'
|
|
require 'prism'
|
|
|
|
class RateLimiterMemoryStoreTest < Minitest::Test
|
|
def setup
|
|
@store = Prism::RateLimiter::MemoryStore.new
|
|
end
|
|
|
|
def test_count_increments_within_the_same_window
|
|
assert_equal 1, @store.count('ip:1', 60)
|
|
assert_equal 2, @store.count('ip:1', 60)
|
|
assert_equal 3, @store.count('ip:1', 60)
|
|
end
|
|
|
|
def test_count_is_independent_per_key
|
|
assert_equal 1, @store.count('ip:1', 60)
|
|
assert_equal 1, @store.count('ip:2', 60)
|
|
end
|
|
|
|
def test_count_resets_after_the_window_elapses
|
|
assert_equal 1, @store.count('ip:1', 0.05)
|
|
assert_equal 2, @store.count('ip:1', 0.05)
|
|
sleep 0.08
|
|
assert_equal 1, @store.count('ip:1', 0.05)
|
|
end
|
|
|
|
def test_banned_is_false_when_never_banned
|
|
refute @store.banned?('ip:1')
|
|
end
|
|
|
|
def test_ban_makes_banned_true_until_it_expires
|
|
@store.ban!('ip:1', 0.05)
|
|
assert @store.banned?('ip:1')
|
|
sleep 0.08
|
|
refute @store.banned?('ip:1')
|
|
end
|
|
|
|
def test_reset_clears_windows_and_bans
|
|
@store.count('ip:1', 60)
|
|
@store.ban!('ip:2', 60)
|
|
@store.reset!
|
|
assert_equal 1, @store.count('ip:1', 60)
|
|
refute @store.banned?('ip:2')
|
|
end
|
|
|
|
def test_concurrent_increments_do_not_lose_updates
|
|
threads = Array.new(20) do
|
|
Thread.new { 100.times { @store.count('shared', 60) } }
|
|
end
|
|
threads.each(&:join)
|
|
assert_equal 2001, @store.count('shared', 60)
|
|
end
|
|
end
|