Add unauthorized_entity to authenticate_for's default for called by c…

…allback (nsarno#228)

- Add unauthorized_entity to authenticate_for's default

- Remove unused argument from unauthorized_entity

CHANGELOG.md

@@ -2,6 +2,9 @@
 All notable changes to this project will be documented in this file.
 This project adheres to [Semantic Versioning](http://semver.org/).
 
+### Fixed
+- Add unauthorized_entity to authenticate_for's default for called by callback
+
 ## [2.1.1] - 2017-02-11
 ### Fixed
 - Stop trying to retrieve user from empty payload when no token is given

lib/knock/authenticable.rb

@@ -1,8 +1,12 @@
 module Knock::Authenticable
-  def authenticate_for entity_class
+  def authenticate_for entity_class, by_callback: true
     getter_name = "current_#{entity_class.to_s.parameterize.underscore}"
     define_current_entity_getter(entity_class, getter_name)
-    public_send(getter_name)
+    if by_callback
+      unauthorized_entity unless public_send(getter_name)
+    else
+      public_send(getter_name)
+    end
   end
 
   private
@@ -15,7 +19,7 @@ def method_missing(method, *args)
     prefix, entity_name = method.to_s.split('_', 2)
     case prefix
     when 'authenticate'
-      unauthorized_entity(entity_name) unless authenticate_entity(entity_name)
+      unauthorized_entity unless authenticate_entity(entity_name)
     when 'current'
       authenticate_entity(entity_name)
     else
@@ -26,11 +30,11 @@ def method_missing(method, *args)
   def authenticate_entity(entity_name)
     if token
       entity_class = entity_name.camelize.constantize
-      send(:authenticate_for, entity_class)
+      send(:authenticate_for, entity_class, by_callback: false)
     end
   end
 
-  def unauthorized_entity(entity_name)
+  def unauthorized_entity
     head(:unauthorized)
   end
 

test/dummy/app/controllers/custom_unauthorized_entity_controller.rb

@@ -7,7 +7,7 @@ def index
 
   private
 
-  def unauthorized_entity(entity)
+  def unauthorized_entity
     head :not_found
   end
 end