allowed 0.1.4 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7af153f66450cb6df34f81701dc3cd967bdc51c4
-  data.tar.gz: 2b0899d05b5a3dfa6d3166631eab8299548aaca3
+  metadata.gz: e450c5a1540f74c95fa6da6c5fc675b93b38272f
+  data.tar.gz: 077515cf12e15f52c1fd65ec5c6c0d13933e2925
 SHA512:
-  metadata.gz: 3b33a0f054d66a70805b00e562a4784048089cdba366e5b9e5c78af2c82aa40550cb477ae99a49f8c82f98a655b6082de375b569a9b3aa3433c5ec4476d381a3
-  data.tar.gz: d8dd767fa5548e1fccad3f5f9c629ec63923c8ce399e673318dcda48002a3280c45c04665f71f78222713fc019fa7079e1767b75dbdf050957a07b9e508c5930
+  metadata.gz: 1c7cab82212758d6722e60b6ee310ccb7dc50632aeed7e9a8d5992746642e2572e95f8fd39afbd90811246ca6e8693e9562e1a73a00f5d8d524259d98a044311
+  data.tar.gz: 64d5c21ce51dcffbcbf08866735c9489ddc09e4a621c7b4540b60bae2122504e6bccf2bcfa43240f6148673e8bad08230807d520c71b0e9c5ec58ae64ca81760

data/lib/allowed/throttle.rb

@@ -18,7 +18,7 @@ module Allowed
     def valid?(record)
       return true if skip?(record)
 
-      scope_for(record).count < limit
+      scope_for(record).count < allowed_count(record)
     end
 
     private
@@ -44,5 +44,14 @@ module Allowed
     def timeframe
       options.fetch(:per, 1.day).ago
     end
+
+    def allowed_count(record)
+      case limit
+      when Integer
+        limit
+      when Symbol
+        record.__send__(limit)
+      end
+    end
   end
 end

data/spec/lib/allowed/throttle_spec.rb

@@ -102,6 +102,36 @@ describe Allowed::Throttle, "#valid?, above limit" do
   end
 end
 
+describe Allowed::Throttle, "#valid?, with limit method symbol" do
+  subject { Allowed::Throttle.new(:custom_limit) }
+
+  let(:record) { ExampleRecord.new }
+
+  before do
+    2.times { ExampleRecord.create }
+  end
+
+  it "returns true if higher than the count" do
+    ExampleRecord.class_eval do
+      def custom_limit
+        3
+      end
+    end
+
+    expect(subject).to be_valid(record)
+  end
+
+  it "returns false if lower than the count" do
+    ExampleRecord.class_eval do
+      def custom_limit
+        1
+      end
+    end
+
+    expect(subject).to_not be_valid(record)
+  end
+end
+
 describe Allowed::Throttle, "#valid?, with custom timeframe" do
   subject { Allowed::Throttle.new(1, per: 5.minutes) }
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: allowed
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.2.0
 platform: ruby
 authors:
 - Tristan Dunn