lib/wp/hmac.rb in wp-hmac-0.1.1 vs lib/wp/hmac.rb in wp-hmac-0.2.1

@@ -1,6 +1,6 @@
-require "wp/hmac/version"
+require 'wp/hmac/version'
 require 'rubygems'
 require 'bundler/setup'
 Bundler.require
 require 'rack/client'
 require 'ey_api_hmac'
@@ -32,11 +32,16 @@
     def self.auth_id
       raise MissingConfiguration('Set get_auth_id_for_request in the initializer') unless @callable
       @callable.call
     end
 
+    def self.lookup_auth_key_with(&block)
+      KeyCabinet.lookup_block = block
+    end
+
     def self.reset
       KeyCabinet.keys = {}
+      KeyCabinet.lookup_block = nil
       Server.hmac_enabled_routes = []
     end
   end
 end