rack-pooledthrottle 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f17681ae19c501bb8f4194f2d7e9389f3aae51ee
-  data.tar.gz: 6349f697567a2cbc6f3a04248e710dbd36c253eb
+  metadata.gz: 93f678ed4ba1803eb76022038615a86360dcf110
+  data.tar.gz: 487c6c5332cc4469e6bba491a2233210d0db9468
 SHA512:
-  metadata.gz: 2431de24d6e9c40b2f5e0b2b0f5bfdca57d5ae38b5b99e3c3cec3698c258f3cb56d52efbd30472071ca568046b5c5dcbc1784fff2208880ec31eb080eb8ecf1d
-  data.tar.gz: 68eec81e9a722fc99bcbdc66a3cec630ed20500188854bea77ca48f77c83b46a950958b2756a0061f97569ccb4908e5ddbb5cf8e57f3307ad9772552251309f7
+  metadata.gz: 8c4f7308d1fa6628ce4fbfee90a68719de280646f5a47d4a79056cdd95d3835f906489a21c75f74213ceea3b1c6eb7eb49527b5d2fbdfaa80746964f57cf4328
+  data.tar.gz: 4d39db54a3c1a065ad50ece2638705639281fdedb982087ae1ecd8d334feaef1f20bb9228a0f5bd0f6af349a2e5edbd4b3affb32e548a7f7a78544083acac04a

data/.gitignore

@@ -12,3 +12,4 @@
 *.o
 *.a
 mkmf.log
+rack-pooled*.gem

data/lib/rack/pooledthrottle/throttle.rb

@@ -28,11 +28,19 @@ module Rack
       end
 
       def whitelisted?(request)
-        false
+        if wl = options[:whitelisted]
+          wl.call(request)
+        else
+          false
+        end
       end
 
       def blacklisted?(request)
-        false
+        if bl = options[:blacklisted]
+          bl.call(request)
+        else
+          false
+        end
       end
 
       def cache_key(request)

data/lib/rack/pooledthrottle/version.rb

@@ -1,5 +1,5 @@
 module Rack
   module Pooledthrottle
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/spec/memecached_spec.rb

@@ -45,8 +45,30 @@ describe Rack::PooledThrottle::MemcachedThrottle do
     expect(last_response.body).to show_throttled_response
   end
   
+  it "should allow blacklisted? to be overridden" do
+    @options[:blacklisted] = ->(request){true}
+    get "/foo"
+    expect(last_response.body).to show_throttled_response
+  end
+
+  it "should allow whitelisted? to be overridden" do
+    @options[:whitelisted] = ->(request){true}
+    4.times {get "/foo"}
+    expect(last_response.body).to show_allowed_response
+  end
+  
+  it 'should execute the rate_limit_exceeded_callback if it is defined' do 
+    callback = Object.new
+    expect(callback).to receive(:call)
+    @options[:rate_limit_exceeded_callback] = callback
+    @options[:max] = 0
+    get '/foo'
+    expect(last_response.body).to show_throttled_response
+    
+  end
+  
   
-  it 'should allow the client_identifier to be overriden and pass' do 
+  it 'should allow the client_identifier to be overridden and pass' do 
     @options[:max] = 2
     @options[:client_identifier] = ->(request){request.params['email']}
     2.times {get('/foo', email: 'scottwater@gmail.com')}
@@ -54,7 +76,7 @@ describe Rack::PooledThrottle::MemcachedThrottle do
     expect(last_response.body).to show_allowed_response
   end
 
-  it 'should allow the client_identifier to be overriden and fail' do 
+  it 'should allow the client_identifier to be overridden and fail' do 
     @options[:max] = 2
     @options[:client_identifier] = ->(request){request.params['email']}
     2.times {get('/foo', email: 'scottwater@gmail.com')}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rack-pooledthrottle
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Scott Watermasysk
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-09-08 00:00:00.000000000 Z
+date: 2015-09-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack