verify takes 2 params, second being payload

jwt · Sep 19, 2017 · e348ee3 · e348ee3
1 parent b9b220a
commit e348ee3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/jwt/verify.rb b/lib/jwt/verify.rb
@@ -63,7 +63,8 @@ def verify_jti
       jti = @payload['jti']
 
       if options_verify_jti.respond_to?(:call)
-        raise(JWT::InvalidJtiError, 'Invalid jti') unless options_verify_jti.call(jti)
+        verified = options_verify_jti.arity == 2 ? options_verify_jti.call(jti, @payload) : options_verify_jti.call(jti)
+        raise(JWT::InvalidJtiError, 'Invalid jti') unless verified
       elsif jti.to_s.strip.empty?
         raise(JWT::InvalidJtiError, 'Missing jti')
       end

diff --git a/spec/jwt/verify_spec.rb b/spec/jwt/verify_spec.rb
@@ -182,6 +182,19 @@ module JWT
       it 'true proc should not raise JWT::InvalidJtiError' do
         Verify.verify_jti(payload, options.merge(verify_jti: ->(_jti) { true }))
       end
+
+      it 'it should not throw arguement error with 2 args' do
+        expect do
+          Verify.verify_jti(payload, options.merge(verify_jti: ->(_jti, pl) {
+            true
+          }))
+        end.to_not raise_error
+      end
+      it 'should have payload as second param in proc' do
+        Verify.verify_jti(payload, options.merge(verify_jti: ->(_jti, pl) {
+          expect(pl).to eq(payload)
+        }))
+      end
     end
 
     context '.verify_not_before(payload, options)' do